Add tests for emulator.detect_conflicts

tests/test_emulator.py

@@ -0,0 +1,119 @@
+import contextlib
+import json
+
+from tabi.emulator import detect_conflicts
+
+
+@contextlib.contextmanager
+def dict_opener(line):
+    yield [json.dumps(line)]
+
+
+def test_detect_conflicts_boundary_1():
+    """Check if conflicts detection fails on empty file list"""
+    try:
+        conflicts = detect_conflicts("collector", [])
+        conflicts.next()
+        raise Exception("Should raise an exception on empty file list")
+    except ValueError as error:
+        assert error.message == "no bviews were loaded"
+
+
+def test_detect_conflicts_boundary_2():
+    """Check if conflicts detection fails on a file with partial data"""
+    try:
+        conflicts = detect_conflicts("collector", [{"type": "table_dump_v2"}], opener=dict_opener)
+        conflicts.next()
+        raise Exception("There shouldn't be any conflict to report")
+    except StopIteration:
+        pass
+
+
+def test_detect_conflicts_rib_update():
+    """Check if conflicts detection works for a conflict between rib and update files"""
+    rib = {
+        "entries": [{
+            "peer_ip": "11:33:55:77",
+            "peer_as": 99999.0,
+            "originated_timestamp": 0.0,
+            "as_path": "22 333 4444 55555"
+        }],
+        "type": "table_dump_v2",
+        "timestamp": 1451601234.0,
+        "prefix": "1.2.3.0/24"
+    }
+    update = {
+        "type": "update",
+        "timestamp": 1451606698.0,
+        "peer_as": 11111.0,
+        "peer_ip": "22.44.66.88",
+        "as_path": "1111 2222 3333",
+        "announce": ["1.2.3.0/25"],
+        "withdraw": []
+    }
+    conflicts = detect_conflicts("collector", [rib, update], opener=dict_opener)
+    conflict = conflicts.next()
+    expected = {
+        'timestamp': 1451606698.0,
+        'collector': 'collector',
+        'peer_as': 11111,
+        'peer_ip': "22.44.66.88",
+        'announce': {
+            'type': 'U',
+            'prefix': '1.2.3.0/25',
+            'asn': 3333,
+            'as_path': '1111 2222 3333'
+        },
+        'conflict_with': {
+            'asn': 55555,
+            'prefix': '1.2.3.0/24'
+        },
+        'asn': 55555
+    }
+    assert conflict == expected
+
+
+def test_detect_conflicts_rib_only():
+    """Check if conflicts detection works for a conflict within rib file"""
+    rib1 = {
+        "entries": [{
+            "peer_ip": "11.33.55.77",
+            "peer_as": 66666.0,
+            "originated_timestamp": 1451606699.0,
+            "as_path": "22 333 4444 66666"
+        }],
+        "type": "table_dump_v2",
+        "timestamp": 1451606699.0,
+        "prefix": "1.2.3.0/24"
+    }
+    rib2 = {
+        "entries": [{
+            "peer_ip": "22.44.66.88",
+            "peer_as": 99999.0,
+            "originated_timestamp": 1451605511.0,
+            "as_path": "22 333 4444 55555"
+        }],
+        "type": "table_dump_v2",
+        "timestamp": 1451605511.0,
+        "prefix": "1.2.3.0/24"
+    }
+    conflicts = detect_conflicts("collector", [rib1, rib2], opener=dict_opener)
+    conflict = conflicts.next()
+    expected = {
+        'timestamp': 1451606699.0,
+        'collector': 'collector',
+        'peer_as': 66666,
+        'peer_ip': "11.33.55.77",
+        'announce': {
+            'type': 'F',
+            'prefix': '1.2.3.0/24',
+            'asn': 66666,
+            'as_path': '22 333 4444 66666'
+        },
+        'conflict_with': {
+            'asn': 55555,
+            'prefix': '1.2.3.0/24'
+        },
+        'asn': 55555
+    }
+    assert conflict == expected