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 testGetFlowRaisesIfFlowWasNotCreatedBySameUser(self):
flow_id = flow_test_lib.StartFlow(
file_finder.FileFinder, self.client_id, creator=self.another_username)
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_id),
token=self.token)
def testGetFlowRaisesIfFlowWasNotCreatedBySameRouter(self):
flow_urn = flow.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)
