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
|
# 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 mock
from twisted.internet import defer
from twisted.trial import unittest
from buildbot.test.fake.fakemaster import make_master
class ChangeSourceMixin(object):
"""
This class is used for testing change sources, and handles a few things:
- starting and stopping a ChangeSource service
- a fake C{self.master.addChange}, which adds its args
to the list C{self.changes_added}
"""
changesource = None
started = False
def setUpChangeSource(self):
"Set up the mixin - returns a deferred."
self.changes_added = []
def addChange(**kwargs):
# check for 8-bit strings
for k,v in kwargs.items():
if type(v) == type(""):
try:
v.decode('ascii')
except UnicodeDecodeError:
raise unittest.FailTest(
"non-ascii string for key '%s': %r" % (k,v))
self.changes_added.append(kwargs)
return defer.succeed(mock.Mock())
self.master = make_master(testcase=self, wantDb=True)
self.master.addChange = addChange
return defer.succeed(None)
def tearDownChangeSource(self):
"Tear down the mixin - returns a deferred."
if not self.started:
return defer.succeed(None)
if self.changesource.running:
return defer.maybeDeferred(self.changesource.stopService)
return defer.succeed(None)
def attachChangeSource(self, cs):
"Set up a change source for testing; sets its .master attribute"
self.changesource = cs
self.changesource.master = self.master
def startChangeSource(self):
"start the change source as a service"
self.started = True
self.changesource.startService()
def stopChangeSource(self):
"stop the change source again; returns a deferred"
d = self.changesource.stopService()
def mark_stopped(_):
self.started = False
d.addCallback(mark_stopped)
return d
|