From 5d4e4ca770d1bc0870bc44731bfdb759f6cc8731 Mon Sep 17 00:00:00 2001 From: Ruslan Date: Wed, 27 Apr 2022 09:41:04 +0300 Subject: [PATCH] added return from insert method --- src/infi/clickhouse_orm/database.py | 2 +- tests/test_database.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/infi/clickhouse_orm/database.py b/src/infi/clickhouse_orm/database.py index 703e982..9435946 100644 --- a/src/infi/clickhouse_orm/database.py +++ b/src/infi/clickhouse_orm/database.py @@ -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): ''' diff --git a/tests/test_database.py b/tests/test_database.py index 44971e1..96b47da 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -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)