datacrunch 1.15.0__py3-none-any.whl → 1.16.0__py3-none-any.whl

tests/integration_tests/test_locations.py

@@ -1,65 +0,0 @@
-import os
-import pytest
-from datacrunch.datacrunch import DataCrunchClient
-from datacrunch.constants import Locations
-
-IN_GITHUB_ACTIONS = os.getenv("GITHUB_ACTIONS") == "true"
-
-location_codes = [Locations.FIN_01, Locations.ICE_01]
-
-
-@pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test doesn't work in Github Actions.")
-@pytest.mark.withoutresponses
-class TestLocations():
-
-    def test_specific_instance_availability_in_specific_location(self, datacrunch_client: DataCrunchClient):
-        # call the instance availability endpoint, for a specific location
-        availability = datacrunch_client.instances.is_available(
-            'CPU.4V', location_code=Locations.FIN_01)
-
-        assert availability is not None
-        assert isinstance(availability, bool)
-
-    def test_all_availabilies_in_specific_location(self, datacrunch_client: DataCrunchClient):
-
-        # call the instance availability endpoint, for a specific location
-        availabilities = datacrunch_client.instances.get_availabilities(
-            location_code=Locations.FIN_01)
-
-        assert availabilities is not None
-        assert isinstance(availabilities, list)
-        assert len(availabilities) == 1
-        assert availabilities[0]['location_code'] in location_codes
-        assert isinstance(availabilities[0]['availabilities'], list)
-        assert len(availabilities[0]['availabilities']) > 0
-
-    def test_all_availabilites(self, datacrunch_client: DataCrunchClient):
-        # call the instance availability endpoint, for all locations
-        all_availabilities = datacrunch_client.instances.get_availabilities()
-
-        assert all_availabilities is not None
-        assert isinstance(all_availabilities, list)
-        assert len(all_availabilities) > 1
-        assert all_availabilities[0]['location_code'] in location_codes
-        assert all_availabilities[1]['location_code'] in location_codes
-        assert isinstance(all_availabilities[0]['availabilities'], list)
-        assert len(all_availabilities[0]['availabilities']) > 0
-
-    def test_get_all_locations(self, datacrunch_client: DataCrunchClient):
-        # call the locations endpoint
-        locations = datacrunch_client.locations.get()
-
-        assert locations is not None
-        assert isinstance(locations, list)
-
-        assert locations[0]['code'] in location_codes
-        assert locations[1]['code'] in location_codes
-        assert locations[0]['code'] != locations[1]['code']
-
-        assert locations[0]['name'] is not None
-        assert locations[1]['name'] is not None
-        assert locations[0]['name'] != locations[1]['name']
-
-        assert locations[0]['country_code'] is not None
-        assert locations[1]['country_code'] is not None
-        assert locations[0]['country_code'] != locations[1]['country_code']

tests/integration_tests/test_volumes.py

@@ -1,94 +0,0 @@
-import os
-import time
-import pytest
-from datacrunch.datacrunch import DataCrunchClient
-from datacrunch.constants import Locations, VolumeTypes, VolumeStatus
-
-IN_GITHUB_ACTIONS = os.getenv("GITHUB_ACTIONS") == "true"
-
-
-NVMe = VolumeTypes.NVMe
-
-
-@pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test doesn't work in Github Actions.")
-@pytest.mark.withoutresponses
-class TestVolumes():
-
-    def test_get_volumes_from_trash(self, datacrunch_client: DataCrunchClient):
-        # create new volume
-        volume = datacrunch_client.volumes.create(
-            type=NVMe, name="test_volume", size=100)
-
-        # delete volume
-        datacrunch_client.volumes.delete(volume.id)
-
-        # get volumes from trash
-        volumes = datacrunch_client.volumes.get_in_trash()
-
-        # assert volume is in trash
-        assert volume.id in [v.id for v in volumes]
-
-        # cleaning: permanently delete the volume
-        datacrunch_client.volumes.delete(volume.id, is_permanent=True)
-
-    def test_permanently_delete_detached_volumes(seld, datacrunch_client):
-        # create new volume
-        volume = datacrunch_client.volumes.create(
-            type=NVMe, name="test_volume", size=100)
-
-        # permanently delete the detached volume
-        datacrunch_client.volumes.delete(volume.id, is_permanent=True)
-
-        # sleep for 2 seconds
-        time.sleep(2)
-
-        # make sure the volume is not in trash
-        volumes = datacrunch_client.volumes.get_in_trash()
-
-        # assert volume is not in trash
-        assert volume.id not in [v.id for v in volumes]
-
-        # get the volume
-        volume = datacrunch_client.volumes.get_by_id(volume.id)
-
-        # assert volume status is deleted
-        assert volume.status == datacrunch_client.constants.volume_status.DELETED
-
-    def test_permanently_delete_a_deleted_volume_from_trash(self, datacrunch_client):
-        # create new volume
-        volume = datacrunch_client.volumes.create(
-            type=NVMe, name="test_volume", size=100)
-
-        # delete volume
-        datacrunch_client.volumes.delete(volume.id)
-
-        # sleep for 2 seconds
-        time.sleep(2)
-
-        # permanently delete the volume
-        datacrunch_client.volumes.delete(volume.id, is_permanent=True)
-
-        # get the volume
-        volume = datacrunch_client.volumes.get_by_id(volume.id)
-
-        # assert volume status is deleted
-        assert volume.status == datacrunch_client.constants.volume_status.DELETED
-
-        # make sure the volume is not in trash
-        volumes = datacrunch_client.volumes.get_in_trash()
-
-        # assert volume is not in trash
-        assert volume.id not in [v.id for v in volumes]
-
-    def test_create_volume(self, datacrunch_client):
-        # create new volume
-        volume = datacrunch_client.volumes.create(
-            type=NVMe, name="test_volume", size=100, location=Locations.FIN_01)
-
-        # assert volume is created
-        assert volume.id is not None
-        assert volume.location == Locations.FIN_01
-        assert volume.status == VolumeStatus.ORDERED or volume.status == VolumeStatus.DETACHED
-
-        # cleaning: delete volume
-        datacrunch_client.volumes.delete(volume.id, is_permanent=True)

