class TestBuildNotifications(TrialTestCase):
layer = LaunchpadZopelessLayer
def setUp(self):
super(TestBuildNotifications, self).setUp()
from lp.testing.factory import LaunchpadObjectFactory
self.factory = LaunchpadObjectFactory()
def prepareBehaviour(self, fake_successful_upload=False):
self.queue_record = (
self.factory.makeSourcePackageRecipeBuild().queueBuild())
build = self.queue_record.specific_build
if fake_successful_upload:
removeSecurityProxy(build).verifySuccessfulUpload = FakeMethod(
result=True)
# We overwrite the buildmaster root to use a temp directory.
tempdir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, tempdir)
self.upload_root = tempdir
tmp_builddmaster_root = """
[builddmaster]
root: %s
""" % self.upload_root
config.push('tmp_builddmaster_root', tmp_builddmaster_root)
self.addCleanup(config.pop, 'tmp_builddmaster_root')
self.queue_record.builder = self.factory.makeBuilder()
slave = WaitingSlave('BuildStatus.OK')
return BuilderInteractor.getBuildBehaviour(
self.queue_record, self.queue_record.builder, slave)
def assertDeferredNotifyCount(self, status, behaviour, expected_count):
d = behaviour.handleStatus(self.queue_record, status, {'filemap': {}})
def cb(result):
self.assertEqual(expected_count, len(pop_notifications()))
d.addCallback(cb)
return d
def test_handleStatus_PACKAGEFAIL(self):
"""Failing to build the package immediately sends a notification."""
return self.assertDeferredNotifyCount(
"PACKAGEFAIL", self.prepareBehaviour(), 1)
def test_handleStatus_OK(self):
"""Building the source package does _not_ immediately send mail.
(The archive uploader mail send one later.
"""
return self.assertDeferredNotifyCount(
"OK", self.prepareBehaviour(), 0)
def test_handleStatus_OK_successful_upload(self):
return self.assertDeferredNotifyCount(
"OK", self.prepareBehaviour(True), 0)
