kytos-ng /
sdntrace_cp
| @@ 61-108 (lines=48) @@ | ||
| 58 | mock_find_circuits.assert_called_once() |
|
| 59 | self.assertEqual(result, formatted) |
|
| 60 | ||
| 61 | @patch("napps.amlight.sdntrace_cp.automate.Automate.find_circuits") |
|
| 62 | def test_get_circuit_not_found(self, mock_find_circuits): |
|
| 63 | """Verify get circuit not finding a circuit.""" |
|
| 64 | ||
| 65 | formatted = [ |
|
| 66 | { |
|
| 67 | "dpid": "00:00:00:00:00:00:00:01", |
|
| 68 | "in_port": 3, |
|
| 69 | "in_vlan": 100, |
|
| 70 | "out_port": 2, |
|
| 71 | "out_vlan": 200, |
|
| 72 | }, |
|
| 73 | { |
|
| 74 | "dpid": "00:00:00:00:00:00:00:02", |
|
| 75 | "in_port": 3, |
|
| 76 | "in_vlan": 200, |
|
| 77 | "out_port": 2, |
|
| 78 | "out_vlan": 100, |
|
| 79 | }, |
|
| 80 | { |
|
| 81 | "dpid": "00:00:00:00:00:00:00:03", |
|
| 82 | "in_port": 3, |
|
| 83 | "in_vlan": 100, |
|
| 84 | "out_port": 2, |
|
| 85 | "out_vlan": 200, |
|
| 86 | }, |
|
| 87 | ] |
|
| 88 | ||
| 89 | circuits = [] |
|
| 90 | circuits.append({"circuit": formatted, "entries": []}) |
|
| 91 | ||
| 92 | circuit = { |
|
| 93 | "dpid_a": circuits[0]["circuit"][0]["dpid"], |
|
| 94 | "port_a": circuits[0]["circuit"][0]["in_port"], |
|
| 95 | "vlan_a": circuits[0]["circuit"][0]["in_vlan"], |
|
| 96 | "dpid_z": circuits[0]["circuit"][0]["dpid"], |
|
| 97 | "port_z": circuits[0]["circuit"][0]["out_port"], |
|
| 98 | "vlan_z": circuits[0]["circuit"][0]["out_vlan"], |
|
| 99 | } |
|
| 100 | ||
| 101 | tracer = MagicMock() |
|
| 102 | automate = Automate(tracer) |
|
| 103 | automate._circuits = circuits |
|
| 104 | ||
| 105 | result = automate.get_circuit(circuit) |
|
| 106 | ||
| 107 | mock_find_circuits.assert_called_once() |
|
| 108 | self.assertIsNone(result) |
|
| 109 | ||
| 110 | @patch("napps.amlight.sdntrace_cp.automate.Automate.find_circuits") |
|
| 111 | def test_get_circuit_empty(self, mock_find_circuits): |
|
| @@ 14-59 (lines=46) @@ | ||
| 11 | class TestAutomate(TestCase): |
|
| 12 | """Test class Automate.""" |
|
| 13 | ||
| 14 | @patch("napps.amlight.sdntrace_cp.automate.Automate.find_circuits") |
|
| 15 | def test_get_circuit(self, mock_find_circuits): |
|
| 16 | """Verify get circuit success.""" |
|
| 17 | formatted = [ |
|
| 18 | { |
|
| 19 | "dpid": "00:00:00:00:00:00:00:01", |
|
| 20 | "in_port": 3, |
|
| 21 | "in_vlan": 100, |
|
| 22 | "out_port": 2, |
|
| 23 | "out_vlan": 200, |
|
| 24 | }, |
|
| 25 | { |
|
| 26 | "dpid": "00:00:00:00:00:00:00:02", |
|
| 27 | "in_port": 3, |
|
| 28 | "in_vlan": 200, |
|
| 29 | "out_port": 2, |
|
| 30 | "out_vlan": 100, |
|
| 31 | }, |
|
| 32 | { |
|
| 33 | "dpid": "00:00:00:00:00:00:00:03", |
|
| 34 | "in_port": 3, |
|
| 35 | "in_vlan": 100, |
|
| 36 | "out_port": 2, |
|
| 37 | "out_vlan": 200, |
|
| 38 | }, |
|
| 39 | ] |
|
| 40 | ||
| 41 | circuits = [] |
|
| 42 | circuits.append({"circuit": formatted, "entries": []}) |
|
| 43 | ||
| 44 | circuit = { |
|
| 45 | "dpid_a": circuits[0]["circuit"][0]["dpid"], |
|
| 46 | "port_a": circuits[0]["circuit"][0]["in_port"], |
|
| 47 | "vlan_a": circuits[0]["circuit"][0]["in_vlan"], |
|
| 48 | "dpid_z": circuits[0]["circuit"][2]["dpid"], |
|
| 49 | "port_z": circuits[0]["circuit"][2]["out_port"], |
|
| 50 | "vlan_z": circuits[0]["circuit"][2]["out_vlan"], |
|
| 51 | } |
|
| 52 | ||
| 53 | tracer = MagicMock() |
|
| 54 | automate = Automate(tracer) |
|
| 55 | mock_find_circuits.return_value = circuits |
|
| 56 | result = automate.get_circuit(circuit) |
|
| 57 | ||
| 58 | mock_find_circuits.assert_called_once() |
|
| 59 | self.assertEqual(result, formatted) |
|
| 60 | ||
| 61 | @patch("napps.amlight.sdntrace_cp.automate.Automate.find_circuits") |
|
| 62 | def test_get_circuit_not_found(self, mock_find_circuits): |
|
