mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
eb74c996a6
Imports read `datastore_driver`, distribution `datastore-driver`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.2.9 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
236 lines
8.7 KiB
Python
236 lines
8.7 KiB
Python
from datetime import date
|
|
|
|
from tests.testcase import BaseTestCase
|
|
from datastore_driver import errors
|
|
from datastore_driver.errors import ServerException
|
|
from tests.util import require_server_version
|
|
|
|
|
|
class InsertTestCase(BaseTestCase):
|
|
def test_type_mismatch(self):
|
|
with self.create_table('a Float32'):
|
|
with self.assertRaises(errors.TypeMismatchError):
|
|
data = [(date(2012, 10, 25), )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data
|
|
)
|
|
|
|
def test_no_such_column(self):
|
|
with self.create_table('a Float32'):
|
|
with self.assertRaises(ServerException):
|
|
data = [(1, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (b) VALUES', data
|
|
)
|
|
|
|
def test_data_malformed_rows(self):
|
|
with self.create_table('a Int8'), self.assertRaises(TypeError):
|
|
data = [1]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data
|
|
)
|
|
|
|
def test_data_less_columns_then_expected(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
self.assertEqual(str(e.exception), 'Expected 2 columns, got 1')
|
|
|
|
def test_data_more_columns_then_expected(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, 2, 3)]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
self.assertEqual(str(e.exception), 'Expected 2 columns, got 3')
|
|
|
|
def test_data_different_rows_length(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, 2), (3, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
self.assertEqual(str(e.exception), 'Different rows length')
|
|
|
|
def test_data_different_rows_length_from_dicts(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(KeyError):
|
|
data = [{'a': 1, 'b': 2}, {'a': 3}]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
|
|
def test_data_unsupported_row_type(self):
|
|
with self.create_table('a Int8'):
|
|
with self.assertRaises(TypeError) as e:
|
|
data = [1]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data, types_check=True
|
|
)
|
|
self.assertIn('dict, list or tuple is expected', str(e.exception))
|
|
|
|
def test_data_dicts_ok(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
data = [{'a': 1, 'b': 2}, {'a': 3, 'b': 4}]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '1\t2\n3\t4\n')
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, [(1, 2), (3, 4)])
|
|
|
|
def test_data_generator_type(self):
|
|
with self.create_table('a Int8'):
|
|
data = ((x, ) for x in range(3))
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data
|
|
)
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '0\n1\n2\n')
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, [(0,), (1, ), (2, )])
|
|
|
|
def test_data_dicts_mixed_with_lists(self):
|
|
with self.create_table('a Int8'):
|
|
with self.assertRaises(TypeError) as e:
|
|
data = [{'a': 1}, (2, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data, types_check=True
|
|
)
|
|
|
|
self.assertIn('dict is expected', str(e.exception))
|
|
|
|
with self.assertRaises(TypeError) as e:
|
|
data = [(1, ), {'a': 2}]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data, types_check=True
|
|
)
|
|
|
|
self.assertIn('list or tuple is expected', str(e.exception))
|
|
|
|
def test_empty_insert(self):
|
|
with self.create_table('a Int8'):
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', []
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '')
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, [])
|
|
|
|
def test_insert_from_select(self):
|
|
with self.create_table('a UInt64'):
|
|
inserted = self.client.execute(
|
|
'INSERT INTO test (a) '
|
|
'SELECT number FROM system.numbers LIMIT 5'
|
|
)
|
|
self.assertEqual(inserted, [])
|
|
|
|
def test_insert_return(self):
|
|
with self.create_table('a Int8'):
|
|
rv = self.client.execute(
|
|
'INSERT INTO test (a) VALUES', []
|
|
)
|
|
self.assertEqual(rv, 0)
|
|
|
|
rv = self.client.execute(
|
|
'INSERT INTO test (a) VALUES', [(x,) for x in range(5)]
|
|
)
|
|
self.assertEqual(rv, 5)
|
|
|
|
@require_server_version(22, 3, 6)
|
|
def test_insert_from_input(self):
|
|
with self.create_table('a Int8'):
|
|
data = [{'a': 1}]
|
|
self.client.execute(
|
|
"INSERT INTO test (a) "
|
|
"SELECT a FROM input ('a Int8') FORMAT Native",
|
|
data, settings={'session_timezone': 'UTC'}
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '1\n')
|
|
|
|
def test_profile_events(self):
|
|
with self.create_table('x Int32'):
|
|
data = [{'x': 1}]
|
|
self.client.execute(
|
|
'INSERT INTO test (x) VALUES', data
|
|
)
|
|
|
|
|
|
class InsertColumnarTestCase(BaseTestCase):
|
|
def test_insert_tuple_ok(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
data = [(1, 2, 3), (4, 5, 6)]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data, columnar=True
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '1\t4\n2\t5\n3\t6\n')
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, [(1, 4), (2, 5), (3, 6)])
|
|
inserted = self.client.execute(query, columnar=True)
|
|
self.assertEqual(inserted, [(1, 2, 3), (4, 5, 6)])
|
|
|
|
def test_insert_data_different_column_length(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, 2, 3), (4, 5)]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data, columnar=True
|
|
)
|
|
self.assertEqual(str(e.exception), 'Expected 3 rows, got 2')
|
|
|
|
def test_data_less_columns_then_expected(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, 2)]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data, columnar=True
|
|
)
|
|
self.assertEqual(str(e.exception), 'Expected 2 columns, got 1')
|
|
|
|
def test_data_more_columns_then_expected(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(ValueError) as e:
|
|
data = [(1, 2), (3, 4), (5, 6)]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data, columnar=True
|
|
)
|
|
self.assertEqual(str(e.exception), 'Expected 2 columns, got 3')
|
|
|
|
def test_data_invalid_types(self):
|
|
with self.create_table('a Int8, b Int8'):
|
|
with self.assertRaises(TypeError) as e:
|
|
data = [(1, 2), {'a': 3, 'b': 4}]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data,
|
|
types_check=True, columnar=True
|
|
)
|
|
|
|
self.assertIn('list or tuple is expected', str(e.exception))
|
|
|
|
with self.assertRaises(TypeError) as e:
|
|
data = [(1, 2), 3]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data,
|
|
types_check=True, columnar=True
|
|
)
|
|
|
|
self.assertIn('list or tuple is expected', str(e.exception))
|