def testPassesFlowArgsThroughIfNoOverridesSpecified(self):
h = rr.ApiRobotCreateFlowHandler(robot_id="foo")
args = api_flow.ApiCreateFlowArgs(client_id=self.client_id.Basename())
args.flow.name = file_finder.FileFinder.__name__
args.flow.args = rdf_file_finder.FileFinderArgs(paths=["foo"])
f = h.Handle(args=args, token=self.token)
self.assertEqual(f.args.paths, ["foo"])
def _CreateFlowWithRobotId(self, flow_name=None, flow_args=None):
flow_name = flow_name or file_finder.FileFinder.__name__
handler = rr.ApiRobotCreateFlowHandler(robot_id=self.robot_id)
flow_result = handler.Handle(api_flow.ApiCreateFlowArgs(
client_id=self.client_id,
flow=api_flow.ApiFlow(name=flow_name, args=flow_args)),
token=self.token)
return flow_result.flow_id
def testOverridesFlowArgsThroughIfOverridesSpecified(self):
override_flow_args = rdf_file_finder.FileFinderArgs(paths=["bar"])
h = rr.ApiRobotCreateFlowHandler(override_flow_args=override_flow_args)
args = api_flow.ApiCreateFlowArgs(client_id=self.client_id)
args.flow.name = file_finder.FileFinder.__name__
args.flow.args = rdf_file_finder.FileFinderArgs(paths=["foo"])
f = h.Handle(args=args, context=self.context)
self.assertEqual(f.args.paths, ["bar"])
def testOverridesFlowNameIfOverrideArgIsSpecified(self):
h = rr.ApiRobotCreateFlowHandler(
override_flow_name=AnotherFileFinder.__name__) # pylint: disable=undefined-variable
args = api_flow.ApiCreateFlowArgs(client_id=self.client_id)
args.flow.name = file_finder.FileFinder.__name__
args.flow.args = rdf_file_finder.FileFinderArgs(paths=["foo"])
f = h.Handle(args=args, context=self.context)
self.assertEqual(f.name, AnotherFileFinder.__name__) # pylint: disable=undefined-variable
