Exemplo n.º 1
0
    def test_initialize_options(self):
        """
        HTTP client and API options are set with `initialize` method.
        """
        initialize(api_key=API_KEY,
                   app_key=APP_KEY,
                   api_host=API_HOST,
                   proxies=FAKE_PROXY,
                   cacert=False)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.call_args()

        # Assert `requests` parameters
        self.assertIn('proxies', options)
        self.assertEqual(options['proxies'], FAKE_PROXY)

        self.assertIn('verify', options)
        self.assertEqual(options['verify'], False)

        # Arm the `requests` to raise
        self.arm_requests_to_raise()

        # No exception should be raised (mute=True by default)
        MyCreatable.create()

        # Repeat with mute to False
        initialize(api_key=API_KEY, mute=False)
        self.assertRaises(ApiError, MyCreatable.create)
Exemplo n.º 2
0
    def test_creatable(self):
        MyCreatable.create(mydata="val")
        self.request_called_with('POST', "host/api/v1/creatables", data={'mydata': "val"})

        MyCreatable.create(mydata="val", attach_host_name=True)
        self.request_called_with('POST', "host/api/v1/creatables",
                                 data={'mydata': "val", 'host': api._host_name})
Exemplo n.º 3
0
    def test_creatable(self):
        MyCreatable.create(mydata="val")
        self.request_called_with('POST', "host/api/v1/creatables", data={'mydata': "val"})

        MyCreatable.create(mydata="val", attach_host_name=True)
        self.request_called_with('POST', "host/api/v1/creatables",
                                 data={'mydata': "val", 'host': api._host_name})
Exemplo n.º 4
0
    def test_initialize_options(self):
        """
        HTTP client and API options are set with `initialize` method.
        """
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST,
                   proxies=FAKE_PROXY, cacert=False)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.request.call_args

        # Assert `requests` parameters
        self.assertIn('proxies', options)
        self.assertEquals(options['proxies'], FAKE_PROXY)

        self.assertIn('verify', options)
        self.assertEquals(options['verify'], False)

        # Arm the `requests` to raise
        self.arm_requests_to_raise()

        # No exception should be raised (mute=True by default)
        MyCreatable.create()

        # Repeat with mute to False
        initialize(api_key=API_KEY, mute=False)
        self.assertRaises(ApiError, MyCreatable.create)
Exemplo n.º 5
0
    def test_creatable(self):
        """
        Creatable resource logic.
        """
        MyCreatable.create(mydata="val")
        self.request_called_with('POST', API_HOST + "/api/v1/creatables", data={'mydata': "val"})

        MyCreatable.create(mydata="val", attach_host_name=True)
        self.request_called_with('POST', API_HOST + "/api/v1/creatables",
                                 data={'mydata': "val", 'host': api._host_name})
Exemplo n.º 6
0
    def test_no_initialization_fails(self, test='sisi'):
        assert_raises(ApiNotInitialized, MyCreatable.create)

        # No API key => only stats in statsd mode should work
        initialize()
        api._api_key = None
        assert_raises(ApiNotInitialized, MyCreatable.create)

        # Finally, initialize with an API key
        initialize(api_key=API_KEY, api_host=API_HOST)
        MyCreatable.create()
        assert self.request_mock.request.call_count == 1
Exemplo n.º 7
0
    def test_no_initialization_fails(self, test='sisi'):
        assert_raises(ApiNotInitialized, MyCreatable.create)

        # No API key => only stats in statsd mode should work
        initialize()
        api._api_key = None
        assert_raises(ApiNotInitialized, MyCreatable.create)

        # Finally, initialize with an API key
        initialize(api_key=API_KEY, api_host=API_HOST)
        MyCreatable.create()
        assert self.request_mock.request.call_count == 1
Exemplo n.º 8
0
    def test_no_initialization_fails(self):
        """
        Raise ApiNotInitialized exception when `initialize` has not ran or no API key was set.
        """
        self.assertRaises(ApiNotInitialized, MyCreatable.create)

        # No API key => only stats in statsd mode should work
        initialize()
        api._api_key = None
        self.assertRaises(ApiNotInitialized, MyCreatable.create)

        # Finally, initialize with an API key
        initialize(api_key=API_KEY, api_host=API_HOST)
        MyCreatable.create()
        self.assertEqual(self.request_mock.call_count(), 1)
