missing tests

Author: tsolakoua

specs/client/test_access_token.py

@@ -0,0 +1,53 @@
+import pytest
+import mock
+from amadeus import Client
+from amadeus.client.access_token import AccessToken
+
+
+class TokenResponse:
+    def __init__(self, result):
+        self.result = result
+
+
+@pytest.fixture
+def self():
+    self.request = mock.MagicMock(
+        return_value=TokenResponse({'access_token': 'abc', 'expires_in': 1799}))
+    self.client = mock.MagicMock(Client)
+    self.client._unauthenticated_request = self.request
+    self.client.client_id = '123'
+    self.client.client_secret = '234'
+    self.access_token = AccessToken(self.client)
+    return self
+
+
+def test_bearer_token(self):
+    token = self.access_token._bearer_token()
+    assert token == 'Bearer abc'
+    self.client._unauthenticated_request.assert_called_with(
+        'POST', '/v1/security/oauth2/token',
+        {
+            'grant_type': 'client_credentials',
+            'client_id': '123',
+            'client_secret': '234'
+        })
+
+
+def test_cached_token_valid(self):
+    access_token = self.access_token
+    token = access_token._bearer_token()
+    assert token == 'Bearer abc'
+    self.client._unauthenticated_request.assert_called_once()
+    token2 = access_token._bearer_token()
+    assert token2 == 'Bearer abc'
+    self.client._unauthenticated_request.assert_called_once()
+
+
+def test_cached_token_expired(self):
+    access_token = self.access_token
+    token = access_token._bearer_token()
+    assert token == 'Bearer abc'
+    access_token.expires_at = 0
+    token2 = access_token._bearer_token()
+    assert token2 == 'Bearer abc'
+    self.client._unauthenticated_request.call_count == 2

specs/client/test_request.py

@@ -0,0 +1,114 @@
+import pytest
+from amadeus import Request
+from urllib.request import Request as HTTPRequest
+
+
+@pytest.fixture
+def self():
+    self.host = 'example.com'
+    self.verb = 'GET'
+    self.path = '/foo/bar'
+    self.params = {'foo': 'bar'}
+    self.bearer_token = 'Bearer 123'
+    self.client_version = '1.2.3'
+    self.lang_version = '2.3.4'
+    self.app_id = 'amadeus-cli'
+    self.app_version = '3.4.5'
+    self.ssl = True
+    self.port = 443
+    self.request = Request({
+        'host': self.host,
+        'verb': self.verb,
+        'path': self.path,
+        'params': self.params,
+        'bearer_token': self.bearer_token,
+        'client_version': self.client_version,
+        'language_version': self.lang_version,
+        'app_id': self.app_id,
+        'app_version': self.app_version,
+        'ssl': self.ssl,
+        'port': self.port
+    })
+    return self
+
+
+def test_init(self):
+    assert self.request.host == self.host
+    assert self.request.port == self.port
+    assert self.request.ssl is True
+    assert self.request.scheme == 'https'
+    assert self.request.verb == self.verb
+    assert self.request.path == self.path
+    assert self.request.params == self.params
+    assert self.request.bearer_token == self.bearer_token
+    assert self.request.client_version == self.client_version
+    assert self.request.language_version == self.lang_version
+    assert self.request.app_id == self.app_id
+    assert self.request.app_version == self.app_version
+
+
+def test_http_request(self):
+    assert isinstance(self.request.http_request, HTTPRequest)
+    assert self.request.http_request.data is None
+    assert self.request.http_request.get_header('Content-Type', None) is None
+    assert self.request.http_request.get_header(
+        'Authorization') == self.bearer_token
+    assert self.request.http_request.get_header(
+        'Accept') == 'application/json, application/vnd.amadeus+json'
+    assert self.request.http_request.get_header(
+        'User-agent') == 'amadeus-python/1.2.3 python/2.3.4 amadeus-cli/3.4.5'
+
+    self.request = Request({
+                'host': self.host,
+                'verb': 'POST',
+                'path': self.path,
+                'params': self.params,
+                'bearer_token': self.bearer_token,
+                'client_version': self.client_version,
+                'language_version': self.lang_version,
+                'app_id': self.app_id,
+                'app_version': self.app_version,
+                'port': self.port,
+                'ssl': self.ssl
+    })
+    assert isinstance(self.request.http_request, HTTPRequest)
+    url = self.request.http_request.get_full_url()
+    assert url == 'https://example.com/foo/bar'
+    assert self.request.http_request.data == b'{\"foo\": \"bar\"}'
+
+
+def test_x_http_method_override(self):
+    for path in Request.list_httpoverride:
+        self.request = Request({
+                    'host': self.host,
+                    'verb': 'POST',
+                    'path': path,
+                    'params': self.params,
+                    'bearer_token': self.bearer_token,
+                    'client_version': self.client_version,
+                    'language_version': self.lang_version,
+                    'app_id': self.app_id,
+                    'app_version': self.app_version,
+                    'port': self.port,
+                    'ssl': self.ssl
+                })
+        assert isinstance(self.request.http_request, HTTPRequest)
+        assert self.request.headers['X-HTTP-Method-Override'] == 'GET'
+
+
+def test_custom_scheme_and_port(self):
+    self.request = Request({
+                'host': self.host,
+                'verb': 'POST',
+                'path': self.path,
+                'params': self.params,
+                'bearer_token': self.bearer_token,
+                'client_version': self.client_version,
+                'language_version': self.lang_version,
+                'app_id': self.app_id,
+                'app_version': self.app_version,
+                'port': 8080,
+                'ssl': False
+            })
+    url = self.request.http_request.get_full_url()
+    assert url == 'http://example.com:8080/foo/bar'