def setUp(self):
        settings.set('BASE_URL', 'http://localhost:9000')
        settings.set('REQUEST_CONNECTION_TIMEOUT', 0.5)
        self.client = ApiClient()

        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind(('', 9000))
示例#2
0
    def test_check_api_response_without_okta_error(self):
        """Ensure that an invalid response without an OKTA error gets handled properly"""

        api_client = ApiClient()
        try:
            api_client.check_api_response({}, 400)
            raise AssertionError("check_api_response should have raised an exception and didn't.")
        except ApiException as err:
            self.assertIn('Okta returned 400', str(err))
示例#3
0
    def test_non_json_response(self):
        """Ensure that the API client handles non-JSON responses gracefully"""

        imposter = self.mb.create_imposter('test_api_client/stubs/test_client_response.json')
        api_client = ApiClient(BASE_URL=self.mb.get_imposter_url(imposter), API_TOKEN='dummy')

        response, _ = api_client.get('/non-json')
        self.assertEqual('<html><body><h1>Testing!</h1></body></html>', response)
        response, _ = api_client.post('/non-json')
        self.assertEqual('<html><body><h1>Testing!</h1></body></html>', response)
    def test_check_api_response_without_okta_error(self):
        """Ensure that an invalid response without an OKTA error gets handled properly"""

        api_client = ApiClient()
        try:
            api_client.check_api_response({}, 400)
            raise AssertionError(
                "check_api_response should have raised an exception and didn't."
            )
        except ApiException as err:
            self.assertIn('Okta returned 400', str(err))
示例#5
0
    def test_check_api_response_no_causes(self):
        api_client = ApiClient()

        err_msg = {
            "errorSummary": "This is a short error message",
            "errorCauses": []
        }
        try:
            api_client.check_api_response(err_msg, 400)
            raise AssertionError("check_api_response should have raised an exception and didn't.")
        except ApiException as err:
            self.assertIn(err_msg['errorSummary'], str(err))
示例#6
0
    def test_check_status(self):
        imposter = self.mb.create_imposter('test_api_client/stubs/test_client_response.json')
        imposter_url = self.mb.get_imposter_url(imposter)
        api_client = ApiClient(BASE_URL=imposter_url)
        status = api_client.check_status()

        self.assertTrue(status['okta']['is_available'])
        self.assertEqual(1199, status['okta']['calls_remaining'])
        self.assertTrue(isinstance(status['okta']['time_of_reset'], datetime))

        # this is the fail case
        status = api_client.check_status()
        self.assertFalse(status['okta']['is_available'])
    def test_check_api_response_no_causes(self):
        api_client = ApiClient()

        err_msg = {
            "errorSummary": "This is a short error message",
            "errorCauses": []
        }
        try:
            api_client.check_api_response(err_msg, 400)
            raise AssertionError(
                "check_api_response should have raised an exception and didn't."
            )
        except ApiException as err:
            self.assertIn(err_msg['errorSummary'], str(err))
    def test_check_status(self):
        imposter = self.mb.create_imposter(
            'test_api_client/stubs/test_client_response.json')
        imposter_url = self.mb.get_imposter_url(imposter)
        api_client = ApiClient(BASE_URL=imposter_url)
        status = api_client.check_status()

        self.assertTrue(status['okta']['is_available'])
        self.assertEqual(1199, status['okta']['calls_remaining'])
        self.assertTrue(isinstance(status['okta']['time_of_reset'], datetime))

        # this is the fail case
        status = api_client.check_status()
        self.assertFalse(status['okta']['is_available'])
    def test_non_json_response(self):
        """Ensure that the API client handles non-JSON responses gracefully"""

        imposter = self.mb.create_imposter(
            'test_api_client/stubs/test_client_response.json')
        api_client = ApiClient(BASE_URL=self.mb.get_imposter_url(imposter),
                               API_TOKEN='dummy')

        response, _ = api_client.get('/non-json')
        self.assertEqual('<html><body><h1>Testing!</h1></body></html>',
                         response)
        response, _ = api_client.post('/non-json')
        self.assertEqual('<html><body><h1>Testing!</h1></body></html>',
                         response)
示例#10
0
    def setUp(self):
        settings.set('BASE_URL', 'http://localhost:9000')
        settings.set('REQUEST_CONNECTION_TIMEOUT', 0.5)
        self.client = ApiClient()

        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind(('', 9000))
示例#11
0
    def test_no_response(self):
        """Ensure that API client handles no response from the server gracefully"""
        api_client = ApiClient(BASE_URL='http://example.com', API_TOKEN='dummy')

        # fake out the requests.get method
        requests.get = MagicMock(return_value=None)
        response, status_code = api_client.get('/')
        self.assertIsNone(response)
        self.assertIsNone(status_code)

        # fake out the requests.post method
        requests.post = MagicMock(return_value=None)
        response, status_code = api_client.post('/')
        self.assertIsNone(response)
        self.assertIsNone(status_code)

        # reload requests so that the methods are back to normal
        reload(requests)
