Fixing disable issue

Change-Id: I2ff36d15aaf770528ac3f0c207a3944799355acc
diff --git a/xos/synchronizer/model_policies/test_model_policy_att_workflow_driver_whitelistentry.py b/xos/synchronizer/model_policies/test_model_policy_att_workflow_driver_whitelistentry.py
index 68e1752..369ef70 100644
--- a/xos/synchronizer/model_policies/test_model_policy_att_workflow_driver_whitelistentry.py
+++ b/xos/synchronizer/model_policies/test_model_policy_att_workflow_driver_whitelistentry.py
@@ -73,48 +73,62 @@
         sys.path = self.sys_path_save
         self.service = None
 
+    def test_enable_onu(self):
+        from helpers import AttHelpers
+        si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id, valid="invalid")
+        with patch.object(AttHelpers, "validate_onu") as validate_onu, \
+            patch.object(si, "save") as save_si:
+            validate_onu.return_value = [True, "valid onu"]
+
+            self.policy.validate_onu_state(si)
+
+            save_si.assert_called_once()
+            save_si.assert_called_with(always_update_timestamp=True, update_fields=['no_sync', 'updated', 'onu_state', 'status_message', 'authentication_state'])
+
+            self.assertEqual("valid onu", si.status_message)
+
+    def test_disable_onu(self):
+        from helpers import AttHelpers
+        si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id, valid="invalid")
+        with patch.object(AttHelpers, "validate_onu") as validate_onu, \
+            patch.object(si, "save") as save_si:
+            validate_onu.return_value = [False, "invalid onu"]
+
+            self.policy.validate_onu_state(si)
+
+            save_si.assert_called_once()
+            save_si.assert_called_with(always_update_timestamp=True, update_fields=['no_sync', 'updated', 'onu_state', 'status_message', 'authentication_state'])
+
+            self.assertEqual("invalid onu", si.status_message)
+
     def test_whitelist_update(self):
-        """
-        When a whitelist entry is added, see that the AttWorkflowDriverIServicenstance was set to valid
-        """
-        with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as oss_si_items:
-            si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id, valid="invalid")
+        si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id)
+        wle = AttWorkflowDriverWhiteListEntry(serial_number="brcm333", owner_id=self.service.id, owner=self.service)
+        with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as oss_si_items, \
+            patch.object(self.policy, "validate_onu_state") as validate_onu_state, \
+            patch.object(wle, "save") as wle_save:
             oss_si_items.return_value = [si]
 
-            wle = AttWorkflowDriverWhiteListEntry(serial_number="BRCM333", owner_id=self.service.id, owner=self.service)
 
             self.policy.handle_update(wle)
 
-            self.assertEqual(si.valid, "valid")
-
-    def test_whitelist_update_case_insensitive(self):
-        """
-        When a whitelist entry is added, see that the AttWorkflowDriverIServicenstance was set to valid
-        """
-        with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as oss_si_items:
-            si = AttWorkflowDriverServiceInstance(serial_number="brcm333", owner_id=self.service.id, valid="invalid")
-            oss_si_items.return_value = [si]
-
-            wle = AttWorkflowDriverWhiteListEntry(serial_number="BRCM333", owner_id=self.service.id, owner=self.service)
-
-            self.policy.handle_update(wle)
-
-            self.assertEqual(si.valid, "valid")
+            validate_onu_state.assert_called_with(si)
+            self.assertTrue(wle.backend_need_delete_policy)
+            wle_save.assert_called_with(update_fields=["backend_need_delete_policy"])
 
     def test_whitelist_delete(self):
-        """
-        When a whitelist entry is deleted, see that the AttWorkflowDriverIServicenstance was set to invalid
-        """
-        with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as oss_si_items:
-            si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id, valid="valid")
+        si = AttWorkflowDriverServiceInstance(serial_number="BRCM333", owner_id=self.service.id)
+        wle = AttWorkflowDriverWhiteListEntry(serial_number="brcm333", owner_id=self.service.id, owner=self.service)
+        with patch.object(AttWorkflowDriverServiceInstance.objects, "get_items") as oss_si_items, \
+                patch.object(self.policy, "validate_onu_state") as validate_onu_state, \
+                patch.object(wle, "save") as wle_save:
             oss_si_items.return_value = [si]
 
-            wle = AttWorkflowDriverWhiteListEntry(serial_number="BRCM333", owner_id=self.service.id, owner=self.service)
-
             self.policy.handle_delete(wle)
 
-            self.assertEqual(si.valid, "invalid")
-
+            validate_onu_state.assert_called_with(si)
+            self.assertTrue(wle.backend_need_reap)
+            wle_save.assert_called_with(update_fields=["backend_need_reap"])
 if __name__ == '__main__':
     unittest.main()