Add process response tests (#86)

Maria Korlotian · web-flow · commit 4dd6a558b68f · 2020-11-13T15:07:59.000+01:00
diff --git a/castle/test/core/process_response_test.py b/castle/test/core/process_response_test.py
@@ -21,13 +21,35 @@ def test_response_none(self):
     def test_response_empty(self):
         self.assertEqual(CoreProcessResponse(response(body=b'')).call(), {})
 
-    def test_response_authenticate(self):
+    def test_response_authenticate_allow(self):
         self.assertEqual(
             CoreProcessResponse(
                 response(body=b'{"action":"allow","user_id":"12345"}')).call(),
             {"action": "allow", "user_id": "12345"}
         )
 
+    def test_response_authenticate_allow_with_props(self):
+        self.assertEqual(
+            CoreProcessResponse(
+                response(body=b'{"action":"allow","user_id":"12345","internal":{}}')).call(),
+            {"action": "allow", "user_id": "12345", "internal": {}}
+        )
+
+    def test_response_authenticate_deny_without_rp(self):
+        self.assertEqual(
+            CoreProcessResponse(
+                response(body=b'{"action":"deny","user_id":"1","device_token":"abc"}')).call(),
+            {"action": "deny", "user_id": "1", "device_token": "abc"}
+        )
+
+    def test_response_authenticate_deny_with_rp(self):
+        self.assertEqual(
+            CoreProcessResponse(
+                response(body=b'{"action":"deny","user_id":"1","device_token":"abc","risk_policy":{"id":"123","revision_id":"abc","name":"def","type":"bot"}}')).call(),
+            {"action": "deny", "user_id": "1", "device_token": "abc", "risk_policy": {
+                "id": "123", "revision_id": "abc", "name": "def", "type": "bot"}}
+        )
+
     def test_verify_200_299(self):
         for status_code in range(200, 299):
             self.assertEqual(
