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"]