def test_write_2(self):
proto_table_storage = ProtoTableStorage()
proto_table_storage.initialize_from_file(file_name=self.TEST_DATA_4)
proto_table_storage.write(data={'test': self.EXAMPLE_PROTO_1})
proto_table_storage.write(data={'test_1': self.EXAMPLE_PROTO_2})
result_proto = proto_table_storage.read(
params={
'key': 'test_1',
'message_type': OperatorSnapshot
})
self.assertEqual(result_proto, self.EXAMPLE_PROTO_2)
proto_table_storage.write(data={'test_1': self.EXAMPLE_PROTO_3})
result_proto = proto_table_storage.read(
params={
'key': 'test_1',
'message_type': OperatorSnapshot
})
self.assertEqual(result_proto, self.EXAMPLE_PROTO_3)
result = proto_table_storage.read_all()
self.assertDictEqual(
result, {
'test': ProtoUtil.message_to_any(self.EXAMPLE_PROTO_1),
'test_1': ProtoUtil.message_to_any(self.EXAMPLE_PROTO_3),
})
self.assertEqual(proto_table_storage.get_num_entries(), 2)
gclient_ext.cp_file(self.TEST_DATA_2, self.TEST_DATA_4)
def main():
gclient.create_dir_if_not_exist("/galaxy/aa-d/test_from_python")
print(gclient.list_dirs_in_dir("/galaxy/aa-d/"))
print(gclient.list_dirs_in_dir("/home/pslx/Downloads"))
print(gclient.get_attr("/home/pslx/Downloads"))
print(gclient.get_attr("/galaxy/aa-d/test_from_python"))
gclient.create_file_if_not_exist("/galaxy/aa-d/test_from_python/test.txt")
gclient_ext.cp_file("/galaxy/aa-d/test_from_python/test.txt",
"/galaxy/aa-d/test_from_python/test1.txt")
gclient_ext.mv_file("/galaxy/aa-d/test_from_python/test.txt",
"/galaxy/aa-d/test_from_python/test3.txt")
print(gclient.list_dirs_in_dir_recursive("/galaxy/aa-d/test_from_python"))
print(gclient.list_files_in_dir_recursive("/galaxy/aa-d/test_from_python"))
t = time.time()
gclient.read("/galaxy/aa-d/large_test.txt")
print(time.time() - t)
data = gclient.read_multiple(
["/galaxy/aa-d/test_from_python/test1.txt", "/galaxy/aa-d/test3.txt"])
for key, val in data.items():
print(key, val)
data = gclient_ext.read_txts(
["/galaxy/aa-d/test_from_python/test1.txt", "/galaxy/aa-d/test3.txt"])
for key, val in data.items():
print(key, val)
print(gclient.list_cells())
print(gclient.read('/galaxy/aa-d/test.pb'))
print(gclient_ext.read_txt('/galaxy/aa-d/test.pb'))
message = TestMessage()
message.name = "test"
gclient_ext.write_proto_message('/galaxy/aa-d/test1.pb', message)
print(gclient_ext.read_proto_message('/galaxy/aa-d/test1.pb', TestMessage))
print(gclient.check_health("aa"))
gclient.write_multiple(path_data_map={
'/galaxy/aa-d/test_from_python/test2.txt': '123',
'/galaxy/aa-d/test_from_python/test4.txt': '234',
},
mode='a')
gclient_ext.write_proto_messages(
path_data_map={
'/galaxy/aa-d/test_from_python/test2.pb': message,
'/galaxy/aa-d/test_from_python/test4.pb': message,
})
print(
gclient_ext.read_proto_messages(paths=[
'/galaxy/aa-d/test_from_python/test2.pb',
'/galaxy/aa-d/test_from_python/test4.pb'
],
message_type=TestMessage))
print(gclient.create_file_if_not_exist("/LOCAL/test/test.txt"))
print(gclient.create_file_if_not_exist("/galaxy/aa-d/test/test.txt"))
def test_write_1(self):
proto_table_storage = ProtoTableStorage()
proto_table_storage.initialize_from_file(file_name=self.TEST_DATA_3)
proto_table_storage.write(data={'test': self.EXAMPLE_PROTO_1}, )
result_proto = proto_table_storage.read(params={
'key': 'test',
'message_type': NodeSnapshot
})
self.assertEqual(result_proto, self.EXAMPLE_PROTO_1)
gclient_ext.cp_file(self.TEST_DATA_1, self.TEST_DATA_3)
def test_write_from_beginning(self):
default_storage = DefaultStorage()
default_storage.initialize_from_file(file_name=self.TEST_DATA_2)
data = [3, 4, 5]
default_storage.set_config(
config={
'write_rule_type': WriteRuleType.WRITE_FROM_BEGINNING,
})
default_storage.write(data=data)
data = default_storage.read()
gclient_ext.cp_file(self.TEST_DATA_1, self.TEST_DATA_2)
self.assertListEqual(data, ['3,4,5'])
def test_write_from_end(self):
default_storage = DefaultStorage()
default_storage.initialize_from_file(file_name=self.TEST_DATA_2)
data = [3, 4, 5]
default_storage.set_config(
config={
'read_rule_type': ReadRuleType.READ_FROM_END,
})
default_storage.write(data=data)
data = default_storage.read()
gclient_ext.cp_file(self.TEST_DATA_1, self.TEST_DATA_2)
self.assertListEqual(data, ['3,4,5'])