def test_buildMessage_addLogs(self):
mn = MailNotifier('*****@*****.**', mode=("change",), addLogs=True)
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body":"body", "type":"text",
"subject":"subject"}
mn.createEmail = Mock("createEmail")
mn._gotRecipients = Mock("_gotReceipients")
mn._gotRecipients.return_value = None
mn.master_status = Mock()
mn.master_status.getTitle.return_value = 'TITLE'
bldr = Mock(name="builder")
builds = [ FakeBuildStatus(name='build1'),
FakeBuildStatus(name='build2') ]
logs = [ FakeLog('log1'), FakeLog('log2') ]
for b, l in zip(builds, logs):
b.builder = bldr
b.results = 0
b.getSourceStamp.return_value = ss = Mock(name='ss')
ss.patch = None
ss.changes = []
b.getLogs.return_value = [ l ]
d = mn.buildMessage("mybldr", builds, 0)
def check(_):
mn.createEmail.assert_called_with(
dict(body='body\n\nbody\n\n', type='text', subject='subject'),
'mybldr', 'TITLE', 0, builds, [], logs)
d.addCallback(check)
return d
def test_buildMessage_addLogs(self):
mn = MailNotifier("*****@*****.**", mode=("change",), addLogs=True)
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text", "subject": "subject"}
mn.createEmail = Mock("createEmail")
mn._gotRecipients = Mock("_gotReceipients")
mn._gotRecipients.return_value = None
mn.master_status = Mock()
mn.master_status.getTitle.return_value = "TITLE"
bldr = Mock(name="builder")
builds = [FakeBuildStatus(name="build1"), FakeBuildStatus(name="build2")]
logs = [FakeLog("log1"), FakeLog("log2")]
for b, l in zip(builds, logs):
b.builder = bldr
b.results = 0
ss = Mock(name="ss")
b.getSourceStamps.return_value = [ss]
ss.patch = None
ss.changes = []
b.getLogs.return_value = [l]
d = mn.buildMessage("mybldr", builds, 0)
def check(_):
mn.createEmail.assert_called_with(
dict(body="body\n\nbody\n\n", type="text", subject="subject"), "mybldr", "TITLE", 0, builds, [], logs
)
d.addCallback(check)
return d
def test_buildMessage_addLogs(self):
mn = MailNotifier('*****@*****.**', mode="change", addLogs=True)
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body":"body", "type":"text",
"subject":"subject"}
mn.createEmail = Mock("createEmail")
mn._gotRecipients = Mock("_gotReceipients")
mn._gotRecipients.return_value = None
mn.master_status = Mock()
mn.master_status.getTitle.return_value = 'TITLE'
bldr = Mock(name="builder")
builds = [ FakeBuildStatus(name='build1'),
FakeBuildStatus(name='build2') ]
logs = [ FakeLog('log1'), FakeLog('log2') ]
for b, l in zip(builds, logs):
b.builder = bldr
b.results = 0
b.getSourceStamp.return_value = ss = Mock(name='ss')
ss.patch = None
ss.changes = []
b.getLogs.return_value = [ l ]
d = mn.buildMessage("mybldr", builds, 0)
def check(_):
mn.createEmail.assert_called_with(
dict(body='body\n\nbody\n\n', type='text', subject='subject'),
'mybldr', 'TITLE', 0, builds, [], logs)
d.addCallback(check)
return d
def test_buildsetFinished_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "passing", "warnings"),
builders=["Builder1", "Builder2"])
mn.buildMessage = fakeBuildMessage
builder1 = Mock()
builder1.getBuild = lambda number: build1
builder1.name = "Builder1"
build1 = FakeBuildStatus()
build1.results = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.getBuilder.return_value = builder1
builder2 = Mock()
builder2.getBuild = lambda number: build2
builder2.name = "Builder2"
build2 = FakeBuildStatus()
build2.results = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.getBuilder.return_value = builder1
def fakeGetBuilder(buildername):
return {"Builder1": builder1, "Builder2": builder2}[buildername]
self.db = fakedb.FakeDBConnector(self)
self.db.insertTestData([
fakedb.SourceStampSet(id=127),
fakedb.Buildset(id=99,
sourcestampsetid=127,
results=SUCCESS,
reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99, buildername='Builder1'),
fakedb.Build(number=0, brid=11),
fakedb.BuildRequest(id=12, buildsetid=99, buildername='Builder2'),
fakedb.Build(number=0, brid=12),
])
mn.master = self # FIXME: Should be FakeMaster
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {
"body": "body",
"type": "text",
"subject": "subject"
}
mn.buildsetFinished(99, FAILURE)
fakeBuildMessage.assert_called_with("(whole buildset)",
[build1, build2], SUCCESS)
def test_buildsetComplete_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "passing", "warnings"),
builders=["Builder1", "Builder2"])
mn.buildMessage = fakeBuildMessage
builder1 = Mock()
builder1.getBuild = lambda number: build1
builder1.name = "Builder1"
build1 = FakeBuildStatus()
build1.results = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.getBuilder.return_value = builder1
build1.getResults.return_value = build1.results
builder2 = Mock()
builder2.getBuild = lambda number: build2
builder2.name = "Builder2"
build2 = FakeBuildStatus()
build2.results = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.getBuilder.return_value = builder1
build2.getResults.return_value = build2.results
def fakeGetBuilder(buildername):
return {"Builder1": builder1, "Builder2": builder2}[buildername]
self.db = fakedb.FakeDBConnector(self)
self.db.insertTestData([fakedb.SourceStampSet(id=127),
fakedb.Buildset(id=99, sourcestampsetid=127,
results=SUCCESS,
reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99,
buildername='Builder1'),
fakedb.Build(number=0, brid=11),
fakedb.BuildRequest(id=12, buildsetid=99,
buildername='Builder2'),
fakedb.Build(number=0, brid=12),
])
mn.master = self # FIXME: Should be FakeMaster
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text",
"subject": "subject"}
mn._buildsetComplete(99, FAILURE)
fakeBuildMessage.assert_called_with("(whole buildset)",
[build1, build2], SUCCESS)
def test_buildsetFinished_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "passing", "warnings"),
builders=["Builder"])
mn.buildMessage = fakeBuildMessage
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus()
build.results = FAILURE
build.finished = True
build.reason = "testReason"
build.getBuilder.return_value = builder
self.db = fakedb.FakeDBConnector(self)
self.db.insertTestData([
fakedb.SourceStampSet(id=127),
fakedb.Buildset(id=99,
sourcestampsetid=127,
results=SUCCESS,
reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99, buildername='Builder'),
fakedb.Build(number=0, brid=11),
])
mn.master = self # FIXME: Should be FakeMaster
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {
"body": "body",
"type": "text",
"subject": "subject"
}
mn.buildsetFinished(99, FAILURE)
fakeBuildMessage.assert_called_with("Buildset Complete: testReason",
[build], SUCCESS)
def test_buildsetComplete_doesnt_send_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "warnings"),
builders=["Builder"])
mn.buildMessage = fakeBuildMessage
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus()
build.results = SUCCESS
build.finished = True
build.reason = "testReason"
build.getBuilder.return_value = builder
build.getResults.return_value = build.results
self.db = self.master.db
self.db.insertTestData([
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name='sl'),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99, buildername='Builder'),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13,
masterid=92),
])
mn.master = self.master
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text",
"subject": "subject"}
d = mn._buildsetComplete('buildset.99.complete',
dict(bsid=99, result=FAILURE))
@d.addCallback
def check(_):
self.assertFalse(fakeBuildMessage.called)
return d
def test_buildsetFinished_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode="all",
builders=["Builder"])
mn.buildMessage = fakeBuildMessage
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername":"Builder", "brid":1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = Mock()
build.result = FAILURE
build.finished = True
build.reason = "testReason"
build.builder = builder
build.getBuilder = Mock()
build.getBuilder.return_value = builder
self.db = fakedb.FakeDBConnector(self)
self.db.insertTestData([fakedb.Buildset(id=99, sourcestampid=127,
results=SUCCESS,
reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99,
buildername='Builder'),
fakedb.Build(number=0, brid=11, results=SUCCESS)
])
mn.parent = self
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body":"body", "type":"text",
"subject":"subject"}
mn.buildsetFinished(99, FAILURE)
fakeBuildMessage.assert_called_with("Buildset Complete: testReason",
[build], SUCCESS)
def test_buildsetComplete_doesnt_send_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "warnings"),
builders=["Builder"])
mn.buildMessage = fakeBuildMessage
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus()
build.results = SUCCESS
build.finished = True
build.reason = "testReason"
build.getBuilder.return_value = builder
build.getResults.return_value = build.results
self.db = self.master.db
self.db.insertTestData([
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name='sl'),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.Builder(id=80, name='Builder'),
fakedb.BuildRequest(id=11, buildsetid=99, builderid=80),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13,
masterid=92),
])
self.setupMailNotifier(mn)
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text",
"subject": "subject"}
d = mn._buildsetComplete('buildset.99.complete',
dict(bsid=99, result=FAILURE))
@d.addCallback
def check(_):
self.assertFalse(fakeBuildMessage.called)
return d
def test_buildsetFinished_doesnt_send_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "warnings"),
builders=["Builder"])
mn.buildMessage = fakeBuildMessage
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername":"Builder", "brid":1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus()
build.results = SUCCESS
build.finished = True
build.reason = "testReason"
build.getBuilder.return_value = builder
self.db = fakedb.FakeDBConnector(self)
self.db.insertTestData([fakedb.SourceStampSet(id=127),
fakedb.Buildset(id=99, sourcestampsetid=127,
results=SUCCESS,
reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99,
buildername='Builder'),
fakedb.Build(number=0, brid=11),
])
mn.master = self
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body":"body", "type":"text",
"subject":"subject"}
mn.buildsetFinished(99, FAILURE)
self.assertFalse(fakeBuildMessage.called)
def test_buildsetComplete_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier(
"*****@*****.**",
buildSetSummary=True,
mode=("failing", "passing", "warnings"),
builders=["Builder1", "Builder2"],
)
mn.buildMessage = fakeBuildMessage
builder1 = Mock()
builder1.getBuild = lambda number: build1
builder1.name = "Builder1"
build1 = FakeBuildStatus()
build1.results = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.getBuilder.return_value = builder1
build1.getResults.return_value = build1.results
builder2 = Mock()
builder2.getBuild = lambda number: build2
builder2.name = "Builder2"
build2 = FakeBuildStatus()
build2.results = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.getBuilder.return_value = builder1
build2.getResults.return_value = build2.results
def fakeGetBuilder(buildername):
return {"Builder1": builder1, "Builder2": builder2}[buildername]
self.db = self.master.db
self.db.insertTestData(
[
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name="sl"),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.Builder(id=80, name="Builder1"),
fakedb.Builder(id=81, name="Builder2"),
fakedb.BuildRequest(id=11, buildsetid=99, builderid=80),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13, masterid=92),
fakedb.BuildRequest(id=12, buildsetid=99, builderid=81),
fakedb.Build(number=0, buildrequestid=12, buildslaveid=13, masterid=92),
]
)
mn.master = self.master
self.status = Mock()
mn.master_status = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text", "subject": "subject"}
d = mn._buildsetComplete("buildset.99.complete", dict(bsid=99, result=FAILURE))
@d.addCallback
def check(_):
fakeBuildMessage.assert_called_with("(whole buildset)", [build1, build2], SUCCESS)
return d
def test_sendToInterestedUsers_two_builds(self):
from email.message import Message
m = Message()
mn = MailNotifier(fromaddr="*****@*****.**", lookup=None)
mn.sendMessage = Mock()
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.name = "Builder"
build1 = FakeBuildStatus(name="build")
build1.result = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.builder = builder
build2 = FakeBuildStatus(name="build")
build2.result = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.builder = builder
def fakeCreateEmail(msgdict, builderName, title, results, builds=None, patches=None, logs=None):
# only concerned with m['To'] and m['CC'], which are added in
# _got_recipients later
return defer.succeed(m)
mn.createEmail = fakeCreateEmail
self.db = self.master.db
self.db.insertTestData(
[
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name="sl"),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99, buildername="Builder"),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13, masterid=92),
fakedb.Build(number=1, buildrequestid=11, buildslaveid=13, masterid=92),
fakedb.Change(changeid=9123),
fakedb.Change(changeid=9124),
fakedb.ChangeUser(changeid=9123, uid=1),
fakedb.ChangeUser(changeid=9124, uid=2),
fakedb.User(uid=1, identifier="tdurden"),
fakedb.User(uid=2, identifier="user2"),
fakedb.UserInfo(uid=1, attr_type="email", attr_data="*****@*****.**"),
fakedb.UserInfo(uid=2, attr_type="email", attr_data="*****@*****.**"),
]
)
def _getInterestedUsers():
# 'narrator' in this case is the owner, which tests the lookup
return ["narrator"]
build1.getInterestedUsers = _getInterestedUsers
build2.getInterestedUsers = _getInterestedUsers
def _getResponsibleUsers():
return ["Big Bob <*****@*****.**>"]
build1.getResponsibleUsers = _getResponsibleUsers
build2.getResponsibleUsers = _getResponsibleUsers
# fake sourcestamp with relevant user bits
ss1 = Mock(name="sourcestamp")
fake_change1 = Mock(name="change")
fake_change1.number = 9123
ss1.changes = [fake_change1]
ss1.patch, ss1.addPatch = None, None
ss2 = Mock(name="sourcestamp")
fake_change2 = Mock(name="change")
fake_change2.number = 9124
ss2.changes = [fake_change2]
ss2.patch, ss1.addPatch = None, None
def fakeGetSSlist(ss):
return lambda: [ss]
build1.getSourceStamps = fakeGetSSlist(ss1)
build2.getSourceStamps = fakeGetSSlist(ss2)
self.setupMailNotifier(mn)
mn.buildMessageDict = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict.return_value = {"body": "body", "type": "text"}
mn.buildMessage(builder.name, [build1, build2], build1.result)
self.assertEqual(m["To"], "[email protected], [email protected]")
def do_test_sendToInterestedUsers(
self,
lookup=None,
extraRecipients=[],
sendToInterestedUsers=True,
exp_called_with=None,
exp_TO=None,
exp_CC=None,
):
from email.message import Message
m = Message()
mn = MailNotifier(
fromaddr="*****@*****.**",
lookup=lookup,
sendToInterestedUsers=sendToInterestedUsers,
extraRecipients=extraRecipients,
)
mn.sendMessage = Mock()
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus(name="build")
build.result = FAILURE
build.finished = True
build.reason = "testReason"
build.builder = builder
def fakeCreateEmail(msgdict, builderName, title, results, builds=None, patches=None, logs=None):
# only concerned with m['To'] and m['CC'], which are added in
# _got_recipients later
return defer.succeed(m)
mn.createEmail = fakeCreateEmail
self.db = self.master.db
self.db.insertTestData(
[
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name="sl"),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.Builder(id=80, name="Builder"),
fakedb.BuildRequest(id=11, buildsetid=99, builderid=80),
fakedb.Build(number=0, buildrequestid=11, masterid=92, buildslaveid=13),
fakedb.Change(changeid=9123),
fakedb.ChangeUser(changeid=9123, uid=1),
fakedb.User(uid=1, identifier="tdurden"),
fakedb.UserInfo(uid=1, attr_type="svn", attr_data="tdurden"),
fakedb.UserInfo(uid=1, attr_type="email", attr_data="*****@*****.**"),
]
)
# fake sourcestamp with relevant user bits
ss = Mock(name="sourcestamp")
fake_change = Mock(name="change")
fake_change.number = 9123
ss.changes = [fake_change]
ss.patch, ss.addPatch = None, None
def fakeGetSSlist():
return [ss]
build.getSourceStamps = fakeGetSSlist
def _getInterestedUsers():
# 'narrator' in this case is the owner, which tests the lookup
return ["narrator"]
build.getInterestedUsers = _getInterestedUsers
def _getResponsibleUsers():
return ["Big Bob <*****@*****.**>"]
build.getResponsibleUsers = _getResponsibleUsers
self.setupMailNotifier(mn)
mn.buildMessageDict = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict.return_value = {"body": "body", "type": "text"}
mn.buildMessage(builder.name, [build], build.result)
mn.sendMessage.assert_called_with(m, exp_called_with)
self.assertEqual(m["To"], exp_TO)
self.assertEqual(m["CC"], exp_CC)
def test_sendToInterestedUsers_two_builds(self):
from email.message import Message
m = Message()
mn = MailNotifier(fromaddr="*****@*****.**", lookup=None)
mn.sendMessage = Mock()
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.name = "Builder"
build1 = FakeBuildStatus(name="build")
build1.result = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.builder = builder
build2 = FakeBuildStatus(name="build")
build2.result = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.builder = builder
def fakeCreateEmail(msgdict, builderName, title, results, builds=None,
patches=None, logs=None):
# only concerned with m['To'] and m['CC'], which are added in
# _got_recipients later
return defer.succeed(m)
mn.createEmail = fakeCreateEmail
self.db = self.master.db
self.db.insertTestData([
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name='sl'),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.BuildRequest(id=11, buildsetid=99, buildername='Builder'),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13,
masterid=92),
fakedb.Build(number=1, buildrequestid=11, buildslaveid=13,
masterid=92),
fakedb.Change(changeid=9123),
fakedb.Change(changeid=9124),
fakedb.ChangeUser(changeid=9123, uid=1),
fakedb.ChangeUser(changeid=9124, uid=2),
fakedb.User(uid=1, identifier="tdurden"),
fakedb.User(uid=2, identifier="user2"),
fakedb.UserInfo(uid=1, attr_type='email',
attr_data="*****@*****.**"),
fakedb.UserInfo(uid=2, attr_type='email',
attr_data="*****@*****.**")
])
def _getInterestedUsers():
# 'narrator' in this case is the owner, which tests the lookup
return ["narrator"]
build1.getInterestedUsers = _getInterestedUsers
build2.getInterestedUsers = _getInterestedUsers
def _getResponsibleUsers():
return ["Big Bob <*****@*****.**>"]
build1.getResponsibleUsers = _getResponsibleUsers
build2.getResponsibleUsers = _getResponsibleUsers
# fake sourcestamp with relevant user bits
ss1 = Mock(name="sourcestamp")
fake_change1 = Mock(name="change")
fake_change1.number = 9123
ss1.changes = [fake_change1]
ss1.patch, ss1.addPatch = None, None
ss2 = Mock(name="sourcestamp")
fake_change2 = Mock(name="change")
fake_change2.number = 9124
ss2.changes = [fake_change2]
ss2.patch, ss1.addPatch = None, None
def fakeGetSSlist(ss):
return lambda: [ss]
build1.getSourceStamps = fakeGetSSlist(ss1)
build2.getSourceStamps = fakeGetSSlist(ss2)
self.setupMailNotifier(mn)
mn.buildMessageDict = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict.return_value = {"body": "body", "type": "text"}
mn.buildMessage(builder.name, [build1, build2], build1.result)
self.assertEqual(m['To'], "[email protected], [email protected]")
def do_test_sendToInterestedUsers(self, lookup=None, extraRecipients=[],
sendToInterestedUsers=True,
exp_called_with=None, exp_TO=None,
exp_CC=None):
from email.message import Message
m = Message()
mn = MailNotifier(fromaddr='*****@*****.**',
lookup=lookup,
sendToInterestedUsers=sendToInterestedUsers,
extraRecipients=extraRecipients)
mn.sendMessage = Mock()
def fakeGetBuild(number):
return build
def fakeGetBuilder(buildername):
if buildername == builder.name:
return builder
return None
def fakeGetBuildRequests(self, bsid):
return defer.succeed([{"buildername": "Builder", "brid": 1}])
builder = Mock()
builder.getBuild = fakeGetBuild
builder.name = "Builder"
build = FakeBuildStatus(name="build")
build.result = FAILURE
build.finished = True
build.reason = "testReason"
build.builder = builder
def fakeCreateEmail(msgdict, builderName, title, results, builds=None,
patches=None, logs=None):
# only concerned with m['To'] and m['CC'], which are added in
# _got_recipients later
return defer.succeed(m)
mn.createEmail = fakeCreateEmail
self.db = self.master.db
self.db.insertTestData([
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name='sl'),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.Builder(id=80, name='Builder'),
fakedb.BuildRequest(id=11, buildsetid=99, builderid=80),
fakedb.Build(number=0, buildrequestid=11, masterid=92,
buildslaveid=13),
fakedb.Change(changeid=9123),
fakedb.ChangeUser(changeid=9123, uid=1),
fakedb.User(uid=1, identifier="tdurden"),
fakedb.UserInfo(uid=1, attr_type='svn', attr_data="tdurden"),
fakedb.UserInfo(uid=1, attr_type='email',
attr_data="*****@*****.**")
])
# fake sourcestamp with relevant user bits
ss = Mock(name="sourcestamp")
fake_change = Mock(name="change")
fake_change.number = 9123
ss.changes = [fake_change]
ss.patch, ss.addPatch = None, None
def fakeGetSSlist():
return [ss]
build.getSourceStamps = fakeGetSSlist
def _getInterestedUsers():
# 'narrator' in this case is the owner, which tests the lookup
return ["narrator"]
build.getInterestedUsers = _getInterestedUsers
def _getResponsibleUsers():
return ["Big Bob <*****@*****.**>"]
build.getResponsibleUsers = _getResponsibleUsers
self.setupMailNotifier(mn)
mn.buildMessageDict = Mock()
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict.return_value = {"body": "body", "type": "text"}
mn.buildMessage(builder.name, [build], build.result)
mn.sendMessage.assert_called_with(m, exp_called_with)
self.assertEqual(m['To'], exp_TO)
self.assertEqual(m['CC'], exp_CC)
def test_buildsetComplete_sends_email(self):
fakeBuildMessage = Mock()
mn = MailNotifier('*****@*****.**',
buildSetSummary=True,
mode=("failing", "passing", "warnings"),
builders=["Builder1", "Builder2"])
mn.buildMessage = fakeBuildMessage
builder1 = Mock()
builder1.getBuild = lambda number: build1
builder1.name = "Builder1"
build1 = FakeBuildStatus()
build1.results = FAILURE
build1.finished = True
build1.reason = "testReason"
build1.getBuilder.return_value = builder1
build1.getResults.return_value = build1.results
builder2 = Mock()
builder2.getBuild = lambda number: build2
builder2.name = "Builder2"
build2 = FakeBuildStatus()
build2.results = FAILURE
build2.finished = True
build2.reason = "testReason"
build2.getBuilder.return_value = builder1
build2.getResults.return_value = build2.results
def fakeGetBuilder(buildername):
return {"Builder1": builder1, "Builder2": builder2}[buildername]
self.db = self.master.db
self.db.insertTestData([
fakedb.Master(id=92),
fakedb.Buildslave(id=13, name='sl'),
fakedb.Buildset(id=99, results=SUCCESS, reason="testReason"),
fakedb.Builder(id=80, name='Builder1'),
fakedb.Builder(id=81, name='Builder2'),
fakedb.BuildRequest(id=11, buildsetid=99, builderid=80),
fakedb.Build(number=0, buildrequestid=11, buildslaveid=13,
masterid=92),
fakedb.BuildRequest(id=12, buildsetid=99, builderid=81),
fakedb.Build(number=0, buildrequestid=12, buildslaveid=13,
masterid=92),
])
self.setupMailNotifier(mn)
mn.master_status.getBuilder = fakeGetBuilder
mn.buildMessageDict = Mock()
mn.buildMessageDict.return_value = {"body": "body", "type": "text",
"subject": "subject"}
d = mn._buildsetComplete('buildset.99.complete',
dict(bsid=99, result=FAILURE))
@d.addCallback
def check(_):
fakeBuildMessage.assert_called_with(
"(whole buildset)",
[build1, build2], SUCCESS)
return d
