[SEBA-164] Forcing subscriber auth if onu goes down
Change-Id: I6d1b721ac67533fb2b58048666771b50d3a84b2b
diff --git a/xos/synchronizer/event_steps/test_onu_events.py b/xos/synchronizer/event_steps/test_onu_events.py
index e81f876..b690663 100644
--- a/xos/synchronizer/event_steps/test_onu_events.py
+++ b/xos/synchronizer/event_steps/test_onu_events.py
@@ -94,7 +94,7 @@
with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as att_si_mock , \
patch.object(AttWorkflowDriverServiceInstance, "save", autospec=True) as mock_save:
- att_si_mock.side_effect = IndexError
+ att_si_mock.return_value = []
self.event_step.process_event(self.event)
@@ -105,3 +105,56 @@
self.assertEqual(att_si.serial_number, self.event_dict['serial_number'])
self.assertEqual(att_si.of_dpid, self.event_dict['of_dpid'])
self.assertEqual(att_si.uni_port_id, self.event_dict['uni_port_id'])
+ self.assertEqual(att_si.onu_state, "ACTIVE")
+
+ def test_reuse_instance(self):
+
+ si = AttWorkflowDriverServiceInstance(
+ serial_number=self.event_dict["serial_number"],
+ of_dpid="foo",
+ uni_port_id="foo"
+ )
+
+ with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as att_si_mock , \
+ patch.object(AttWorkflowDriverServiceInstance, "save", autospec=True) as mock_save:
+
+ att_si_mock.return_value = [si]
+
+ self.event_step.process_event(self.event)
+
+ att_si = mock_save.call_args[0][0]
+
+ self.assertEqual(mock_save.call_count, 1)
+
+ self.assertEqual(att_si.serial_number, self.event_dict['serial_number'])
+ self.assertEqual(att_si.of_dpid, self.event_dict['of_dpid'])
+ self.assertEqual(att_si.uni_port_id, self.event_dict['uni_port_id'])
+ self.assertEqual(att_si.onu_state, "ACTIVE")
+
+ def test_disable_onu(self):
+ self.event_dict = {
+ 'status': 'disabled',
+ 'serial_number': 'BRCM1234',
+ 'of_dpid': 'of:109299321',
+ 'uni_port_id': 16
+ }
+ self.event.value = json.dumps(self.event_dict)
+
+ with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as att_si_mock , \
+ patch.object(AttWorkflowDriverServiceInstance, "save", autospec=True) as mock_save:
+
+ att_si_mock.return_value = []
+
+ self.event_step.process_event(self.event)
+
+ att_si = mock_save.call_args[0][0]
+
+ self.assertEqual(mock_save.call_count, 1)
+
+ self.assertEqual(att_si.serial_number, self.event_dict['serial_number'])
+ self.assertEqual(att_si.of_dpid, self.event_dict['of_dpid'])
+ self.assertEqual(att_si.uni_port_id, self.event_dict['uni_port_id'])
+ self.assertEqual(att_si.onu_state, "DISABLED")
+
+if __name__ == '__main__':
+ unittest.main()
\ No newline at end of file