added return from insert method

This commit is contained in:
Ruslan 2022-04-27 09:41:04 +03:00
parent 232a8d29ad
commit 5d4e4ca770
2 changed files with 6 additions and 1 deletions

View File

@ -245,7 +245,7 @@ class Database(object):
# Return any remaining lines in partial batch
if lines:
yield buf.getvalue()
self._send(gen())
return self._send(gen())
def count(self, model_class, conditions=None):
'''

View File

@ -46,6 +46,11 @@ class DatabaseTestCase(TestCaseWithData):
self.assertEqual(str(t.b), '2020-01-01')
self.assertEqual(t.d, 35)
def test_insert__response(self):
from requests import Response
response = self.database.insert(next(self._sample_data()))
self.assertTrue(isinstance(response, Response))
def test_count(self):
self.database.insert(self._sample_data())
self.assertEqual(self.database.count(Person), 100)