Rietveld: schedule builds on buildbucket

appengine/chromium_rietveld/tests/test_buildbucket.py

 #!/usr/bin/env python
 # Copyright 2011 Google Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
 """Tests for buildbucket client."""
 
 import json
 import logging
 import unittest
 
 import setup
 setup.process_args()
 
 
 from google.appengine.api import app_identity
+from google.appengine.api import users
 from google.appengine.ext import ndb
 
 from utils import TestCase
 
 from codereview import buildbucket
 from codereview import common
 from codereview import models
 from codereview import net
 
 
@@ skipping 132 matching lines @@
     self.requests = []
     def json_request_async(*args, **kwargs):
       self.requests.append((args, kwargs))
       future = ndb.Future()
       future.set_result(self.fake_responses.pop(0))
       return future
 
     self.mock(net, 'json_request_async', json_request_async)
     self.mock(common, 'get_preferred_domain',
               lambda *_, **__: 'codereview.chromium.org')
-
-  def test_get_try_job_results_for_patchset(self):
     models.Account.current_user_account = models.Account(
         email='johndoe@chromium.org')
 
+
+  def test_get_try_job_results_for_patchset(self):
     put_builds_response = {
       'builds': [
         {'id': '1', 'status': 'SCHEDULED'},
         {'id': '2', 'status': 'SCHEDULED'},
       ],
     }
     self.fake_responses = [
       {'delegation_token': 'deltok', 'validity_duration': 1000},
       put_builds_response,
     ]
     actual_builds = buildbucket.get_builds_for_patchset_async(
         'project', 1, 2).get_result()
     self.assertEqual(actual_builds, put_builds_response['builds'])
 
     mint_token_req_body = self.requests[0][1]['payload']
     self.assertEqual(mint_token_req_body, {
       'audience': ['user:test@localhost'],
       'services': ['service:cr-buildbucket-test'],
       'impersonate': 'user:johndoe@chromium.org',
     })
     put_builds_req_headers = self.requests[1][1]['headers']
     self.assertEqual(
         put_builds_req_headers['X-Delegation-Token-V1'],
         'deltok')
 
+  def test_schedule(self):
+    models.Account.current_user_account = models.Account(
+        email='johndoe@chromium.org')

[tandrii(chromium), 2015/09/16 11:02:32]

this is not necessary, you set it up in setUp() now.

[nodir, 2015/09/17 17:44:17]

Done.

+
+    put_builds_response = {
+      'results': [
+        {
+          'build': {'id': '1'},
+        },
+        {
+          'build': {'id': '1'},
+        },
+      ]
+    }
+    self.fake_responses = [
+      {'delegation_token': 'deltok', 'validity_duration': 1000},
+      put_builds_response,
+    ]
+    issue = models.Issue(
+        id='123',
+        project='chromium',
+        owner=users.User(email='owner@chromium.org'),
+    )
+    builds = buildbucket.schedule(issue, '1', [
+        ('tryserver.chromium.linux', 'linux_rel'),
+        ('tryserver.chromium.linux', 'win_rel'),
+    ])
+    self.assertEqual(
+        builds, [r['build'] for r in put_builds_response['results']])
+
 
 if __name__ == '__main__':
   unittest.main()