From 8e535b5b152e9f4318d3df82e1362dab7358d3e4 Mon Sep 17 00:00:00 2001 From: NguyenHoangSon96 Date: Tue, 17 Mar 2026 16:57:07 +0700 Subject: [PATCH 1/3] test: add test for null field cases --- tests/test_influxdb_client_3.py | 6 +++--- tests/util/mocks.py | 8 +++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/test_influxdb_client_3.py b/tests/test_influxdb_client_3.py index c928f60..fbf36e0 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_value': None} in result_list + assert {'data': 'sql_query', 'reference': query, 'value': -1.0, 'null_value': None} in result_list + assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0, 'null_value': None} in result_list def test_write_api_custom_options_no_error(self): write_options = WriteOptions(write_type=WriteType.batching) 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) From 0568cb5285bbf1853d8d92fb72ceea243e1a9097 Mon Sep 17 00:00:00 2001 From: NguyenHoangSon96 Date: Tue, 17 Mar 2026 17:51:45 +0700 Subject: [PATCH 2/3] refactor: variable name for null_field --- tests/test_influxdb_client_3.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_influxdb_client_3.py b/tests/test_influxdb_client_3.py index fbf36e0..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, 'null_value': None} in result_list - assert {'data': 'sql_query', 'reference': query, 'value': -1.0, 'null_value': None} in result_list - assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0, 'null_value': None} 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) From b84ba687b09758260992588e4d54263c24d9aaba Mon Sep 17 00:00:00 2001 From: NguyenHoangSon96 Date: Tue, 17 Mar 2026 17:55:20 +0700 Subject: [PATCH 3/3] test: add null field tests for test_query_async_table --- tests/test_query.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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):