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

tests/unit_tests/http_client/test_http_client.py

@@ -1,193 +0,0 @@
-import pytest
-import responses  # https://github.com/getsentry/responses
-from unittest.mock import Mock
-from datacrunch.exceptions import APIException
-
-INVALID_REQUEST = 'invalid_request'
-INVALID_REQUEST_MESSAGE = 'Your existence is invalid'
-
-UNAUTHORIZED_REQUEST = 'unauthorized_request'
-UNAUTHORIZED_REQUEST_MESSAGE = 'Access token is missing or invalid'
-
-
-class TestHttpClient:
-    def test_add_base_url(self, http_client):
-        # arrange
-        path = "/test"
-        base = http_client._base_url
-
-        # act
-        url = http_client._add_base_url(path)
-
-        # assert
-        assert base == http_client._base_url
-        assert url == base + path
-
-    def test_generate_bearer_header(self, http_client):
-        bearer_string = http_client._generate_bearer_header()
-        access_token = http_client._auth_service._access_token
-
-        assert type(bearer_string) == str
-        assert bearer_string == f'Bearer {access_token}'
-
-    def test_generate_user_agent(self, http_client):
-        # arrange
-        version = http_client._version
-        client_id_truncated = http_client._auth_service._client_id[0:10]
-
-        # act
-        user_agent_string = http_client._generate_user_agent()
-
-        # assert
-        assert type(user_agent_string) == str
-        assert user_agent_string == f'datacrunch-python-v{version}-{client_id_truncated}'
-
-    def test_generate_headers(self, http_client):
-        # arrange / act
-        headers = http_client._generate_headers()
-        authorization_string = http_client._generate_bearer_header()
-        user_agent_string = http_client._generate_user_agent()
-
-        # assert
-        assert type(headers) == dict
-        assert type(headers['Content-Type']) == str
-        assert type(headers['Authorization']) == str
-        assert type(headers['User-Agent']) == str
-        assert headers['Content-Type'] == 'application/json'
-        assert headers['Authorization'] == authorization_string
-        assert headers['User-Agent'] == user_agent_string
-
-    def test_refresh_token_if_expired_refresh_successful(self, http_client):
-        # act
-        http_client._refresh_token_if_expired()
-
-        # assert
-        http_client._auth_service.refresh.assert_called_once()
-        http_client._auth_service.is_expired.assert_called()
-        http_client._auth_service.authenticate.assert_called_once()
-
-    def test_refresh_token_if_expired_refresh_failed(self, http_client):
-        # arrange - make refresh raise an exception
-        http_client._auth_service.refresh = Mock(side_effect=Exception())
-
-        # act
-        http_client._refresh_token_if_expired()
-
-        # assert
-        http_client._auth_service.refresh.assert_called_once()
-        http_client._auth_service.is_expired.assert_called()
-        assert http_client._auth_service.authenticate.call_count == 2
-
-    def test_get_successful(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.GET,
-            url=(http_client._base_url + '/test'),
-            status=200, body='{}',
-            content_type='application/json'
-        )
-
-        # act
-        response = http_client.get('/test')
-
-        # assert
-        assert response.ok is True
-        assert response.status_code == 200
-        assert response.text == '{}'
-        assert response.headers['Content-Type'] == 'application/json'
-        assert response.json() == {}
-        assert responses.assert_call_count(http_client._base_url + '/test', 1) is True
-
-    def test_post_successful(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.POST,
-            url=(http_client._base_url + '/test'),
-            status=200, body='{}',
-            content_type='application/json'
-        )
-
-        # act
-        response = http_client.post('/test', params={})
-
-        # assert
-        assert response.ok is True
-        assert response.status_code == 200
-        assert response.text == '{}'
-        assert response.headers['Content-Type'] == 'application/json'
-        assert response.json() == {}
-        assert responses.assert_call_count(http_client._base_url + '/test', 1) is True
-
-    def test_delete_successful(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.DELETE,
-            url=(http_client._base_url + '/test'),
-            status=200,
-            content_type='application/json'
-        )
-
-        # act
-        response = http_client.delete('/test')
-
-        # assert
-        assert response.ok is True
-        assert response.status_code == 200
-        assert response.headers['Content-Type'] == 'application/json'
-        assert responses.assert_call_count(http_client._base_url + '/test', 1) is True
-
-    def test_get_failed(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.GET,
-            url=(http_client._base_url + '/test'),
-            status=401,
-            json={'code': UNAUTHORIZED_REQUEST, 'message': UNAUTHORIZED_REQUEST_MESSAGE},
-            content_type='application/json'
-        )
-        error_str = f'error code: {UNAUTHORIZED_REQUEST}\nmessage: {UNAUTHORIZED_REQUEST_MESSAGE}'
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            http_client.get('/test')
-
-        # assert
-        assert excinfo.value.code == UNAUTHORIZED_REQUEST
-        assert excinfo.value.message == UNAUTHORIZED_REQUEST_MESSAGE
-        assert excinfo.value.__str__() == error_str
-
-    def test_post_failed(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.POST,
-            url=(http_client._base_url + '/test'),
-            status=400,
-            json={'code': INVALID_REQUEST, 'message': INVALID_REQUEST_MESSAGE},
-            content_type='application/json'
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            http_client.post('/test', json={'data': '42'})
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
-
-    def test_delete_failed(self, http_client):
-        # arrange - add response mock
-        responses.add(
-            method=responses.DELETE,
-            url=(http_client._base_url + '/test'),
-            status=400,
-            json={'code': INVALID_REQUEST, 'message': INVALID_REQUEST_MESSAGE},
-            content_type='application/json'
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            http_client.delete('/test', json={'data': '42'})
-
-        # assert
-        assert excinfo.value.code == INVALID_REQUEST
-        assert excinfo.value.message == INVALID_REQUEST_MESSAGE

tests/unit_tests/images/test_images.py

@@ -1,41 +0,0 @@
-import responses # https://github.com/getsentry/responses
-
-from datacrunch.images.images import ImagesService, Image
-
-
-def test_images(http_client):
-    # arrange - add response mock
-    responses.add(
-        responses.GET,
-        http_client._base_url + "/images",
-        json=[
-            {
-                "id": "0888da25-bb0d-41cc-a191-dccae45d96fd",
-                "name": "Ubuntu 20.04 + CUDA 11.0",
-                "details": [
-                    "Ubuntu 20.04",
-                    "CUDA 11.0"
-                ],
-                "image_type": "ubuntu-20.04-cuda-11.0"
-            }
-        ],
-        status=200
-    )
-
-    image_service = ImagesService(http_client)
-
-    # act
-    images = image_service.get()
-
-    # assert
-    assert type(images) == list
-    assert len(images) == 1
-    assert type(images[0]) == Image
-    assert images[0].id == '0888da25-bb0d-41cc-a191-dccae45d96fd'
-    assert images[0].name == 'Ubuntu 20.04 + CUDA 11.0'
-    assert images[0].image_type == 'ubuntu-20.04-cuda-11.0'
-    assert type(images[0].details) == list
-    assert images[0].details[0] == "Ubuntu 20.04"
-    assert images[0].details[1] == "CUDA 11.0"
-    assert type(images[0].__str__()) == str
-    

tests/unit_tests/instance_types/test_instance_types.py

@@ -1,87 +0,0 @@
-import responses  # https://github.com/getsentry/responses
-
-from datacrunch.instance_types.instance_types import InstanceTypesService, InstanceType
-
-TYPE_ID = "01cf5dc1-a5d2-4972-ae4e-d429115d055b"
-CPU_DESCRIPTION = "48 CPU 3.5GHz"
-NUMBER_OF_CORES = 48
-GPU_DESCRIPTION = "8x NVidia Tesla V100"
-NUMBER_OF_GPUS = 8
-MEMORY_DESCRIPTION = "192GB RAM"
-MEMORY_SIZE = 192
-GPU_MEMORY_DESCRIPTION = "128GB VRAM"
-GPU_MEMORY_SIZE = 128
-STORAGE_DESCRIPTION = "1800GB NVME"
-STORAGE_SIZE = 1800
-INSTANCE_TYPE_DESCRIPTION = "Dedicated Bare metal Server"
-PRICE_PER_HOUR = 5.0
-SPOT_PRICE_PER_HOUR = 2.5
-INSTANCE_TYPE = "8V100.48M"
-
-
-def test_instance_types(http_client):
-    # arrange - add response mock
-    responses.add(
-        responses.GET,
-        http_client._base_url + "/instance-types",
-        json=[
-            {
-                "id": TYPE_ID,
-                "cpu": {
-                    "description": CPU_DESCRIPTION,
-                    "number_of_cores": NUMBER_OF_CORES
-                },
-                "gpu": {
-                    "description": GPU_DESCRIPTION,
-                    "number_of_gpus": NUMBER_OF_GPUS
-                },
-                "memory": {
-                    "description": MEMORY_DESCRIPTION,
-                    "size_in_gigabytes": MEMORY_SIZE
-                },
-                "gpu_memory": {
-                    "description": GPU_MEMORY_DESCRIPTION,
-                    "size_in_gigabytes": GPU_MEMORY_SIZE
-                },
-                "storage": {
-                    "description": STORAGE_DESCRIPTION,
-                    "size_in_gigabytes": STORAGE_SIZE
-                },
-                "description": INSTANCE_TYPE_DESCRIPTION,
-                "price_per_hour": "5.00",
-                "spot_price": "2.50",
-                "instance_type": INSTANCE_TYPE
-            }
-        ],
-        status=200
-    )
-
-    instance_types_service = InstanceTypesService(http_client)
-
-    # act
-    instance_types = instance_types_service.get()
-    instance_type = instance_types[0]
-
-    # assert
-    assert type(instance_types) == list
-    assert len(instance_types) == 1
-    assert type(instance_type) == InstanceType
-    assert instance_type.id == TYPE_ID
-    assert instance_type.description == INSTANCE_TYPE_DESCRIPTION
-    assert instance_type.price_per_hour == PRICE_PER_HOUR
-    assert instance_type.spot_price_per_hour == SPOT_PRICE_PER_HOUR
-    assert instance_type.instance_type == INSTANCE_TYPE
-    assert type(instance_type.cpu) == dict
-    assert type(instance_type.gpu) == dict
-    assert type(instance_type.memory) == dict
-    assert type(instance_type.storage) == dict
-    assert instance_type.cpu['description'] == CPU_DESCRIPTION
-    assert instance_type.gpu['description'] == GPU_DESCRIPTION
-    assert instance_type.memory['description'] == MEMORY_DESCRIPTION
-    assert instance_type.gpu_memory['description'] == GPU_MEMORY_DESCRIPTION
-    assert instance_type.storage['description'] == STORAGE_DESCRIPTION
-    assert instance_type.cpu['number_of_cores'] == NUMBER_OF_CORES
-    assert instance_type.gpu['number_of_gpus'] == NUMBER_OF_GPUS
-    assert instance_type.memory['size_in_gigabytes'] == MEMORY_SIZE
-    assert instance_type.gpu_memory['size_in_gigabytes'] == GPU_MEMORY_SIZE
-    assert instance_type.storage['size_in_gigabytes'] == STORAGE_SIZE