def test_get_attachment(payload): srv = Service({}) model = payload['model'].lower() method = 'get_%s_attachment' % model attachment = getattr(srv, method)(payload) assert type(attachment) == dict assert attachment.has_key('text')
def test_format_item_link(): srv = Service({}) item = {'short_url': '', 'title': '', 'number': 1} link = srv.format_item_link(item) assert type(link) == str assert link[0] == '<' assert link[-1] == '>'
def test_get_post_data(payload): model = payload['model'].lower() srv = Service({}) with patch.object(srv, 'get_%s_attachment' % model) as mock_get_attachment: mock_get_attachment.return_value = {} post_data = srv.get_post_data(payload) assert post_data.has_key('attachments') assert post_data.has_key('icon_url') mock_get_attachment.assert_called_with(payload)
def test_get_attachment_color(): srv = Service({}) assert type(srv.get_attachment_color({'type': 'story'})) == str
def test_get_attachment_color_empty(): srv = Service({}) assert type(srv.get_attachment_color()) == str