Issue 5458, 5457 - Refactor _parse_targetspec, add support for locales and blockTotal

sitescripts/notifications/test/parser.py

 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-present eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
@@ skipping 142 matching lines @@
         self.assertEqual(notifications[1]['severity'], 'critical')
         self.assertEqual(notifications[2]['severity'], 'normal')
         self.assertEqual(notifications[3]['severity'], 'relentless')
 
     def test_urls(self):
         self.notification_to_load = [
             ('1', '\nurls = adblockplus.org\n'),
             ('1', '\nurls = adblockplus.org eyeo.com\n'),
         ]
         notifications = parser.load_notifications()
-        self.assertEqual(len(notifications), 2)
-        self.assertEqual(
-            notifications[0]['urlFilters'], ['ADBLOCKPLUS.ORG^$document'])
-        self.assertEqual(notifications[1]['urlFilters'], [
+
+        assert len(notifications) == 2
+        assert notifications[0]['urlFilters'] == ['ADBLOCKPLUS.ORG^$document']
+        assert notifications[1]['urlFilters'] == [
             'ADBLOCKPLUS.ORG^$document',
-            'EYEO.COM^$document'])
+            'EYEO.COM^$document']
 
     def test_target(self):
         self.notification_to_load = [
             ('1', '\ntarget = extension=adblockplus\n'),
             ('2', '\ntarget = extensionVersion=1.2.3\n'),
             ('3', '\ntarget = extensionVersion>=1.2.3\n'),
             ('4', '\ntarget = extensionVersion<=1.2.3\n'),
             ('5', '\ntarget = application=chrome\n'),
             ('6', '\ntarget = applicationVersion=1.2.3\n'),
             ('7', '\ntarget = applicationVersion>=1.2.3\n'),
             ('8', '\ntarget = applicationVersion<=1.2.3\n'),
             ('9', '\ntarget = platform=chromium\n'),
             ('10', '\ntarget = platformVersion=1.2.3\n'),
             ('11', '\ntarget = platformVersion>=1.2.3\n'),
             ('12', '\ntarget = platformVersion<=1.2.3\n'),
             ('13', '\ntarget = blockedTotal=10\n'),
             ('14', '\ntarget = blockedTotal>=10\n'),
             ('15', '\ntarget = blockedTotal<=10\n'),
             ('16', '\ntarget = locales=en-US\n'),
             ('17', '\ntarget = locales=en-US,de-DE\n'),
         ]
 
         notifications = parser.load_notifications()
-        self.assertEqual(len(notifications), 17)

[Vasily Kuznetsov, 2017/09/20 15:45:33]

These tests are run with pytest nowadays, so you can use

    assert len(notifications) == 17

instead of `self.assertEqual`. It's more compact and readable and pytest
produces very helpful failure messages when the assertion fails.

[wspee, 2017/09/21 08:10:52]

Done.

[Vasily Kuznetsov, 2017/09/21 09:41:30]

I actually meant it for all new assertions (the ones below would benefit more
than this one in terms of readability).

-
-        self.assertEqual(
-            notifications[0]['targets'][0]['extension'], 'adblockplus')
-        self.assertEqual(
-            notifications[1]['targets'][0]['extensionMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[1]['targets'][0]['extensionMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[2]['targets'][0]['extensionMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[3]['targets'][0]['extensionMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[4]['targets'][0]['application'], 'chrome')
-        self.assertEqual(
-            notifications[5]['targets'][0]['applicationMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[5]['targets'][0]['applicationMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[6]['targets'][0]['applicationMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[7]['targets'][0]['applicationMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[8]['targets'][0]['platform'], 'chromium')
-        self.assertEqual(
-            notifications[9]['targets'][0]['platformMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[9]['targets'][0]['platformMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[10]['targets'][0]['platformMinVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[11]['targets'][0]['platformMaxVersion'], '1.2.3')
-        self.assertEqual(
-            notifications[12]['targets'][0]['blockedTotalMin'], 10)
-        self.assertEqual(
-            notifications[12]['targets'][0]['blockedTotalMax'], 10)
-        self.assertEqual(
-            notifications[13]['targets'][0]['blockedTotalMin'], 10)
-        self.assertEqual(
-            notifications[14]['targets'][0]['blockedTotalMax'], 10)
-        self.assertEqual(
-            notifications[15]['targets'][0]['locales'], ['en-US'])
-        self.assertEqual(
-            notifications[16]['targets'][0]['locales'], ['en-US', 'de-DE'])
+
+        assert len(notifications) == 17
+        assert notifications[0]['targets'] == [{'extension': 'adblockplus'}]
+        assert notifications[1]['targets'] == [{
+            'extensionMinVersion': '1.2.3',
+            'extensionMaxVersion': '1.2.3'}]
+        assert notifications[2]['targets'] == [
+            {'extensionMinVersion': '1.2.3'}]
+        assert notifications[3]['targets'] == [{
+            'extensionMaxVersion': '1.2.3'}]
+        assert notifications[4]['targets'] == [{'application': 'chrome'}]
+        assert notifications[5]['targets'] == [{
+            'applicationMinVersion': '1.2.3',
+            'applicationMaxVersion': '1.2.3'}]
+        assert notifications[6]['targets'] == [{
+            'applicationMinVersion': '1.2.3'}]
+        assert notifications[7]['targets'] == [{
+            'applicationMaxVersion': '1.2.3'}]
+        assert notifications[8]['targets'] == [{'platform': 'chromium'}]
+        assert notifications[9]['targets'] == [{
+            'platformMinVersion': '1.2.3',
+            'platformMaxVersion': '1.2.3'}]
+        assert notifications[10]['targets'] == [{
+            'platformMinVersion': '1.2.3'}]
+        assert notifications[11]['targets'] == [{
+            'platformMaxVersion': '1.2.3'}]
+        assert notifications[12]['targets'] == [{
+            'blockedTotalMin': 10,
+            'blockedTotalMax': 10}]
+        assert notifications[13]['targets'] == [{'blockedTotalMin': 10}]
+        assert notifications[14]['targets'] == [{'blockedTotalMax': 10}]
+        assert notifications[15]['targets'] == [{'locales': ['en-US']}]
+        assert notifications[16]['targets'] == [{
+            'locales': ['en-US', 'de-DE']}]
 
 
 if __name__ == '__main__':
     unittest.main()