Upgrade to SDL 2.0.2

[ric-plt/a1.git] / tests / test_controller.py

diff --git a/tests/test_controller.py b/tests/test_controller.py
index 833fc6e..bbfe996 100644 (file)
--- a/tests/test_controller.py
+++ b/tests/test_controller.py
@@ -20,8 +20,9 @@ tests for controller
 import time
 import json
 from rmr.rmr_mocks import rmr_mocks
+from ricsdl.syncstorage import SyncStorage
+from ricsdl.exceptions import RejectedByBackend, NotConnected, BackendError
 from a1 import a1rmr, data
-from .a1test_helpers import MockSDLWrapper
 
 RCV_ID = "test_receiver"
 ADM_CRTL_TID = 6660666
@@ -231,7 +232,7 @@ def _verify_instance_and_status(client, expected_instance, expected_status, expe
 def setup_module():
     """module level setup"""
 
-    data.SDL = MockSDLWrapper()  # patch SDL
+    data.SDL.sdl = SyncStorage(fake_db_backend="dict")
 
     def noop():
         pass
@@ -349,6 +350,18 @@ def test_bad_instances(client, monkeypatch, adm_type_good):
     assert res.status_code == 204
 
     # test 503 handlers
+
+    def monkey_set(key, value):
+        # set a key override function that throws sdl errors on certain keys
+        if key == "a1.policy_type.111":
+            raise RejectedByBackend()
+        if key == "a1.policy_type.112":
+            raise NotConnected()
+        if key == "a1.policy_type.113":
+            raise BackendError()
+
+    monkeypatch.setattr("a1.data.SDL.set", monkey_set)
+
     res = client.put("/a1-p/policytypes/111", json=adm_type_good)
     assert res.status_code == 503
     res = client.put("/a1-p/policytypes/112", json=adm_type_good)