test/test_blacklists.py: simple JSON validation

test/test_blacklists.py

@@ -1,9 +1,13 @@
 #!/usr/bin/env python3
 # coding=utf-8
 
+import json
+from glob import glob
+from os import unlink
+
 import pytest
 
-from glob import glob
+from blacklists import Blacklist, JSONParserCIDR, JSONParserASN
 from helpers import files_changed, blacklist_integrity_check
 
 
@@ -22,3 +26,62 @@ def test_remote_diff():
     false_diff = "h j q t"
     assert files_changed(true_diff, file_set)
     assert not files_changed(false_diff, file_set)
+
+
+def test_json_blacklist():
+    with open('test_ip.json', 'w') as j:
+        json.dump({
+            'schema': 'json_cidr',
+            'schema_version': '2019120601',
+            'items': [
+                {'ip': '1.2.3.4'},
+                {'ip': '2.3.4.5', 'disable': True},
+                {'ip': '3.4.5.6', 'comment': 'comment'},
+            ]}, j)
+    blacklist = Blacklist(('test_ip.json', JSONParserCIDR))
+    with pytest.raises(ValueError) as e:
+        blacklist.add('1.3.34')
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'ip': '1.3.4'})
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'ip': '1.2.3.4'})
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'ip': '2.3.4.5'})
+    with pytest.raises(ValueError) as e:
+        blacklist.remove({'ip': '34.45.56.67'})
+    blacklist.add({'ip': '1.3.4.5'})
+    assert '1.2.3.4' in blacklist.parse()
+    assert '2.3.4.5' not in blacklist.parse()
+    assert '3.4.5.6' in blacklist.parse()
+    blacklist.remove({'ip': '3.4.5.6'})
+    assert '3.4.5.6' not in blacklist.parse()
+    unlink('test_ip.json')
+
+
+def test_json_asn():
+    with open('test_asn.json', 'w') as j:
+        json.dump({
+            'schema': 'json_asn',
+            'schema_version': '2019120601',
+            'items': [
+                {'asn': '123'},
+                {'asn': '234', 'disable': True},
+                {'asn': '345', 'comment': 'comment'},
+            ]}, j)
+    blacklist = Blacklist(('test_asn.json', JSONParserASN))
+    with pytest.raises(ValueError) as e:
+        blacklist.add('123')
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'asn': 'invalid'})
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'asn': '123'})
+    with pytest.raises(ValueError) as e:
+        blacklist.add({'asn': '234'})
+    with pytest.raises(ValueError) as e:
+        blacklist.remove({'asn': '9897'})
+    assert '123' in blacklist.parse()
+    assert '234' not in blacklist.parse()
+    assert '345' in blacklist.parse()
+    blacklist.remove({'asn': '345'})
+    assert '345' not in blacklist.parse()
+    unlink('test_asn.json')