class TestGetURI(unittest.TestCase):
data = get_test_data()
def setUp(self):
self.base_api = BaseAPI()
self.base_api.set_connection(
user_name="test_user", password="******", end_point="http://api.test.com", session_verify=False
)
def test_get_uri_with_multiple_resaults(self, mock):
headers = self.data["good-header"]
response_data = self.data["good-content"]
self.base_api.set_connection(
user_name="test", password="******", end_point="http:://test.com", session_verify=False
)
mock.get(requests_mock.ANY, json=response_data, headers=headers)
test = self.base_api.get_uri(search_string="Get/Delete Access Locks", search_spec="test")
self.assertEquals(test["Get/Delete Access Locks"], "/api/access_lock")
self.assertEquals(test["Get/Update/Add/Delete User Accounts"], "/api/account")
def test_get_uri_with_single_resaults(self, mock):
headers = self.data["good-header"]
response_data = self.data["get_uri_with_single_resaults"]
self.base_api.set_connection(
user_name="test", password="******", end_point="http:://test.com", session_verify=False
)
mock.get(requests_mock.ANY, json=response_data, headers=headers)
test = self.base_api.get_uri(search_string="Get/Delete Access Locks", search_spec="test")
self.assertEquals(test["Get/Delete Access Locks"], "/api/access_lock")
class TestGet(unittest.TestCase):
data = get_test_data()
def setUp(self):
self.base_api = BaseAPI()
self.base_api.set_connection(
user_name="test_user", password="******", end_point="http://api.test.com", session_verify=False
)
def test__perform_request_returns_requests_object(self, mock):
headers = self.data["good-header"]
response_data = self.data["good-content"]
mock.get(requests_mock.ANY, json=response_data, headers=headers)
test = self.base_api._perform_request(self.base_api.end_point)
self.assertIsInstance(test, requests.models.Response)
class TestSetConnection(unittest.TestCase):
data = get_test_data()
def setUp(self):
self.base_api = BaseAPI()
self.base_api.set_connection(
user_name="test_user", password="******", end_point="http://api.test.com", session_verify=False
)
def test_set_connection(self, mock):
self.assertEqual("test_user", self.base_api.user_name)
self.assertEqual("test_password", self.base_api.password)
self.assertEqual("http://api.test.com", self.base_api.end_point)
def test_set_connection_with_bad_values(self, mock):
self.assertNotEqual("user", self.base_api.user_name)
self.assertNotEqual("password", self.base_api.password)
self.assertNotEqual("http://test.com", self.base_api.end_point)
def setUp(self):
self.base_api = BaseAPI()
self.base_api.set_connection(
user_name="test_user", password="******", end_point="http://api.test.com", session_verify=False
)
