def test_process_query_view_parity(self, process_props: ProcessProps):
local_client = DgraphClient(DgraphClientStub("localhost:9080"))
created_proc = get_or_create_process(self, local_client, process_props,)
queried_proc = (
ProcessQuery()
.with_node_key(eq=created_proc.node_key)
.query_first(local_client)
)
assert queried_proc
assert process_props["node_key"] == queried_proc.node_key
assert "Process" == queried_proc.get_node_type()
assert process_props["process_id"] == queried_proc.get_process_id()
assert process_props["arguments"] == escape_dgraph_str(
queried_proc.get_arguments()
)
assert (
process_props["created_timestamp"] == queried_proc.get_created_timestamp()
)
assert None == queried_proc.get_asset()
assert process_props["terminate_time"] == queried_proc.get_terminate_time()
assert process_props["image_name"] == escape_dgraph_str(
queried_proc.get_image_name()
)
assert process_props["process_name"] == escape_dgraph_str(
queried_proc.get_process_name()
)
def upsert(
client: DgraphClient,
type_name: str,
view_type: Type[Viewable],
node_key: str,
node_props: Dict[str, Property],
) -> Viewable:
node_props["node_key"] = node_key
node_props["dgraph.type"] = type_name
for key, value in node_props.items():
if isinstance(value, str):
node_props[key] = escape_dgraph_str(value)
uid = _upsert(client, node_props)
# print(f'uid: {uid}')
node_props["uid"] = uid
# print(node_props['node_key'])
return view_type.from_dict(client, node_props)