Support colorizing attachment prefix or line

Thanks to @Informatic for the initial implementation of this in PR #426.

I have seen that most attachment colors are without the leading #, but a
few does include it, so we have to handle it.

Fixes #424, closes #426

2 files changed, 25 insertions, 0 deletions

diff --git a/_pytest/conftest.py b/_pytest/conftest.py
index ad0949e..1ad2310 100644
--- a/_pytest/conftest.py
+++ b/_pytest/conftest.py
@@ -117,6 +117,11 @@ class FakeWeechat():
         return ""
     def color(self, name):
         return "<[color {}]>".format(name)
+    def info_get(self, info_name, arguments):
+        if info_name == "color_rgb2term":
+            return arguments
+        else:
+            return ""
     def __getattr__(self, name):
         def method(*args):
             pass
diff --git a/_pytest/test_unwrap_attachments.py b/_pytest/test_unwrap_attachments.py
index ea3a35d..054c0af 100644
--- a/_pytest/test_unwrap_attachments.py
+++ b/_pytest/test_unwrap_attachments.py
@@ -267,6 +267,26 @@ import pytest
             "| 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",
+        ]),
+    },
 ))
 def test_unwrap_attachments(case):
     result = wee_slack.unwrap_attachments(