from __future__ import print_function, unicode_literals
import wee_slack
import pytest
@pytest.mark.parametrize('case', (
{
'input_message': {'attachments': [{
'title': 'Title',
}]},
'input_text_before': "Text before",
'output': "\n".join([
"",
"| Title",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'title_link': 'http://title.link',
'from_url': 'http://from.url',
'fallback': 'Fallback',
}]},
'input_text_before': "",
'output': "\n".join([
"| Title (http://title.link)",
"| http://from.url",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'title_link': 'http://title.link',
'image_url': 'http://image.url',
'fallback': 'Fallback',
}]},
'input_text_before': "",
'output': "\n".join([
"| Title (http://title.link)",
"| Attachment text",
"| http://image.url",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'title_link': 'http://link?a=1&b=2',
'from_url': 'http://link?a=1&b=2',
'image_url': 'http://link?a=1&b=2',
}]},
'input_text_before': "http://link?a=1&b=2",
'output': "\n".join([
"",
"| Title",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'title_link': 'http://link?a=1&b=2',
'from_url': 'http://link?a=1&b=2',
'image_url': 'http://link?a=1&b=2',
}]},
'input_text_before': "http://link?a=1&b=2",
'output': "\n".join([
"",
"| Title",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'title_link': 'http://link',
'from_url': 'http://link',
'image_url': 'http://link',
}]},
'input_text_before': "",
'output': "\n".join([
"| Title (http://link)",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text',
'from_url': 'http://link',
'image_url': 'http://link',
}]},
'input_text_before': "",
'output': "\n".join([
"| Title",
"| http://link",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'text': 'Attachment text\n\n\nWith multiple lines',
}]},
'input_text_before': "",
'output': "\n".join([
"| Title",
"| Attachment text",
"| With multiple lines",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'author_name': 'Author',
'pretext': 'Pretext',
'text': 'Attachment text',
'title_link': 'http://title.link',
'from_url': 'http://from.url',
}]},
'input_text_before': "",
'output': "\n".join([
"| Pretext",
"| Author: Title (http://title.link)",
"| http://from.url",
"| Attachment text",
]),
},
{
'input_message': {'attachments': [{
'author_name': 'Author',
'text': 'Attachment text',
'title_link': 'http://title.link',
'from_url': 'http://from.url',
}]},
'input_text_before': "",
'output': "\n".join([
"| http://from.url",
"| Author: Attachment text",
]),
},
{
'input_message': {'attachments': [{
'fallback': 'Fallback',
}]},
'input_text_before': "",
'output': "| Fallback",
},
{
'input_message': {'attachments': [{
'fallback': 'Fallback',
'title_link': 'http://link',
}]},
'input_text_before': "http://link",
'output': "",
},
{
'input_message': {'attachments': [{
'fallback': 'Fallback',
'from_url': 'http://link',
}]},
'input_text_before': "http://link",
'output': "",
},
{
'input_message': {'attachments': [{
'fallback': 'Fallback',
'image_url': 'http://link',
}]},
'input_text_before': "http://link",
'output': "",
},
{
'input_message': {'attachments': [{
'text': 'Some message',
'footer': 'Thread in #general'
}]},
'input_text_before': "",
'output': "\n".join([
"| Some message",
"| Thread in #general",
]),
},
{
'input_message': {'attachments': [{
'ts': 1584986782,
'text': 'Some message',
'footer': 'Thread in #general'
}]},
'input_text_before': "",
'output': "\n".join([
"| Some message",
"| Thread in #general | Mar 23, 2020",
]),
},
{
'input_message': {'attachments': [{
'ts': '1584986782.261400',
'text': 'Some message',
'footer': 'Thread in #general'
}]},
'input_text_before': "",
'output': "\n".join([
"| Some message",
"| Thread in #general | Mar 23, 2020",
]),
},
{
'input_message': {'attachments': [{
'text': 'Original message',
'files': [
{
'title': 'File',
'url_private': 'http://link',
}
],
}]},
'input_text_before': "",
'output': "\n".join([
"| Original message",
"| http://link (File)",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'fields': [{
'title': 'First field title',
'value': 'First field value',
}, {
'title': '',
'value': 'Second field value',
}],
}]},
'input_text_before': "",
'output': "\n".join([
"| Title",
"| First field title: First field value",
"| Second field value",
]),
},
{
'input_message': {'attachments': [{
'title': 'First attachment title',
'text': 'First attachment text',
'title_link': 'http://title.link.1',
'from_url': 'http://from.url.1',
}, {
'title': 'Second attachment title',
'text': 'Second attachment text',
'title_link': 'http://title.link.2',
'from_url': 'http://from.url.2',
}]},
'input_text_before': "",
'output': "\n".join([
"| First attachment title (http://title.link.1)",
"| http://from.url.1",
"| First attachment text",
"| Second attachment title (http://title.link.2)",
"| http://from.url.2",
"| Second attachment text",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'color': 'ff0000',
}]},
'input_text_before': "",
'output': "\n".join([
"<[color 16711680]>|<[color reset]> Title",
]),
},
{
'input_message': {'attachments': [{
'title': 'Title',
'color': '#ff0000',
}]},
'input_text_before': "",
'output': "\n".join([
"<[color 16711680]>|<[color reset]> Title",
]),
},
{
'input_message': {'attachments': [{
'text': 'Attachment text',
'original_url': 'http://from.url',
}]},
'input_text_before': "",
'output': "\n".join([
"| Attachment text",
]),
'link_previews': True
},
{
'input_message': {'attachments': [{
'text': 'Attachment text',
'original_url': 'http://from.url',
}]},
'input_text_before': "",
'output': '',
'link_previews': False
},
{
'input_message': {'attachments': [{
'text': 'Attachment text',
}]},
'input_text_before': "",
'output': "\n".join([
"| Attachment text",
]),
'link_previews': False
},
))
def test_unwrap_attachments(case):
wee_slack.config.link_previews = case.get('link_previews')
result = wee_slack.unwrap_attachments(
case['input_message'], case['input_text_before'])
assert result == case['output']
