From 32e5d633f43ca81dc94ce4f4d6be2a5b59926e23 Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Tue, 7 Apr 2020 20:59:26 +0200
Subject: [PATCH] [core/output] Add tests for pango formatting

---
 core/output.py            |  2 +-
 tests/core/test_output.py | 51 +++++++++++++++++++++++++++++++++++++--
 2 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/core/output.py b/core/output.py
index 4b19260..436c4af 100644
--- a/core/output.py
+++ b/core/output.py
@@ -52,7 +52,7 @@ class block(object):
         text = attr.get('full_text', '')
         if 'full_text' in attr:
             del attr['full_text']
-        result = '<span '
+        result = '<span'
         for key, value in attr.items():
             result = '{} {}="{}"'.format(result, key, value)
         result = '{}>{}</span>'.format(result, text)
diff --git a/tests/core/test_output.py b/tests/core/test_output.py
index ea1750f..aeb2d70 100644
--- a/tests/core/test_output.py
+++ b/tests/core/test_output.py
@@ -17,11 +17,16 @@ class i3(unittest.TestCase):
             widgets=[widget, widget, widget])
         self.paddedTheme = core.theme.Theme(raw_data = {
             'defaults': { 'padding': ' ' }
-        });
+        })
         self.separator = '***';
         self.separatorTheme = core.theme.Theme(raw_data = {
             'defaults': { 'separator': self.separator, 'fg': 'red', 'bg': 'blue' }
-        });
+        })
+        self.someBlock = core.output.block(
+            theme=self.separatorTheme,
+            module=self.someModule, 
+            widget=self.someModule.widget()
+        )
 
     def test_start(self):
         core.event.clear()
@@ -89,4 +94,46 @@ class i3(unittest.TestCase):
         core.output.dump_json(obj)
         obj.dict_assert_called_once_with()
 
+    def test_assign(self):
+        src = {
+            'a': 'x', 'b': 'y', 'c': 'z'
+        }
+        dst = {}
+
+        core.output.assign(src, dst, 'a')
+        self.assertEqual(dst['a'], src['a'])
+
+        core.output.assign(src, dst, '123', 'b')
+        self.assertEqual(dst['123'], src['b'])
+
+        core.output.assign(src, dst, 'blub', default='def')
+        self.assertEqual('def', dst['blub'])
+
+    def test_pango_detection(self):
+        self.assertFalse(self.someBlock.is_pango({}))
+        self.assertTrue(self.someBlock.is_pango({ 'pango': {} }))
+
+    def test_pangoize(self):
+        self.assertEqual('test', self.someBlock.pangoize('test'))
+        self.assertFalse('markup' in self.someBlock.dict())
+
+        self.assertEqual('<span attr="blub" x="y">test</span>',
+            self.someBlock.pangoize({ 'pango': { 'attr': 'blub', 'x': 'y', 'full_text': 'test' } })
+        )
+        self.assertEqual('pango', self.someBlock.dict()['markup'])
+
+    def test_padding(self):
+        self.someBlock.set('padding', '***')
+        self.someBlock.set('full_text', 'test')
+
+        self.assertEqual('***test***', self.someBlock.dict()['full_text'])
+
+    def test_pre_suffix(self):
+        self.someBlock.set('padding', '*')
+        self.someBlock.set('prefix', 'pre')
+        self.someBlock.set('suffix', 'suf')
+        self.someBlock.set('full_text', 'test')
+
+        self.assertEqual('*pre*test*suf*', self.someBlock.dict()['full_text'])
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4