commit 242c29672003dc654c1f45eb0eb9ede648d88506
author Matteo Scandolo <teo@opennetworking.org> Tue Aug 28 16:50:47 2018 -0700
committer Matteo Scandolo <teo@opennetworking.org> Tue Aug 28 17:42:34 2018 -0700
tree 6e2e03f855f7d91a2a8aae7ad3e867ab32e4749d
parent b7597155b464418610638625ecf1a7d67273fd3b

Fixing SI creation

Change-Id: I7b60169fd2e4f39f8bc1a74c8db1df0e90f69a0e

xos/synchronizer/event_steps/test_onu_events.py

diff --git a/xos/synchronizer/event_steps/test_onu_events.py b/xos/synchronizer/event_steps/test_onu_events.py
index b690663..bea8c0c 100644
--- a/xos/synchronizer/event_steps/test_onu_events.py
+++ b/xos/synchronizer/event_steps/test_onu_events.py
@@ -85,6 +85,8 @@
         }
         self.event.value = json.dumps(self.event_dict)
 
+        self.att = AttWorkflowDriverService(name="att-workflow-driver")
+
     def tearDown(self):
         sys.path = self.sys_path_save
 
@@ -92,9 +94,11 @@
     def test_create_instance(self):
 
         with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as att_si_mock , \
+            patch.object(AttWorkflowDriverService.objects, "get_items") as service_mock, \
             patch.object(AttWorkflowDriverServiceInstance, "save", autospec=True) as mock_save:
 
             att_si_mock.return_value = []
+            service_mock.return_value = [self.att]
 
             self.event_step.process_event(self.event)
 
@@ -105,7 +109,7 @@
             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")
+            self.assertEqual(att_si.onu_state, "ENABLED")
 
     def test_reuse_instance(self):
 
@@ -129,7 +133,7 @@
             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")
+            self.assertEqual(att_si.onu_state, "ENABLED")
 
     def test_disable_onu(self):
         self.event_dict = {
@@ -141,9 +145,11 @@
         self.event.value = json.dumps(self.event_dict)
 
         with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as att_si_mock , \
+            patch.object(AttWorkflowDriverService.objects, "get_items") as service_mock, \
             patch.object(AttWorkflowDriverServiceInstance, "save", autospec=True) as mock_save:
 
             att_si_mock.return_value = []
+            service_mock.return_value = [self.att]
 
             self.event_step.process_event(self.event)