def test_application_data():
name = 'test_app'
key = get_random_bytes(16)
key_index = get_random_bytes(2)
network = f'test_net'
num_nodes = 10
nodes = []
for x in range(num_nodes):
nodes.append(f'test_node{x}')
data = ApplicationData(name=name,
key=key,
key_index=key_index,
network=network,
nodes=nodes)
assert file_helper.file_exist(base_dir + app_dir + name + '.yml') is \
False
data.save()
assert file_helper.file_exist(base_dir + app_dir + name + '.yml') is \
True
r_data = ApplicationData.load(base_dir + app_dir + name + '.yml')
assert data == r_data
def test_application_data():
name = 'test_node'
addr = get_random_bytes(2)
devkey = get_random_bytes(16)
device_uuid = get_random_bytes(16)
network = f'test_net'
num_apps = 10
apps = []
for x in range(num_apps):
apps.append(f'test_app{x}')
data = NodeData(name=name,
addr=addr,
network=network,
device_uuid=device_uuid,
devkey=devkey,
apps=apps)
assert file_helper.file_exist(base_dir + node_dir + name + '.yml') is \
False
data.save()
assert file_helper.file_exist(base_dir + node_dir + name + '.yml') is \
True
r_data = NodeData.load(base_dir + node_dir + name + '.yml')
assert data == r_data
def test_network_data():
name = 'test_net'
key = get_random_bytes(16)
key_index = get_random_bytes(2)
iv_index = get_random_bytes(4)
num_apps = 10
num_nodes = 15
apps = []
for x in range(num_apps):
apps.append(f'test_app{x}')
nodes = []
for x in range(num_nodes):
nodes.append(f'test_node{x}')
data = NetworkData(name=name,
key=key,
key_index=key_index,
iv_index=iv_index,
apps=apps,
nodes=nodes)
assert file_helper.file_exist(base_dir + net_dir + name + '.yml') is \
False
data.save()
assert file_helper.file_exist(base_dir + net_dir + name + '.yml') is \
True
r_data = NetworkData.load(base_dir + net_dir + name + '.yml')
assert data == r_data
def test_write_read():
content = {
'val_int': 95,
'val_str': 'october',
'val_bytes': b'\x24',
'val_bool': True
}
assert file_helper.file_exist('write_read_file.yml') is False
file_helper.write('write_read_file.yml', content)
assert file_helper.file_exist('write_read_file.yml') is True
r_content = file_helper.read('write_read_file.yml')
assert r_content['val_int'] == 95
assert r_content['val_str'] == 'october'
assert r_content['val_bytes'] == b'\x24'
assert r_content['val_bool'] is True
def test_load():
assert file_helper.file_exist('load_file.yml') is False
content = file_helper.load('load_file.yml',
default_content={
'val_int': 95,
'val_str': 'october',
'val_bytes': b'\x24',
'val_bool': True
})
assert file_helper.file_exist('load_file.yml') is True
assert content['val_int'] == 95
assert content['val_str'] == 'october'
assert content['val_bytes'] == b'\x24'
assert content['val_bool'] is True
content = file_helper.load('load_file.yml', default_content={})
assert content['val_int'] == 95
assert content['val_str'] == 'october'
assert content['val_bytes'] == b'\x24'
assert content['val_bool'] is True
def test_file_exist():
assert file_helper.file_exist('non_exist_file.yml') is False