def testCreateAndSave(self): gitiles_project = 'chromium/src' bucket = 'ci' builder_name = 'b' step_name = 's' test_name = 't' git_hash = 'a1b2c3' legacy_master_name = 'm' build_number = 123 build_url = 'build_url' pass_rate = 0.5 task_ids = ['task'] commit_position = 1000 try_job_url = 'try_job_url' iterations = 100 elapsed_seconds = 300 error = None commit_timestamp = datetime(2018, 12, 4) failed_swarming_task_attempts = 0 data_point = DataPoint.CreateAndSave( gitiles_project, bucket, builder_name, step_name, test_name, git_hash, legacy_master_name=legacy_master_name, build_number=build_number, build_url=build_url, pass_rate=pass_rate, task_ids=task_ids, commit_position=commit_position, try_job_url=try_job_url, iterations=iterations, elapsed_seconds=elapsed_seconds, error=error, commit_timestamp=commit_timestamp, failed_swarming_task_attempts=failed_swarming_task_attempts) self.assertEqual( data_point, DataPoint.Get(gitiles_project, bucket, builder_name, step_name, test_name, git_hash)) self.assertEqual(legacy_master_name, data_point.legacy_master_name) self.assertEqual(build_number, data_point.build_number) self.assertEqual(build_url, data_point.build_url) self.assertEqual(pass_rate, data_point.pass_rate) self.assertEqual(task_ids, data_point.task_ids) self.assertEqual(commit_position, data_point.commit_position) self.assertEqual(try_job_url, data_point.try_job_url) self.assertEqual(iterations, data_point.iterations) self.assertEqual(elapsed_seconds, data_point.elapsed_seconds) self.assertEqual(error, data_point.error) self.assertEqual(commit_timestamp, data_point.commit_timestamp) self.assertEqual(failed_swarming_task_attempts, data_point.failed_swarming_task_attempts)
def testGet(self): gitiles_project = 'chromium/src' bucket = 'ci' builder_name = 'b' step_name = 's' test_name = 't' git_hash = 'a1b2c3' data_point = DataPoint( key=DataPoint._CreateKey(gitiles_project, bucket, builder_name, step_name, test_name, git_hash)) data_point.put() self.assertIsNone( DataPoint.Get(gitiles_project, bucket, builder_name, step_name, test_name, 'wrong_hash')) self.assertEqual( data_point, DataPoint.Get(gitiles_project, bucket, builder_name, step_name, test_name, git_hash))