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)