From 7e83d0876ddb84a45e130eeba28bc40ef53c074b Mon Sep 17 00:00:00 2001
From: Yaron Yogev <yaronyogev@gmail.com>
Date: Thu, 27 Jul 2017 09:02:54 +0300
Subject: Calipso initial release for OPNFV

Change-Id: I7210c244b0c10fa80bfa8c77cb86c9d6ddf8bc88
Signed-off-by: Yaron Yogev <yaronyogev@gmail.com>
---
 app/test/api/test_base.py | 101 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 app/test/api/test_base.py

(limited to 'app/test/api/test_base.py')

diff --git a/app/test/api/test_base.py b/app/test/api/test_base.py
new file mode 100644
index 0000000..c126b2b
--- /dev/null
+++ b/app/test/api/test_base.py
@@ -0,0 +1,101 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+import copy
+
+
+from api.app import App
+from api.middleware.authentication import AuthenticationMiddleware
+from api.responders.responder_base import ResponderBase
+from api.backends.ldap_access import LDAPAccess
+from falcon.testing import TestCase
+from test.api.responders_test.test_data import base
+from unittest.mock import MagicMock
+from utils.mongo_access import MongoAccess
+
+
+def mock_auth_method(*args):
+    return None
+
+
+class TestBase(TestCase):
+
+    def setUp(self, authenticate=False):
+        super().setUp()
+        # mock
+        self.authenticate = authenticate
+        if not authenticate:
+            self.original_auth_method = AuthenticationMiddleware.process_request
+            AuthenticationMiddleware.process_request = mock_auth_method
+
+        ResponderBase.get_constants_by_name = MagicMock(side_effect=
+                                                        lambda name: base.CONSTANTS_BY_NAMES[name])
+        # mock mongo access
+        MongoAccess.mongo_connect = MagicMock()
+        MongoAccess.db = MagicMock()
+        MongoAccess.client = MagicMock()
+        # mock ldap access
+        LDAPAccess.get_ldap_params = MagicMock()
+        LDAPAccess.connect_ldap_server = MagicMock()
+
+        log_level = 'debug'
+        self.app = App(log_level=log_level).get_app()
+
+    def validate_get_request(self, url, params={}, headers=None, mocks={},
+                             side_effects={},
+                             expected_code=base.SUCCESSFUL_CODE,
+                             expected_response=None):
+        self.validate_request("GET", url, params, headers, "",
+                              mocks, side_effects,
+                              expected_code,
+                              expected_response)
+
+    def validate_request(self, action, url, params, headers, body,
+                         mocks, side_effects, expected_code,
+                         expected_response):
+        for mock_method, mock_data in mocks.items():
+            mock_method.return_value = mock_data
+
+        for mock_method, side_effect in side_effects.items():
+            mock_method.side_effect = side_effect
+
+        result = self.simulate_request(action, url, params=params, headers=headers, body=body)
+        self.assertEqual(result.status, expected_code)
+        if expected_response:
+            self.assertEqual(result.json, expected_response)
+
+    def validate_post_request(self, url, headers={}, body="", mocks={},
+                              side_effects={},
+                              expected_code=base.CREATED_CODE, expected_response=None):
+        self.validate_request("POST", url, {}, headers, body, mocks, side_effects,
+                              expected_code, expected_response)
+
+    def validate_delete_request(self, url, params={}, headers={}, mocks={},
+                                side_effects={},
+                                expected_code=base.SUCCESSFUL_CODE, expected_response=None):
+        self.validate_request("DELETE", url, params, headers, "",
+                              mocks, side_effects,
+                              expected_code,
+                              expected_response)
+
+    def get_updated_data(self, original_data, deleted_keys=[], updates={}):
+        copy_data = copy.deepcopy(original_data)
+
+        for key in deleted_keys:
+            del copy_data[key]
+
+        for key, value in updates.items():
+            copy_data[key] = value
+
+        return copy_data
+
+    def tearDown(self):
+        # if the authentication method has been mocked, it needs to be reset after test
+        if not self.authenticate:
+            AuthenticationMiddleware.process_request = self.original_auth_method
-- 
cgit 1.2.3-korg