1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
# This file is part of Buildbot. Buildbot is free software: you can
# redistribute it and/or modify it under the terms of the GNU General Public
# License as published by the Free Software Foundation, version 2.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Copyright Buildbot Team Members
import os
from twisted.trial import unittest
from twisted.internet import defer
from buildbot.changes import hgpoller
from buildbot.test.util import changesource, gpo
from buildbot.test.fake.fakedb import FakeDBConnector
from buildbot.util import epoch2datetime
ENVIRON_2116_KEY = 'TEST_THAT_ENVIRONMENT_GETS_PASSED_TO_SUBPROCESSES'
class TestHgPoller(gpo.GetProcessOutputMixin,
changesource.ChangeSourceMixin,
unittest.TestCase):
def setUp(self):
# To test that environment variables get propagated to subprocesses
# (See #2116)
os.environ[ENVIRON_2116_KEY] = 'TRUE'
self.setUpGetProcessOutput()
d = self.setUpChangeSource()
self.remote_repo = 'ssh://example.com/foo/baz'
self.repo_ready = True
def _isRepositoryReady():
return self.repo_ready
def create_poller(_):
self.poller = hgpoller.HgPoller(self.remote_repo,
workdir='/some/dir')
self.poller.master = self.master
self.poller._isRepositoryReady = _isRepositoryReady
def create_db(_):
db = self.master.db = FakeDBConnector(self)
return db.setup()
d.addCallback(create_poller)
d.addCallback(create_db)
return d
def tearDown(self):
del os.environ[ENVIRON_2116_KEY]
return self.tearDownChangeSource()
def gpoFullcommandPattern(self, commandName, *expected_args):
"""Match if the command is commandName and arg list start as expected.
This allows to test a bit more if expected GPO are issued, be it
by obscure failures due to the result not being given.
"""
def matchesSubcommand(bin, given_args, **kwargs):
return bin == commandName and tuple(
given_args[:len(expected_args)]) == expected_args
return matchesSubcommand
def test_describe(self):
self.assertSubstring("HgPoller", self.poller.describe())
def test_hgbin_default(self):
self.assertEqual(self.poller.hgbin, "hg")
def test_poll_initial(self):
self.repo_ready = False
# Test that environment variables get propagated to subprocesses
# (See #2116)
expected_env = {ENVIRON_2116_KEY: 'TRUE'}
self.addGetProcessOutputExpectEnv(expected_env)
self.expectCommands(
gpo.Expect('hg', 'init', '/some/dir'),
gpo.Expect('hg', 'pull', '-b', 'default',
'ssh://example.com/foo/baz')
.path('/some/dir'),
gpo.Expect('hg', 'heads', 'default', '--template={rev}' + os.linesep)
.path('/some/dir').stdout("73591"),
gpo.Expect('hg', 'log', '-b', 'default', '-r', '73591:73591', # only fetches that head
'--template={rev}:{node}\\n')
.path('/some/dir').stdout(os.linesep.join(['73591:4423cdb'])),
gpo.Expect('hg', 'log', '-r', '4423cdb',
'--template={date|hgdate}' + os.linesep + '{author}' + os.linesep + '{files}' + os.linesep + '{desc|strip}')
.path('/some/dir').stdout(os.linesep.join([
'1273258100.0 -7200',
'Bob Test <bobtest@example.org>',
'file1 dir/file2',
'This is rev 73591',
''])),
)
# do the poll
d = self.poller.poll()
# check the results
def check_changes(_):
self.assertEqual(len(self.changes_added), 1)
change = self.changes_added[0]
self.assertEqual(change['revision'], '4423cdb')
self.assertEqual(change['author'],
'Bob Test <bobtest@example.org>')
self.assertEqual(change['when_timestamp'],
epoch2datetime(1273258100)),
self.assertEqual(change['files'], ['file1', 'dir/file2'])
self.assertEqual(change['src'], 'hg')
self.assertEqual(change['branch'], 'default')
self.assertEqual(change['comments'], 'This is rev 73591')
d.addCallback(check_changes)
d.addCallback(self.check_current_rev(73591))
return d
def check_current_rev(self, wished):
def check_on_rev(_):
d = self.poller._getCurrentRev()
d.addCallback(lambda oid_rev: self.assertEqual(oid_rev[1], wished))
return check_on_rev
@defer.inlineCallbacks
def test_poll_several_heads(self):
# If there are several heads on the named branch, the poller musn't
# climb (good enough for now, ideally it should even go to the common
# ancestor)
self.expectCommands(
gpo.Expect('hg', 'pull', '-b', 'default',
'ssh://example.com/foo/baz')
.path('/some/dir'),
gpo.Expect('hg', 'heads', 'default', '--template={rev}' + os.linesep)
.path('/some/dir').stdout('5' + os.linesep + '6' + os.linesep),
)
yield self.poller._setCurrentRev(3)
# do the poll: we must stay at rev 3
d = self.poller.poll()
d.addCallback(self.check_current_rev(3))
@defer.inlineCallbacks
def test_poll_regular(self):
# normal operation. There's a previous revision, we get a new one.
self.expectCommands(
gpo.Expect('hg', 'pull', '-b', 'default',
'ssh://example.com/foo/baz')
.path('/some/dir'),
gpo.Expect('hg', 'heads', 'default', '--template={rev}' + os.linesep)
.path('/some/dir').stdout('5' + os.linesep),
gpo.Expect('hg', 'log', '-b', 'default', '-r', '5:5',
'--template={rev}:{node}\\n')
.path('/some/dir').stdout('5:784bd' + os.linesep),
gpo.Expect('hg', 'log', '-r', '784bd',
'--template={date|hgdate}' + os.linesep + '{author}' + os.linesep + '{files}' + os.linesep + '{desc|strip}')
.path('/some/dir').stdout(os.linesep.join([
'1273258009.0 -7200',
'Joe Test <joetest@example.org>',
'file1 file2',
'Comment for rev 5',
''])),
)
yield self.poller._setCurrentRev(4)
d = self.poller.poll()
d.addCallback(self.check_current_rev(5))
def check_changes(_):
self.assertEquals(len(self.changes_added), 1)
change = self.changes_added[0]
self.assertEqual(change['revision'], '784bd')
self.assertEqual(change['comments'], 'Comment for rev 5')
d.addCallback(check_changes)
|
