def testGetFlowWorksIfFlowWasCreatedBySameRouter(self):
flow_id = self._CreateFlowWithRobotId()
router = self._CreateRouter(get_flow=rr.RobotRouterGetFlowParams(
enabled=True))
router.GetFlow(api_flow.ApiGetFlowArgs(client_id=self.client_id,
flow_id=flow_id),
token=self.token)
def testGetFlowRaisesIfFlowWasNotCreatedBySameRouter(self):
flow_urn = flow.GRRFlow.StartFlow(
client_id=self.client_id,
flow_name=file_finder.FileFinder.__name__,
token=self.token)
router = self._CreateRouter(get_flow=rr.RobotRouterGetFlowParams(
enabled=True))
with self.assertRaises(access_control.UnauthorizedAccess):
router.GetFlow(api_flow.ApiGetFlowArgs(
client_id=self.client_id, flow_id=flow_urn.Basename()),
token=self.token)
