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!!")