def get_status_query_response(cls): query_response = Response() query_response.responseCode = ResponseCode.OK query_response.result = Result() summaries = GetJobUpdateSummariesResult() query_response.result.getJobUpdateSummariesResult = summaries summaries.updateSummaries = [ JobUpdateSummary( updateId="hello", jobKey=AuroraJobKey('west', 'mcc', 'test', 'hello'), user="******", state=JobUpdateState(status=JobUpdateStatus.ROLLING_FORWARD, createdTimestampMs=1411404927, lastModifiedTimestampMs=14114056030)), JobUpdateSummary( updateId="goodbye", jobKey=AuroraJobKey('west', 'mch', 'prod', 'goodbye'), user="******", state=JobUpdateState(status=JobUpdateStatus.ROLLING_BACK, createdTimestampMs=1411300632, lastModifiedTimestampMs=14114092632)), JobUpdateSummary(updateId="gasp", jobKey=AuroraJobKey('west', 'mcq', 'devel', 'gasp'), user="******", state=JobUpdateState( status=JobUpdateStatus.ROLL_FORWARD_PAUSED, createdTimestampMs=1411600891, lastModifiedTimestampMs=1411800891)) ] return query_response
def get_status_query_response(count=1, status=JobUpdateStatus.ROLLED_FORWARD): return Response( responseCode=ResponseCode.OK, result=Result(getJobUpdateSummariesResult=GetJobUpdateSummariesResult( updateSummaries=[ JobUpdateSummary(key=UPDATE_KEY, user="******", state=JobUpdateState( status=status, createdTimestampMs=1411404927, lastModifiedTimestampMs=14114056030)) for i in range(count) ])))
def get_status_query_response(self, count=3): query_response = Response() query_response.responseCode = ResponseCode.OK query_response.result = Result() summaries = GetJobUpdateSummariesResult() query_response.result.getJobUpdateSummariesResult = summaries summaries.updateSummaries = [JobUpdateSummary( updateId="%s" % i, jobKey=self.TEST_JOBKEY.to_thrift(), user="******", state=JobUpdateState( status=JobUpdateStatus.ROLLED_FORWARD, createdTimestampMs=1411404927, lastModifiedTimestampMs=14114056030)) for i in range(count)] return query_response
def get_update_details_response(cls): query_response = Response() query_response.responseCode = ResponseCode.OK query_response.result = Result() details = JobUpdateDetails( update=JobUpdate(summary=JobUpdateSummary( key=UPDATE_KEY, user="******", state=JobUpdateState(status=JobUpdateStatus.ROLLING_FORWARD, createdTimestampMs=1000, lastModifiedTimestampMs=2000), metadata={ Metadata("issue", "test"), Metadata("country", "America"), Metadata("country", "Canada") })), updateEvents=[ JobUpdateEvent(status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=3000), JobUpdateEvent(status=JobUpdateStatus.ROLL_FORWARD_PAUSED, message="Investigating issues", timestampMs=4000), JobUpdateEvent(status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=5000) ], instanceEvents=[ JobInstanceUpdateEvent( instanceId=1, timestampMs=6000, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent( instanceId=2, timestampMs=7000, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent( instanceId=1, timestampMs=8000, action=JobUpdateAction.INSTANCE_UPDATED), JobInstanceUpdateEvent(instanceId=2, timestampMs=9000, action=JobUpdateAction.INSTANCE_UPDATED) ]) query_response.result.getJobUpdateDetailsResult = GetJobUpdateDetailsResult( detailsList=[details]) return query_response
def get_update_details_response(self): query_response = Response() query_response.responseCode = ResponseCode.OK query_response.result = Result() details = JobUpdateDetails( update=JobUpdate( summary=JobUpdateSummary( jobKey=self.TEST_JOBKEY.to_thrift(), updateId="0", user="******", state=JobUpdateState( status=JobUpdateStatus.ROLLING_FORWARD, createdTimestampMs=1411404927, lastModifiedTimestampMs=14114056030))), updateEvents=[ JobUpdateEvent( status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=1411404927), JobUpdateEvent( status=JobUpdateStatus.ROLL_FORWARD_PAUSED, timestampMs=1411405000), JobUpdateEvent( status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=1411405100)], instanceEvents=[ JobInstanceUpdateEvent( instanceId=1, timestampMs=1411404930, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent( instanceId=2, timestampMs=1411404940, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent( instanceId=1, timestampMs=1411404950, action=JobUpdateAction.INSTANCE_UPDATED), JobInstanceUpdateEvent( instanceId=2, timestampMs=1411404960, action=JobUpdateAction.INSTANCE_UPDATED)]) query_response.result.getJobUpdateDetailsResult = GetJobUpdateDetailsResult(details=details) return query_response
def get_update_details_response(cls): query_response = Response() query_response.responseCode = ResponseCode.OK query_response.result = Result() details = JobUpdateDetails() query_response.result.getJobUpdateDetailsResult = GetJobUpdateDetailsResult( details=details) details.update = JobUpdate() details.update.summary = JobUpdateSummary( jobKey=AuroraJobKey('west', 'mcc', 'test', 'hello'), updateId="fake-update-identifier", user="******", state=JobUpdateState(status=JobUpdateStatus.ROLLING_FORWARD, createdTimestampMs=1411404927, lastModifiedTimestampMs=14114056030)) details.updateEvents = [ JobUpdateEvent(status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=1411404927), JobUpdateEvent(status=JobUpdateStatus.ROLL_FORWARD_PAUSED, timestampMs=1411405000), JobUpdateEvent(status=JobUpdateStatus.ROLLING_FORWARD, timestampMs=1411405100) ] details.instanceEvents = [ JobInstanceUpdateEvent(instanceId=1, timestampMs=1411404930, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent(instanceId=2, timestampMs=1411404940, action=JobUpdateAction.INSTANCE_UPDATING), JobInstanceUpdateEvent(instanceId=1, timestampMs=1411404950, action=JobUpdateAction.INSTANCE_UPDATED), JobInstanceUpdateEvent(instanceId=2, timestampMs=1411404960, action=JobUpdateAction.INSTANCE_UPDATED) ] return query_response
def create_start_job_update_result(cls, code, msg, key, metadata): resp = cls.create_blank_response(code, msg) resp.result = Result(startJobUpdateResult=StartJobUpdateResult( key=key, updateSummary=JobUpdateSummary(metadata=metadata))) return resp