def testListFlowResultsWorksIfFlowWasCreatedBySameRouter(self):
flow_id = self._CreateFlowWithRobotId()
router = self._CreateRouter(
list_flow_results=rr.RobotRouterListFlowResultsParams(
enabled=True))
router.ListFlowResults(api_flow.ApiListFlowResultsArgs(
client_id=self.client_id, flow_id=flow_id),
token=self.token)
def testListFlowResultsRaisesIfFlowWasNotCreatedBySameRouter(self):
flow_urn = flow.GRRFlow.StartFlow(
client_id=self.client_id,
flow_name=file_finder.FileFinder.__name__,
token=self.token)
router = self._CreateRouter(
list_flow_results=rr.RobotRouterListFlowResultsParams(
enabled=True))
with self.assertRaises(access_control.UnauthorizedAccess):
router.ListFlowResults(api_flow.ApiListFlowResultsArgs(
client_id=self.client_id, flow_id=flow_urn.Basename()),
token=self.token)