import unittest
from unittest.mock import Mock, patch

import legacy_parser as engine


class VkCountersTests(unittest.TestCase):
    def test_missing_counter_is_not_fabricated_zero(self):
        url = 'https://vk.com/wall-123_456'
        for field in ({}, {'views': {}}, {'views': None}, {'views': {'count': '100'}}):
            with self.subTest(field=field):
                post = dict(id=456, owner_id=-123, **field)
                response = Mock(status_code=200, json=lambda: {'response': {'items': [post], 'groups': [{'id': 123, 'name': 'Сообщество'}]}})
                with patch.object(engine.requests, 'get', return_value=response):
                    result = engine.process_vk_batch([(url, -123, '456')])
                self.assertEqual(result[url], ('Сообщество', None))

    def test_real_zero_and_nonzero_are_preserved(self):
        url = 'https://vk.com/wall-123_456'
        for count in (0, 23000):
            post = {'id': 456, 'owner_id': -123, 'views': {'count': count}}
            response = Mock(status_code=200, json=lambda: {'response': {'items': [post]}})
            with patch.object(engine.requests, 'get', return_value=response):
                result = engine.process_vk_batch([(url, -123, '456')])
            self.assertEqual(result[url][1], count)


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