示例#12
0
    def test_no_response(self):
        """Ensure that API client handles no response from the server gracefully"""
        api_client = ApiClient(BASE_URL='http://example.com',
                               API_TOKEN='dummy')

        # fake out the requests.get method
        requests.get = MagicMock(return_value=None)
        response, status_code = api_client.get('/')
        self.assertIsNone(response)
        self.assertIsNone(status_code)

        # fake out the requests.post method
        requests.post = MagicMock(return_value=None)
        response, status_code = api_client.post('/')
        self.assertIsNone(response)
        self.assertIsNone(status_code)

        # reload requests so that the methods are back to normal
        reload(requests)
示例#13
0
 def setUp(self):
     self.mb = MountebankProcess()
     api_client = ApiClient()
     self.broker = UserBroker(api_client)
示例#14
0
 def setup_imposter(self, file_name):
     file_path = 'test_user_broker/stubs/' + file_name
     imposter = self.mb.create_imposter(file_path)
     self.broker._api_client = ApiClient(
         BASE_URL=self.mb.get_imposter_url(imposter), API_TOKEN='1')
 def setUp(self):
     self.api_client = ApiClient()
     self.auth_client = OktaAuthClient(OktaFactors(self.api_client))
     self.mb = MountebankProcess()
示例#16
0
 def setup_imposter(self, file_name):
     imposter = self.mb.create_imposter('test_auth_client/stubs/' + file_name)
     api_client = ApiClient(BASE_URL=self.mb.get_imposter_url(imposter))
     return OktaAuthClient(OktaFactors(api_client))
示例#17
0
 def setUp(self):
     self.api_client = ApiClient(API_TOKEN='jrr-tolkien')
     self.auth_client = OktaAuthClient(OktaFactors(self.api_client))
     self.mb = MountebankProcess()
示例#18
0
class TestTimeouts(unittest.TestCase):
    # these tests will ONLY work on UNIX-based systems because of the signal package. So no Windows for us.

    def setUp(self):
        settings.set('BASE_URL', 'http://localhost:9000')
        settings.set('REQUEST_CONNECTION_TIMEOUT', 0.5)
        self.client = ApiClient()

        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind(('', 9000))

    def tearDown(self):
        self.socket.close()
        reload(theoktany.conf)

    @staticmethod
    def timeout_handler(signum, frame):
        raise AssertionError('Connection did not time out!')

    def test_get_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3*settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.get('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(
            test_end-test_start, 1.5*settings.get('REQUEST_CONNECTION_TIMEOUT'),
            msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)

    def test_post_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3*settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.post('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(
            test_end-test_start, 1.5*settings.get('REQUEST_CONNECTION_TIMEOUT'),
            msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)

    def test_delete_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3*settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.delete('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(
            test_end-test_start, 1.5*settings.get('REQUEST_CONNECTION_TIMEOUT'),
            msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)
示例#19
0
class TestTimeouts(unittest.TestCase):
    # these tests will ONLY work on UNIX-based systems because of the signal package. So no Windows for us.

    def setUp(self):
        settings.set('BASE_URL', 'http://localhost:9000')
        settings.set('REQUEST_CONNECTION_TIMEOUT', 0.5)
        self.client = ApiClient()

        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind(('', 9000))

    def tearDown(self):
        self.socket.close()
        reload(theoktany.conf)

    @staticmethod
    def timeout_handler(signum, frame):
        raise AssertionError('Connection did not time out!')

    def test_get_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3 * settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.get('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(test_end - test_start,
                        1.5 * settings.get('REQUEST_CONNECTION_TIMEOUT'),
                        msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)

    def test_post_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3 * settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.post('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(test_end - test_start,
                        1.5 * settings.get('REQUEST_CONNECTION_TIMEOUT'),
                        msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)

    def test_delete_timeout(self):
        # set a timeout in case something goes horribly wrong
        signal.signal(signal.SIGALRM, self.timeout_handler)
        signal.alarm(int(3 * settings.get('REQUEST_CONNECTION_TIMEOUT')))

        test_start = time.time()
        response = self.client.delete('/')
        test_end = time.time()

        # stop listening for the signal
        signal.signal(signal.SIGALRM, lambda *args: None)

        # end-start is going to be a bit more than the timeout
        self.assertLess(test_end - test_start,
                        1.5 * settings.get('REQUEST_CONNECTION_TIMEOUT'),
                        msg='Connection timeout did not obey setting.')

        self.assertEqual((None, None), response)