Add some tests for buildbot#2278.

master/buildbot/test/fake/botmaster.py

@@ -0,0 +1,32 @@
+# 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
+
+from twisted.application import service
+
+class FakeBotMaster(service.MultiService):
+    def __init__(self, master):
+        service.MultiService.__init__(self)
+        self.setName("fake-botmaster")
+        self.master = master
+        self.locks = {}
+
+    def getLockByID(self, lockid):
+        if not lockid in self.locks:
+            self.locks[lockid] = lockid.lockClass(lockid)
+        # if the master.cfg file has changed maxCount= on the lock, the next
+        # time a build is started, they'll get a new RealLock instance. Note
+        # that this requires that MasterLock and SlaveLock (marker) instances
+        # be hashable and that they should compare properly.
+        return self.locks[lockid]

master/buildbot/test/unit/test_buildslave.py

@@ -16,10 +16,11 @@
 import mock
 from twisted.trial import unittest
 from twisted.internet import defer
-from buildbot import buildslave, config
+from buildbot import buildslave, config, locks
 from buildbot.test.fake import fakemaster
+from buildbot.test.fake.botmaster import FakeBotMaster
 
-class AbstractBuildSlave(unittest.TestCase):
+class TestAbstractBuildSlave(unittest.TestCase):
 
     class ConcreteBuildSlave(buildslave.AbstractBuildSlave):
         pass
@@ -203,3 +204,35 @@ def test_missing_timer(self):
         bs.stopMissingTimer()
         self.assertEqual(bs.missing_timer, None)
 
+    def test_setServiceParent_started(self):
+        master = self.master = fakemaster.make_master()
+        botmaster = FakeBotMaster(master)
+        botmaster.startService()
+        bs = self.ConcreteBuildSlave('bot', 'pass')
+        bs.setServiceParent(botmaster)
+        self.assertEqual(bs.botmaster, botmaster)
+        self.assertEqual(bs.master, master)
+
+    def test_setServiceParent_masterLocks(self):
+        """
+        http://trac.buildbot.net/ticket/2278
+        """
+        master = self.master = fakemaster.make_master()
+        botmaster = FakeBotMaster(master)
+        botmaster.startService()
+        lock = locks.MasterLock('masterlock')
+        bs = self.ConcreteBuildSlave('bot', 'pass', locks = [lock])
+        bs.setServiceParent(botmaster)
+
+    def test_setServiceParent_slaveLocks(self):
+        """
+        http://trac.buildbot.net/ticket/2278
+        """
+        master = self.master = fakemaster.make_master()
+        botmaster = FakeBotMaster(master)
+        botmaster.startService()
+        lock = locks.SlaveLock('lock')
+        bs = self.ConcreteBuildSlave('bot', 'pass', locks = [lock])
+        bs.setServiceParent(botmaster)
+
+    test_setServiceParent_slaveLocks.todo = "SlaveLock not support for slave lock"

master/buildbot/test/unit/test_process_botmaster_BotMaster.py

@@ -168,7 +168,6 @@ def test_reconfigServiceSlaves_add_remove(self):
         wfd.getResult()
 
         self.assertIdentical(sl.parent, self.botmaster)
-        self.assertIdentical(sl.master, self.master)
         self.assertEqual(self.botmaster.slaves, { 'sl1' : sl })
 
         self.new_config.slaves = [ ]