diff --git a/tests/test_influxdb_client_3.py b/tests/test_influxdb_client_3.py index c928f60..a37339f 100644 --- a/tests/test_influxdb_client_3.py +++ b/tests/test_influxdb_client_3.py @@ -148,9 +148,9 @@ async def test_query_async(self): for item in cd.to_list(): assert item in result_list - assert {'data': 'database', 'reference': 'my_db', 'value': -1.0} in result_list - assert {'data': 'sql_query', 'reference': query, 'value': -1.0} in result_list - assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0} in result_list + assert {'data': 'database', 'reference': 'my_db', 'value': -1.0, 'null_field': None} in result_list + assert {'data': 'sql_query', 'reference': query, 'value': -1.0, 'null_field': None} in result_list + assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0, 'null_field': None} in result_list def test_write_api_custom_options_no_error(self): write_options = WriteOptions(write_type=WriteType.batching) diff --git a/tests/test_query.py b/tests/test_query.py index f702f17..2d623d0 100644 --- a/tests/test_query.py +++ b/tests/test_query.py @@ -402,9 +402,10 @@ async def test_query_async_table(self): for item in cd.to_list(): assert item in result_list - assert {'data': 'database', 'reference': 'my_database', 'value': -1.0} in result_list - assert {'data': 'sql_query', 'reference': 'SELECT * FROM data', 'value': -1.0} in result_list - assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0} in result_list + assert {'data': 'database', 'reference': 'my_database', 'value': -1.0, 'null_field': None} in result_list + assert {'data': 'sql_query', 'reference': 'SELECT * FROM data', 'value': -1.0, + 'null_field': None} in result_list + assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0, 'null_field': None} in result_list @asyncio_run async def test_query_async_delayed(self): diff --git a/tests/util/mocks.py b/tests/util/mocks.py index 416e020..088fcb8 100644 --- a/tests/util/mocks.py +++ b/tests/util/mocks.py @@ -62,9 +62,10 @@ def __init__(self): self.data = [ array(['temp', 'temp', 'temp']), array(['kitchen', 'common', 'foyer']), - array([36.9, 25.7, 9.8]) + array([36.9, 25.7, 9.8]), + array([None, None, None]) ] - self.names = ['data', 'reference', 'value'] + self.names = ['data', 'reference', 'value', 'null_field'] def to_tuples(self): response = [] @@ -97,7 +98,8 @@ def do_get(self, context, ticket): tkt_data = [ array([key]), array([tkt[key]]), - array([-1.0]) + array([-1.0]), + array([None]) ] result_table = concat_tables([result_table, Table.from_arrays(tkt_data, names=self.cd.names)]) return RecordBatchStream(result_table, options=self.options)