tests/unit_tests/authentication/test_authentication.py

@@ -1,202 +0,0 @@
-import pytest
-import responses  # https://github.com/getsentry/responses
-from responses import matchers
-import time
-
-from datacrunch.exceptions import APIException
-from datacrunch.authentication.authentication import AuthenticationService
-
-INVALID_REQUEST = 'invalid_request'
-INVALID_REQUEST_MESSAGE = 'Your existence is invalid'
-
-BASE_URL = "https://api-testing.datacrunch.io/v1"
-CLIENT_ID = "0123456789xyz"
-CLIENT_SECRET = 'zyx987654321'
-
-ACCESS_TOKEN = 'access'
-REFRESH_TOKEN = 'refresh'
-SCOPE = 'fullAccess'
-TOKEN_TYPE = 'Bearer'
-EXPIRES_IN = 3600
-
-ACCESS_TOKEN2 = 'access2'
-REFRESH_TOKEN2 = 'refresh2'
-
-
-class TestAuthenticationService:
-
-    @pytest.fixture
-    def authentication_service(self):
-        return AuthenticationService(CLIENT_ID, CLIENT_SECRET, BASE_URL)
-
-    @pytest.fixture
-    def endpoint(self, http_client):
-        return http_client._base_url + "/oauth2/token"
-
-    def test_authenticate_successful(self, authentication_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={
-                'access_token': ACCESS_TOKEN,
-                'refresh_token': REFRESH_TOKEN,
-                'scope': SCOPE,
-                'token_type': TOKEN_TYPE,
-                'expires_in': EXPIRES_IN
-            },
-            status=200
-        )
-
-        # act
-        auth_data = authentication_service.authenticate()
-
-        # assert
-        assert type(auth_data) == dict
-        assert authentication_service._access_token == ACCESS_TOKEN
-        assert authentication_service._refresh_token == REFRESH_TOKEN
-        assert authentication_service._scope == SCOPE
-        assert authentication_service._token_type == TOKEN_TYPE
-        assert authentication_service._expires_at != None
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_authenticate_failed(self, authentication_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            authentication_service.authenticate()
-
-        x = responses.calls[0].request
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(endpoint, 1) is True
-        assert responses.calls[0].request.body == f'{{"grant_type": "client_credentials", "client_id": "{CLIENT_ID}", "client_secret": "{CLIENT_SECRET}"}}'.encode(
-        )
-
-    def test_refresh_successful(self, authentication_service, endpoint):
-        # add a response for the client credentials grant
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={
-                'access_token': ACCESS_TOKEN,
-                'refresh_token': REFRESH_TOKEN,
-                'scope': SCOPE,
-                'token_type': TOKEN_TYPE,
-                'expires_in': EXPIRES_IN
-            },
-            match=[matchers.json_params_matcher(
-                {"grant_type": "client_credentials", "client_id": CLIENT_ID, "client_secret": CLIENT_SECRET})],
-            status=200
-        )
-
-        # add another response for the refresh token grant
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={
-                'access_token': ACCESS_TOKEN2,
-                'refresh_token': REFRESH_TOKEN2,
-                'scope': SCOPE,
-                'token_type': TOKEN_TYPE,
-                'expires_in': EXPIRES_IN
-            },
-            match=[matchers.json_params_matcher(
-                {"grant_type": "refresh_token", "refresh_token": REFRESH_TOKEN})],
-            status=200
-        )
-
-        # act
-        auth_data = authentication_service.authenticate()  # authenticate first
-
-        # assert
-        assert type(auth_data) == dict
-        assert authentication_service._access_token == ACCESS_TOKEN
-        assert authentication_service._refresh_token == REFRESH_TOKEN
-        assert authentication_service._scope == SCOPE
-        assert authentication_service._token_type == TOKEN_TYPE
-        assert authentication_service._expires_at != None
-        assert responses.calls[0].request.body == f'{{"grant_type": "client_credentials", "client_id": "{CLIENT_ID}", "client_secret": "{CLIENT_SECRET}"}}'.encode(
-        )
-
-        auth_data2 = authentication_service.refresh()  # refresh
-
-        assert type(auth_data2) == dict
-        assert authentication_service._access_token == ACCESS_TOKEN2
-        assert authentication_service._refresh_token == REFRESH_TOKEN2
-        assert authentication_service._scope == SCOPE
-        assert authentication_service._token_type == TOKEN_TYPE
-        assert authentication_service._expires_at != None
-        assert responses.calls[1].request.body == f'{{"grant_type": "refresh_token", "refresh_token": "{REFRESH_TOKEN}"}}'.encode(
-        )
-        assert responses.assert_call_count(endpoint, 2) is True
-
-    def test_refresh_failed(self, authentication_service, endpoint):
-        # arrange - add responses mock
-        # first response for authentication - ok
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={
-                'access_token': ACCESS_TOKEN,
-                'refresh_token': REFRESH_TOKEN,
-                'scope': SCOPE,
-                'token_type': TOKEN_TYPE,
-                'expires_in': EXPIRES_IN
-            },
-            match=[matchers.json_params_matcher(
-                {"grant_type": "client_credentials", "client_id": CLIENT_ID, "client_secret": CLIENT_SECRET})],
-            status=200
-        )
-
-        # second response for the refresh - failed
-        responses.add(
-            responses.POST,
-            endpoint,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            match=[matchers.json_params_matcher(
-                {"grant_type": "refresh_token", "refresh_token": REFRESH_TOKEN})],
-            status=500
-        )
-
-        # act
-        authentication_service.authenticate()  # authenticate first
-
-        with pytest.raises(APIException) as excinfo:
-            authentication_service.refresh()
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(endpoint, 2) is True
-        assert responses.calls[0].request.body == f'{{"grant_type": "client_credentials", "client_id": "{CLIENT_ID}", "client_secret": "{CLIENT_SECRET}"}}'.encode(
-        )
-        assert responses.calls[1].request.body == f'{{"grant_type": "refresh_token", "refresh_token": "{REFRESH_TOKEN}"}}'.encode(
-        )
-
-    def test_is_expired(self, authentication_service, endpoint):
-        # arrange
-        current_time = time.time()
-        future_time = current_time + 3600
-
-        # act
-        # set the expired_at as current time
-        authentication_service._expires_at = current_time
-        is_expired_current = authentication_service.is_expired()
-
-        # set the expired_at as future time
-        authentication_service._expires_at = future_time
-        is_expired_future = authentication_service.is_expired()
-
-        # assert
-        assert is_expired_current == True
-        assert is_expired_future == False

