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

datacrunch-1.15.0.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2020 DataCrunch Oy
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

datacrunch-1.15.0.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-datacrunch
-tests

tests/integration_tests/conftest.py

@@ -1,20 +0,0 @@
-import os
-import pytest
-from dotenv import load_dotenv
-from datacrunch.datacrunch import DataCrunchClient
-
-"""
-Make sure to run the server and the account has enough balance before running the tests
-"""
-
-BASE_URL = "http://localhost:3010/v1"
-
-# Load env variables, make sure there's an env file with valid client credentials
-load_dotenv()
-CLIENT_SECRET = os.getenv('DATACRUNCH_CLIENT_SECRET')
-CLIENT_ID = os.getenv('DATACRUNCH_CLIENT_ID')
-
-
-@pytest.fixture
-def datacrunch_client():
-    return DataCrunchClient(CLIENT_ID, CLIENT_SECRET, BASE_URL)

tests/integration_tests/test_instances.py

@@ -1,36 +0,0 @@
-import os
-import pytest
-from datacrunch.datacrunch import DataCrunchClient
-from datacrunch.constants import Locations
-
-IN_GITHUB_ACTIONS = os.getenv("GITHUB_ACTIONS") == "true"
-
-
-@pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test doesn't work in Github Actions.")
-@pytest.mark.withoutresponses
-class TestInstances():
-
-    def test_create_instance(self, datacrunch_client: DataCrunchClient):
-        # get ssh key
-        ssh_key = datacrunch_client.ssh_keys.get()[0]
-
-        # create instance
-        instance = datacrunch_client.instances.create(
-            hostname="test-instance",
-            location=Locations.FIN_01,
-            instance_type='CPU.4V',
-            description="test instance",
-            image="ubuntu-18.04",
-            ssh_key_ids=[ssh_key.id])
-
-        # assert instance is created
-        assert instance.id is not None
-        assert instance.status == datacrunch_client.constants.instance_status.PROVISIONING
-
-        # delete instance
-        datacrunch_client.instances.action(instance.id, "delete")
-
-        # permanently delete all volumes in trash
-        trash = datacrunch_client.volumes.get_in_trash()
-        for volume in trash:
-            datacrunch_client.volumes.delete(volume.id, is_permanent=True)

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 @@
-