def test_one_device():
with IoTDBContainer("iotdb:dev") as db:
db: IoTDBContainer
session = Session(db.get_container_host_ip(),
db.get_exposed_port(6667))
session.open(False)
if not session.is_open():
print("can't open session")
exit(1)
# insert string records of one device
time_list = [1, 2, 3]
measurements_list = [
["s_01", "s_02", "s_03"],
["s_01", "s_02", "s_03"],
["s_01", "s_02", "s_03"],
]
values_list = [
["False", "22", "33"],
["True", "1", "23"],
["False", "15", "26"],
]
if (session.insert_string_records_of_one_device(
"root.str_test_01.d_01",
time_list,
measurements_list,
values_list,
) < 0):
test_fail()
print_message("insert string records of one device failed")
# insert aligned string record into the database.
time_list = [1, 2, 3]
measurements_list = [
["s_01", "s_02", "s_03"],
["s_01", "s_02", "s_03"],
["s_01", "s_02", "s_03"],
]
values_list = [
["False", "22", "33"],
["True", "1", "23"],
["False", "15", "26"],
]
if (session.insert_aligned_string_records_of_one_device(
"root.str_test_01.d_02",
time_list,
measurements_list,
values_list,
) < 0):
test_fail()
print_message("insert aligned record of one device failed")
# execute raw data query sql statement
session_data_set = session.execute_raw_data_query(
["root.str_test_01.d_02.s_01", "root.str_test_01.d_02.s_02"], 1, 4)
session_data_set.set_fetch_size(1024)
expect_count = 3
actual_count = 0
while session_data_set.has_next():
print(session_data_set.next())
actual_count += 1
session_data_set.close_operation_handle()
if actual_count != expect_count:
test_fail()
print_message("query count mismatch: expect count: " +
str(expect_count) + " actual count: " +
str(actual_count))
assert actual_count == expect_count
# execute last data query sql statement
session_data_set = session.execute_last_data_query(
["root.str_test_01.d_02.s_01", "root.str_test_01.d_02.s_02"], 0)
session_data_set.set_fetch_size(1024)
expect_time = 3
actual_time = session_data_set.next().get_timestamp()
session_data_set.close_operation_handle()
if actual_time != expect_time:
test_fail()
print_message("query count mismatch: expect count: " +
str(expect_time) + " actual count: " +
str(actual_time))
assert actual_time == expect_time
# close session connection.
session.close()
values_list = [["False", "22", "33"], ["True", "1", "23"], ["False", "15", "26"]]
session.insert_string_records_of_one_device(
"root.sg_test_01.d_03",
time_list,
measurements_list,
values_list,
)
with session.execute_raw_data_query(
["root.sg_test_01.d_03.s_01", "root.sg_test_01.d_03.s_02"], 1, 4
) as session_data_set:
session_data_set.set_fetch_size(1024)
while session_data_set.has_next():
print(session_data_set.next())
with session.execute_last_data_query(
["root.sg_test_01.d_03.s_01", "root.sg_test_01.d_03.s_02"], 0
) as session_data_set:
session_data_set.set_fetch_size(1024)
while session_data_set.has_next():
print(session_data_set.next())
# delete storage group
session.delete_storage_group("root.sg_test_01")
# close session connection.
session.close()
print("All executions done!!")