tests/unit_tests/balance/test_balance.py

@@ -1,25 +0,0 @@
-import responses # https://github.com/getsentry/responses
-
-from datacrunch.balance.balance import BalanceService, Balance
-
-
-def test_balance(http_client):
-    # arrange - add response mock
-    responses.add(
-        responses.GET,
-        http_client._base_url + "/balance",
-        json={"amount": 50.5, "currency": "usd"},
-        status=200
-    )
-
-    balance_service = BalanceService(http_client)
-
-    # act
-    balance = balance_service.get()
-
-    # assert
-    assert type(balance) == Balance
-    assert type(balance.amount) == float
-    assert type(balance.currency) == str
-    assert balance.amount == 50.5
-    assert balance.currency == "usd"

tests/unit_tests/conftest.py

@@ -1,21 +0,0 @@
-import pytest
-from unittest.mock import Mock
-from datacrunch.http_client.http_client import HTTPClient
-
-
-BASE_URL = "https://api-testing.datacrunch.io/v1"
-ACCESS_TOKEN = "test-token"
-CLIENT_ID = "0123456789xyz"
-CLIENT_SECRET = "0123456789xyz"
-
-
-@pytest.fixture
-def http_client():
-    auth_service = Mock()
-    auth_service._access_token = ACCESS_TOKEN
-    auth_service.is_expired = Mock(return_value=True)
-    auth_service.refresh = Mock(return_value=None)
-    auth_service._client_id = CLIENT_ID
-    auth_service._client_secret = CLIENT_SECRET
-
-    return HTTPClient(auth_service, BASE_URL)

tests/unit_tests/containers/__init__.py

@@ -1 +0,0 @@
-