def _StreamBody(
self,
client_id: Text,
flow_id: Text,
) -> api_call_handler_base.ApiBinaryStream:
entries = timeline.Entries(client_id=client_id, flow_id=flow_id)
content = body.Stream(entries)
filename = "timeline_{}.body".format(flow_id)
return api_call_handler_base.ApiBinaryStream(filename, content)
def _Collect(self, root):
args = rdf_timeline.TimelineArgs(root=root)
flow_id = flow_test_lib.TestFlowHelper(
timeline_flow.TimelineFlow.__name__,
action_mocks.ActionMock(timeline_action.Timeline),
client_id=self.client_id,
token=self.token,
args=args)
flow_test_lib.FinishAllFlowsOnClient(self.client_id)
return timeline_flow.Entries(client_id=self.client_id, flow_id=flow_id)