import unittest
from reports import normalize, human_views, render


class ReportTests(unittest.TestCase):
    def test_normalize(self):
        self.assertEqual(normalize('http://vk.ru/wall-123_456')['url'], 'https://vk.com/wall-123_456')
        self.assertEqual(normalize('https://telegram.me/s/Channel/123')['url'], 'https://t.me/channel/123')
        for link in ('https://t.me/c/123/456', 'https://vk.ru.evil.test/wall-123_456', 'https://vk.com/club123'):
            with self.assertRaises(ValueError):
                normalize(link)

    def test_format_without_zero(self):
        for value, expected in ((2000, '2K'), (10600, '10,6K'), (1000000, '1M'), (845, '0,8K'), (100, '0,1K'), (35, '0K'), (0, '0K')):
            self.assertEqual(human_views(value), expected)

    def test_errors_are_not_zero_views(self):
        first = dict(normalize('https://t.me/channel/1'), title='Имя 🗿', views=2000)
        second = dict(normalize('https://t.me/channel/2'), title='Имя', views=None, error='Нет счётчика')
        r = render([first, second])
        self.assertIn('<b>2K</b>', r['html'])
        self.assertNotIn('🗿', r['html'])
        self.assertNotIn('0K', r['html'])
        self.assertIn('Неполный итог', r['html'])
        self.assertEqual(len(r['errors']), 1)

    def test_grouping_and_html_escaping(self):
        records = [dict(normalize('https://max.ru/channel/AaE-viFyB3E'), title='<script>alert(1)</script> Имя', views=1000),
                   dict(normalize('https://max.ru/channel/AaE-viFyB3F'), title='Имя', views=2000)]
        r = render(records)
        self.assertIn('<h2>Макс</h2>', r['html'])
        self.assertIn('ещё</a>', r['html'])
        self.assertNotIn('<script>', r['html'])
        self.assertIn('<b>3K</b>', r['html'])


if __name__ == '__main__':
    unittest.main()