def testInterrogateFlowIsStarted(self):
self.assertEmpty(data_store.REL_DB.ReadAllFlowObjects(self.client_id))
args = client_plugin.ApiInterrogateClientArgs(client_id=self.client_id)
result = self.handler.Handle(args, token=self.token)
results = data_store.REL_DB.ReadAllFlowObjects(self.client_id)
self.assertLen(results, 1)
self.assertEqual(result.operation_id, results[0].flow_id)
def testInterrogateFlowIsStarted(self):
flows_fd = aff4.FACTORY.Open(self.client_id.Add("flows"), token=self.token)
flows_urns = list(flows_fd.ListChildren())
self.assertEqual(len(flows_urns), 0)
args = client_plugin.ApiInterrogateClientArgs(client_id=self.client_id)
result = self.handler.Handle(args, token=self.token)
flows_fd = aff4.FACTORY.Open(self.client_id.Add("flows"), token=self.token)
flows_urns = list(flows_fd.ListChildren())
self.assertEqual(len(flows_urns), 1)
self.assertEqual(str(flows_urns[0]), result.operation_id)
def testClientMethodsAreAccessChecked(self):
args = api_client.ApiInterrogateClientArgs(client_id=self.client_id)
self.CheckMethodIsAccessChecked(
self.router.InterrogateClient, "CheckClientAccess", args=args)
args = api_client.ApiListClientCrashesArgs(client_id=self.client_id)
self.CheckMethodIsAccessChecked(
self.router.ListClientCrashes, "CheckClientAccess", args=args)
args = api_client.ApiGetClientLoadStatsArgs(client_id=self.client_id)
self.CheckMethodIsAccessChecked(
self.router.GetClientLoadStats, "CheckClientAccess", args=args)
