class JiraExtensionTestCase(TestCase):
def setUp(self):
self.jira = JiraExtension(mock_bot(), load=False)
settings.JIRA_URL = 'http://example.com/%(ticket)s'
def test_get_ticket_re_public_matches(self):
ret = self.jira.get_ticket_re('foo', 'helga jira foo bar', True)
assert ret == 'bar'
def test_get_ticket_re_public_no_match(self):
ret = self.jira.get_ticket_re('baz', 'helga jira foo bar', True)
assert ret is None
def test_get_ticket_re_private_matches_with_nick(self):
ret = self.jira.get_ticket_re('foo', 'helga jira foo bar', False)
assert ret == 'bar'
def test_get_ticket_re_private_matches_no_nick(self):
ret = self.jira.get_ticket_re('foo', 'jira foo bar', False)
assert ret == 'bar'
def test_get_ticket_re_private_no_match_with_nick(self):
ret = self.jira.get_ticket_re('baz', 'helga jira foo bar', False)
assert ret is None
def test_get_ticket_re_private_no_match_no_nick(self):
ret = self.jira.get_ticket_re('baz', 'jira foo bar', False)
assert ret is None
def patch_get_ticket_re(self, return_val):
self.jira.get_ticket_re = Mock()
self.jira.get_ticket_re.return_value = return_val
@patch('helga.extensions.jira.db')
def test_add_ticket_re_inserts_new_record(self, db):
self.patch_get_ticket_re('foo')
db.jira.find.return_value = db
db.count.return_value = 0
self.jira.add_ticket_re('foo', True)
assert 'foo' in self.jira.jira_pats
assert db.jira.insert.called
@patch('helga.extensions.jira.db')
def test_add_ticket_re_has_existing_record_in_db(self, db):
self.patch_get_ticket_re('foo')
db.jira.find.return_value = db
db.count.return_value = 1
self.jira.add_ticket_re('foo', True)
assert 'foo' in self.jira.jira_pats
assert not db.jira.insert.called
@patch('helga.extensions.jira.db')
def test_add_ticket_re_returns_none(self, db):
self.patch_get_ticket_re(None)
assert self.jira.add_ticket_re('foo', True) is None
def test_add_ticket_re_does_nothing_important(self):
self.patch_get_ticket_re('foo')
self.jira.jira_pats = ('foo',)
assert self.jira.add_ticket_re('foo', True)
@patch('helga.extensions.jira.db')
def test_remove_ticket_re_does_removing(self, db):
self.patch_get_ticket_re('foo')
self.jira.remove_ticket_re('foo', True)
assert db.jira.remove.called
def test_remove_ticket_re_does_nothing(self):
self.patch_get_ticket_re(None)
assert self.jira.remove_ticket_re('foo', True) is None
@patch('helga.extensions.jira.db')
def test_remove_ticket_re_removes_ticket(self, db):
self.patch_get_ticket_re('foo')
self.jira.jira_pats = set(['foo'])
self.jira.remove_ticket_re('foo', True)
assert db.jira.remove.called
assert 'foo' not in self.jira.jira_pats
def test_contextualize_no_patterns(self):
assert self.jira.contextualize('foo') is None
def test_contextualize_no_pattern_match(self):
self.jira.jira_pats = ('foobar',)
assert self.jira.contextualize('barfoo-123') is None
def test_contextualize_responds_with_url(self):
self.jira.jira_pats = ('foobar',)
ret = self.jira.contextualize('my message is foobar-123')
assert 'http://example.com/foobar-123' in ret
def test_contextualize_responds_many_urls(self):
self.jira.jira_pats = ('foobar',)
ret = self.jira.contextualize('look at foobar-123 and foobar-42')
assert 'http://example.com/foobar-123' in ret
assert 'http://example.com/foobar-42' in ret
def test_contextualize_responds_many_url_patterns(self):
self.jira.jira_pats = ('foobar', 'bazqux')
ret = self.jira.contextualize('look at foobar-123 and bazqux-10')
assert 'http://example.com/foobar-123' in ret
assert 'http://example.com/bazqux-10' in ret
def patch_for_dispatch(self, add_ret, rem_ret, ctx_ret):
self.jira.add_ticket_re = Mock(return_value=add_ret)
self.jira.remove_ticket_re = Mock(return_value=rem_ret)
self.jira.contextualize = Mock(return_value=ctx_ret)
def test_dispatch_responds_to_add(self):
self.patch_for_dispatch('foo', None, None)
ret = self.jira.dispatch('sducnan', '#all', 'check this', False)
assert ret == 'foo'
assert self.jira.add_ticket_re.called
assert not self.jira.remove_ticket_re.called
assert not self.jira.contextualize.called
def test_dispatch_responds_to_remove(self):
self.patch_for_dispatch(None, 'foo', None)
ret = self.jira.dispatch('sducnan', '#all', 'check this', False)
assert ret == 'foo'
assert self.jira.add_ticket_re.called
assert self.jira.remove_ticket_re.called
assert not self.jira.contextualize.called
def test_dispatch_responds_to_contextualize(self):
self.patch_for_dispatch(None, None, 'foo')
ret = self.jira.dispatch('sducnan', '#all', 'check this', False)
assert ret == 'foo'
assert self.jira.add_ticket_re.called
assert self.jira.remove_ticket_re.called
assert self.jira.contextualize.called