def testListResultsForListProcessesFlow(self):
process = rdf_client.Process(pid=2,
ppid=1,
cmdline=["cmd.exe"],
exe="c:\\windows\\cmd.exe",
ctime=long(1333718907.167083 * 1e6),
RSS_size=42)
client_urn = self.SetupClients(1)[0]
client_mock = processes_test.ListProcessesMock([process])
flow_urn = flow.GRRFlow.StartFlow(
client_id=client_urn,
flow_name=processes.ListProcesses.__name__,
token=self.token)
for _ in flow_test_lib.TestFlowHelper(flow_urn,
client_mock,
client_id=client_urn,
token=self.token):
pass
result_flow = self.api.Client(client_id=client_urn.Basename()).Flow(
flow_urn.Basename())
results = list(result_flow.ListResults())
self.assertEqual(len(results), 1)
self.assertEqual(process.AsPrimitiveProto(), results[0].payload)
def ProcessFlow():
time.sleep(1)
client_mock = processes_test.ListProcessesMock([])
for _ in flow_test_lib.TestFlowHelper(flow_urn,
client_mock,
client_id=client_urn,
token=self.token):
pass
