Index: infra/recipe_modules/bot_update/test_api.py |
diff --git a/infra/recipe_modules/bot_update/test_api.py b/infra/recipe_modules/bot_update/test_api.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..723c6f2a166948e440ec981d4643465a465b725c |
--- /dev/null |
+++ b/infra/recipe_modules/bot_update/test_api.py |
@@ -0,0 +1,74 @@ |
+# Copyright 2014 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+import hashlib |
+import os |
+import struct |
+import sys |
+from recipe_engine import recipe_test_api |
+ |
+ |
+class BotUpdateTestApi(recipe_test_api.RecipeTestApi): |
+ def output_json(self, root, first_sln, revision_mapping, git_mode, |
+ force=False, fail_patch=False, output_manifest=False, |
+ fixed_revisions=None): |
+ """Deterministically synthesize json.output test data for gclient's |
+ --output-json option. |
+ """ |
+ active = True |
+ |
+ output = { |
+ 'did_run': active, |
+ 'patch_failure': False |
+ } |
+ |
+ # Add in extra json output if active. |
+ if active: |
+ properties = { |
+ property_name: self.gen_revision(project_name, git_mode) |
+ for project_name, property_name in revision_mapping.iteritems() |
+ } |
+ properties.update({ |
+ '%s_cp' % property_name: ('refs/heads/master@{#%s}' % |
+ self.gen_revision(project_name, False)) |
+ for project_name, property_name in revision_mapping.iteritems() |
+ }) |
+ |
+ output.update({ |
+ 'patch_root': root or first_sln, |
+ 'root': first_sln, |
+ 'properties': properties, |
+ 'step_text': 'Some step text' |
+ }) |
+ |
+ if output_manifest: |
+ output.update({ |
+ 'manifest': { |
+ project_name: { |
+ 'repository': 'https://fake.org/%s.git' % project_name, |
+ 'revision': self.gen_revision(project_name, git_mode), |
+ } |
+ for project_name in revision_mapping |
+ } |
+ }) |
+ |
+ if fixed_revisions: |
+ output['fixed_revisions'] = fixed_revisions |
+ |
+ if fail_patch: |
+ output['log_lines'] = [('patch error', 'Patch failed to apply'),] |
+ output['patch_failure'] = True |
+ output['patch_apply_return_code'] = 1 |
+ if fail_patch == 'download': |
+ output['patch_apply_return_code'] = 3 |
+ return self.m.json.output(output) |
+ |
+ @staticmethod |
+ def gen_revision(project, GIT_MODE): |
+ """Hash project to bogus deterministic revision values.""" |
+ h = hashlib.sha1(project) |
+ if GIT_MODE: |
+ return h.hexdigest() |
+ else: |
+ return struct.unpack('!I', h.digest()[:4])[0] % 300000 |