def test_single_process_contains_key(
self,
node_key,
process_id,
created_timestamp,
asset_id,
terminate_time,
image_name,
process_name,
arguments,
):
node_key = "test_single_process_contains_key" + str(node_key)
local_client = DgraphClient(DgraphClientStub("localhost:9080"))
get_or_create_process_node(
local_client,
node_key,
process_id,
arguments,
created_timestamp,
asset_id,
terminate_time,
image_name,
process_name,
)
queried_proc = ProcessQuery().query_first(local_client,
contains_node_key=node_key)
# assert process_view.process_id == queried_proc.get_process_id()
assert node_key == queried_proc.node_key
assert "Process" == queried_proc.get_node_type()
assert process_id == queried_proc.get_process_id()
assert arguments == queried_proc.get_arguments()
assert created_timestamp == queried_proc.get_created_timestamp()
assert asset_id == queried_proc.get_asset_id()
assert terminate_time == queried_proc.get_terminate_time()
assert image_name == queried_proc.get_image_name()
assert process_name == queried_proc.get_process_name()
def test_single_process_contains_key(self, process_props: ProcessProps) -> None:
local_client = DgraphClient(DgraphClientStub("localhost:9080"))
created_proc = get_or_create_process(self, local_client, process_props)
# Setup complete, do some queries
queried_proc = ProcessQuery().query_first(
local_client, contains_node_key=created_proc.node_key
)
assert queried_proc
assert created_proc.get_process_id() == queried_proc.get_process_id()
assert created_proc.node_key == queried_proc.node_key
assert "Process" == queried_proc.get_node_type()
assert created_proc.get_arguments() == queried_proc.get_arguments()
assert (
created_proc.get_created_timestamp() == queried_proc.get_created_timestamp()
)
assert created_proc.get_terminate_time() == queried_proc.get_terminate_time()
assert created_proc.get_image_name() == queried_proc.get_image_name()
assert created_proc.get_process_name() == queried_proc.get_process_name()
assert not queried_proc.get_asset()
