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

tests/unit_tests/startup_scripts/test_startup_scripts.py

@@ -1,196 +0,0 @@
-import pytest
-import responses  # https://github.com/getsentry/responses
-
-from datacrunch.exceptions import APIException
-from datacrunch.startup_scripts.startup_scripts import StartupScriptsService, StartupScript
-
-INVALID_REQUEST = 'invalid_request'
-INVALID_REQUEST_MESSAGE = 'Your existence is invalid'
-
-SCRIPT_ID = 'deadc0de-a5d2-4972-ae4e-d429115d055b'
-SCRIPT_NAME = 'next episode of _____'
-SCRIPT_VALUE = 'this was not in the script!'
-
-script_ID_2 = 'beefbeef-a5d2-4972-ae4e-d429115d055b'
-
-PAYLOAD = [
-    {
-        'id': SCRIPT_ID,
-        'name': SCRIPT_NAME,
-        'script': SCRIPT_VALUE
-    }
-]
-
-
-class TestStartupScripts:
-
-    @pytest.fixture
-    def startup_script_service(self, http_client):
-        return StartupScriptsService(http_client)
-
-    @pytest.fixture
-    def endpoint(self, http_client):
-        return http_client._base_url + "/scripts"
-
-    def test_get_scripts(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.GET,
-            endpoint,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        scripts = startup_script_service.get()
-
-        # assert
-        assert type(scripts) == list
-        assert len(scripts) == 1
-        assert type(scripts[0]) == StartupScript
-        assert scripts[0].id == SCRIPT_ID
-        assert scripts[0].name == SCRIPT_NAME
-        assert scripts[0].script == SCRIPT_VALUE
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_get_script_by_id_successful(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/' + SCRIPT_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        script = startup_script_service.get_by_id(SCRIPT_ID)
-
-        # assert
-        assert type(script) == StartupScript
-        assert script.id == SCRIPT_ID
-        assert script.name == SCRIPT_NAME
-        assert script.script == SCRIPT_VALUE
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_get_script_by_id_failed(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/x'
-        responses.add(
-            responses.GET,
-            url,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            startup_script_service.get_by_id('x')
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_create_script_successful(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            endpoint,
-            body=SCRIPT_ID,
-            status=201
-        )
-
-        # act
-        script = startup_script_service.create(SCRIPT_NAME, SCRIPT_VALUE)
-
-        # assert
-        assert type(script) == StartupScript
-        assert script.id == SCRIPT_ID
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_create_script_failed(self, startup_script_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:
-            startup_script_service.create(SCRIPT_NAME, SCRIPT_VALUE)
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_delete_scripts_successful(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.DELETE,
-            endpoint,
-            status=200
-        )
-
-        # act
-        result = startup_script_service.delete([SCRIPT_ID, script_ID_2])
-
-        # assert
-        assert result is None
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_delete_scripts_failed(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.DELETE,
-            endpoint,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            startup_script_service.delete(['x'])
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_delete_script_by_id_successful(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/' + SCRIPT_ID
-        responses.add(
-            responses.DELETE,
-            url,
-            status=200
-        )
-
-        # act
-        result = startup_script_service.delete_by_id(SCRIPT_ID)
-
-        # assert
-        assert result is None
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_delete_script_by_id_failed(self, startup_script_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/x'
-        responses.add(
-            responses.DELETE,
-            url,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            startup_script_service.delete_by_id('x')
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-        assert responses.assert_call_count(url, 1) is True

tests/unit_tests/test_datacrunch.py

@@ -1,65 +0,0 @@
-import pytest
-import responses # https://github.com/getsentry/responses
-from datacrunch.datacrunch import DataCrunchClient
-from datacrunch.exceptions import APIException
-
-BASE_URL = "https://api-testing.datacrunch.io/v1"
-
-response_json = {
-    "access_token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJoZXkiOiJ5b3UgYWN1YWxseSBjaGVja2VkIHRoaXM_In0.0RjcdKQ1NJP9gbRyXITE6LFFLwKGzeeshuubnkkfkb8",
-    "token_type": "Bearer",
-    "expires_in": 3600,
-    "refresh_token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ3b3ciOiJhbmQgdGhpcyB0b28_In0.AC5gk-o-MOptUgrouEErlhr8WT3Hg_RR6px6A0I7ZEk",
-    "scope": "fullAccess"
-}
-
-class TestDataCrunchClient:
-
-    def test_client(self):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            BASE_URL + "/oauth2/token",
-            json=response_json,
-            status=200
-        )
-
-        # act
-        client = DataCrunchClient("XXXXXXXXXXXXXX", "XXXXXXXXXXXXXX", BASE_URL)
-
-        # assert
-        assert client.constants.base_url == BASE_URL
-
-    def test_client_with_default_base_url(self):
-        # arrange - add response mock
-        DEFAULT_BASE_URL = "https://api.datacrunch.io/v1"
-        responses.add(
-            responses.POST,
-            DEFAULT_BASE_URL + "/oauth2/token",
-            json=response_json,
-            status=200
-        )
-
-        # act
-        client = DataCrunchClient("XXXXXXXXXXXXXX", "XXXXXXXXXXXXXX")
-
-        # assert
-        assert client.constants.base_url == DEFAULT_BASE_URL
-
-    def test_invalid_client_credentials(self):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            BASE_URL + "/oauth2/token",
-            json={"code": "unauthorized_request", "message": "Invalid client id or client secret"},
-            status=401
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            DataCrunchClient("x", "y", BASE_URL)
-
-        # assert
-        assert excinfo.value.code == 'unauthorized_request'
-        assert excinfo.value.message == 'Invalid client id or client secret'
-        

tests/unit_tests/test_exceptions.py

@@ -1,33 +0,0 @@
-import pytest
-from datacrunch.exceptions import APIException
-
-ERROR_CODE = 'test_code'
-ERROR_MESSAGE = "test message"
-
-
-def test_api_exception_with_code():
-    # arrange
-    error_str = f'error code: {ERROR_CODE}\nmessage: {ERROR_MESSAGE}'
-
-    # act
-    with pytest.raises(APIException) as excinfo:
-        raise APIException(ERROR_CODE, ERROR_MESSAGE)
-
-    # assert
-        assert excinfo.value.code == ERROR_CODE
-        assert excinfo.value.message == ERROR_MESSAGE
-        assert excinfo.value.__str__() == error_str
-
-
-def test_api_exception_without_code():
-    # arrange
-    error_str = f'message: {ERROR_MESSAGE}'
-
-    # act
-    with pytest.raises(APIException) as excinfo:
-        raise APIException(None, ERROR_MESSAGE)
-
-    # assert
-        assert excinfo.value.code is None
-        assert excinfo.value.message == ERROR_MESSAGE
-        assert excinfo.value.__str__() == error_str

tests/unit_tests/volume_types/test_volume_types.py

@@ -1,50 +0,0 @@
-import responses  # https://github.com/getsentry/responses
-
-from datacrunch.volume_types.volume_types import VolumeTypesService, VolumeType
-from datacrunch.constants import VolumeTypes
-
-
-USD = "usd"
-NVMe_PRICE = 0.2
-HDD_PRICE = 0.05
-
-
-def test_volume_types(http_client):
-    responses.add(
-        responses.GET,
-        http_client._base_url + "/volume-types",
-        json=[
-            {
-                "type": VolumeTypes.NVMe,
-                "price": {
-                    "currency": USD,
-                    "price_per_month_per_gb": NVMe_PRICE
-                }
-            },
-            {
-                "type": VolumeTypes.HDD,
-                "price": {
-                    "currency": USD,
-                    "price_per_month_per_gb": HDD_PRICE
-                }
-            }
-        ],
-        status=200
-    )
-
-    volume_types_service = VolumeTypesService(http_client)
-
-    # act
-    volumes_types = volume_types_service.get()
-    nvme_type = volumes_types[0]
-    hdd_type = volumes_types[1]
-
-    # assert
-    assert type(volumes_types) == list
-    assert len(volumes_types) == 2
-    assert type(nvme_type) == VolumeType
-    assert type(hdd_type) == VolumeType
-    assert nvme_type.type == VolumeTypes.NVMe
-    assert nvme_type.price_per_month_per_gb == NVMe_PRICE
-    assert hdd_type.type == VolumeTypes.HDD
-    assert hdd_type.price_per_month_per_gb == HDD_PRICE