Exemplo n.º 9
0
    def test_no_initialization_fails(self):
        """
        Raise ApiNotInitialized exception when `initialize` has not ran or no API key was set.
        """
        self.assertRaises(ApiNotInitialized, MyCreatable.create)

        # No API key => only stats in statsd mode should work
        initialize()
        api._api_key = None
        self.assertRaises(ApiNotInitialized, MyCreatable.create)

        # Finally, initialize with an API key
        initialize(api_key=API_KEY, api_host=API_HOST)
        MyCreatable.create()
        self.assertEquals(self.request_mock.request.call_count, 1)
Exemplo n.º 10
0
    def test_request_parameters(self):
        # Test API, application keys, API host and proxies
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST, proxies=FAKE_PROXY)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.request.call_args

        assert 'params' in options

        assert 'api_key' in options['params']
        assert options['params']['api_key'] == API_KEY
        assert 'application_key' in options['params']
        assert options['params']['application_key'] == APP_KEY

        assert 'proxies' in options
        assert options['proxies'] == FAKE_PROXY

        assert 'headers' in options
        assert options['headers'] == {'Content-Type': 'application/json'}
Exemplo n.º 11
0
    def test_request_parameters(self):
        """
        API parameters are set with `initialize` method.
        """
        # Test API, application keys, API host, and some HTTP client options
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.call_args()

        # Assert `requests` parameters
        self.assertIn('params', options)

        self.assertIn('headers', options)
        self.assertEqual(options['headers']['Content-Type'], 'application/json')
        self.assertEqual(options['headers']['DD-API-KEY'], API_KEY)
        self.assertEqual(options['headers']['DD-APPLICATION-KEY'], APP_KEY)
        assert "api_key" not in options['params']
        assert "application_key" not in options['params']
Exemplo n.º 12
0
    def test_errors_suppressed(self):
        """
        API `errors` field ApiError suppressed when specified
        """
        # Test API, application keys, API host, and some HTTP client options
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST)

        # Make a simple API call
        self.load_request_response(response_body='{"data": {}, "errors": ["foo error"]}')
        resp = MyCreatable.create(params={"suppress_response_errors_on_codes": [200]})
        self.assertNotIsInstance(resp, ApiError)
        self.assertDictEqual({"data": {}, "errors": ["foo error"]}, resp)
Exemplo n.º 13
0
    def test_request_parameters(self):
        # Test API, application keys, API host and proxies
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST, proxies=FAKE_PROXY)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.request.call_args

        assert 'params' in options

        assert 'api_key' in options['params']
        assert options['params']['api_key'] == API_KEY
        assert 'application_key' in options['params']
        assert options['params']['application_key'] == APP_KEY

        assert 'proxies' in options
        assert options['proxies'] == FAKE_PROXY

        assert 'headers' in options
        assert options['headers'] == {'Content-Type': 'application/json'}
Exemplo n.º 14
0
    def test_request_parameters(self):
        """
        API parameters are set with `initialize` method.
        """
        # Test API, application keys, API host, and some HTTP client options
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.request.call_args

        # Assert `requests` parameters
        self.assertIn('params', options)

        self.assertIn('api_key', options['params'])
        self.assertEquals(options['params']['api_key'], API_KEY)
        self.assertIn('application_key', options['params'])
        self.assertEquals(options['params']['application_key'], APP_KEY)

        self.assertIn('headers', options)
        self.assertEquals(options['headers'], {'Content-Type': 'application/json'})
Exemplo n.º 15
0
    def test_request_parameters(self):
        """
        API parameters are set with `initialize` method.
        """
        # Test API, application keys, API host, and some HTTP client options
        initialize(api_key=API_KEY, app_key=APP_KEY, api_host=API_HOST)

        # Make a simple API call
        MyCreatable.create()

        _, options = self.request_mock.request.call_args

        # Assert `requests` parameters
        self.assertIn('params', options)

        self.assertIn('api_key', options['params'])
        self.assertEquals(options['params']['api_key'], API_KEY)
        self.assertIn('application_key', options['params'])
        self.assertEquals(options['params']['application_key'], APP_KEY)

        self.assertIn('headers', options)
        self.assertEquals(options['headers'], {'Content-Type': 'application/json'})