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)