Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(40)

Side by Side Diff: scripts/master/buildbucket/unittests/changestore_test.py

Issue 2157703002: buildbucket: support unique change urls (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/tools/build
Patch Set: nits Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 #!/usr/bin/env python 1 #!/usr/bin/env python
2 # Copyright 2015 The Chromium Authors. All rights reserved. 2 # Copyright 2015 The Chromium Authors. All rights reserved.
3 # Use of this source code is governed by a BSD-style license that can be 3 # Use of this source code is governed by a BSD-style license that can be
4 # found in the LICENSE file. 4 # found in the LICENSE file.
5 5
6 """changes.ChangeStore unit tests""" 6 """changes.ChangeStore unit tests"""
7 7
8 import datetime 8 import datetime
9 import json 9 import json
10 import unittest 10 import unittest
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after
63 change = self.store._find_change(m.revision, m.change_id) 63 change = self.store._find_change(m.revision, m.change_id)
64 cache = self.buildbot.change_cache 64 cache = self.buildbot.change_cache
65 cache.get.assert_called_once_with((m.revision, m.change_id)) 65 cache.get.assert_called_once_with((m.revision, m.change_id))
66 self.assertEqual(change, cache.get.return_value) 66 self.assertEqual(change, cache.get.return_value)
67 67
68 def test_get_change(self): 68 def test_get_change(self):
69 cache = self.buildbot.change_cache 69 cache = self.buildbot.change_cache
70 cache.get.return_value = None 70 cache.get.return_value = None
71 result = run_deferred(self.store.get_change(self.buildbucket_change)) 71 result = run_deferred(self.store.get_change(self.buildbucket_change))
72 72
73 info = json.dumps({ 73 info = {
74 common.BUILDBUCKET_CHANGE_ID_PROPERTY: '1', 74 common.BUILDBUCKET_CHANGE_ID_PROPERTY: '1',
75 }, sort_keys=True) 75 }
76 self.buildbot.add_change_to_db.assert_called_once_with( 76 self.buildbot.add_change_to_db.assert_called_once_with(
77 author=self.buildbucket_change['author']['email'], 77 author=self.buildbucket_change['author']['email'],
78 files=[], 78 files=[],
79 comments=self.buildbucket_change['message'], 79 comments=self.buildbucket_change['message'],
80 revision=self.buildbucket_change['revision'], 80 revision=self.buildbucket_change['revision'],
81 when_timestamp=datetime.datetime(2014, 12, 22), 81 when_timestamp=datetime.datetime(2014, 12, 22),
82 branch=self.buildbucket_change['branch'], 82 branch=self.buildbucket_change['branch'],
83 category=common.CHANGE_CATEGORY, 83 category=common.CHANGE_CATEGORY,
84 revlink=self.buildbucket_change['url'], 84 revlink=self.buildbucket_change['url'],
85 properties={ 85 properties={
86 common.INFO_PROPERTY: (info, 'Change'), 86 common.INFO_PROPERTY: (info, 'Change'),
87 }, 87 },
88 repository=self.buildbucket_change.get('repo_url'), 88 repository=self.buildbucket_change.get('repo_url'),
89 project=self.buildbucket_change.get('project'), 89 project=self.buildbucket_change.get('project'),
90 ) 90 )
91 self.assertEqual(result, self.buildbot.get_change_by_id.return_value) 91 self.assertEqual(result, self.buildbot.get_change_by_id.return_value)
92 92
93 def test_get_change_without_id(self): 93 def test_get_change_without_id(self):
94 cache = self.buildbot.change_cache 94 cache = self.buildbot.change_cache
95 cache.get.return_value = None 95 cache.get.return_value = None
96 result = run_deferred(self.store.get_change({ 96 result = run_deferred(self.store.get_change({
97 'author': { 97 'author': {
98 'email': 'johndoe@chromium.org', 98 'email': 'johndoe@chromium.org',
99 }, 99 },
100 'message': 'Hello world', 100 'message': 'Hello world',
101 })) 101 }))
102 102
103 expected_info = json.dumps({'change_id': None}) 103 expected_info = {'change_id': None}
104 self.buildbot.add_change_to_db.assert_called_once_with( 104 self.buildbot.add_change_to_db.assert_called_once_with(
105 author='johndoe@chromium.org', 105 author='johndoe@chromium.org',
106 files=[], 106 files=[],
107 comments='Hello world', 107 comments='Hello world',
108 revision='', 108 revision='',
109 when_timestamp=None, 109 when_timestamp=None,
110 branch=None, 110 branch=None,
111 category=common.CHANGE_CATEGORY, 111 category=common.CHANGE_CATEGORY,
112 revlink='', 112 revlink='',
113 properties={ 113 properties={
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after
148 } 148 }
149 result = run_deferred( 149 result = run_deferred(
150 self.store.get_source_stamp([self.buildbucket_change], cache=ss_cache)) 150 self.store.get_source_stamp([self.buildbucket_change], cache=ss_cache))
151 self.assertFalse(self.buildbot.change_cache.get.called) 151 self.assertFalse(self.buildbot.change_cache.get.called)
152 self.assertFalse(self.buildbot.insert_source_stamp_to_db.called) 152 self.assertFalse(self.buildbot.insert_source_stamp_to_db.called)
153 self.assertEqual(result, ssid) 153 self.assertEqual(result, ssid)
154 154
155 155
156 if __name__ == '__main__': 156 if __name__ == '__main__':
157 unittest.main() 157 unittest.main()
OLDNEW
« no previous file with comments | « scripts/master/buildbucket/integration.py ('k') | scripts/master/buildbucket/unittests/integration_test.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698