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

tests/unit_tests/instances/test_instances.py

@@ -1,483 +0,0 @@
-import pytest
-import responses  # https://github.com/getsentry/responses
-
-from datacrunch.exceptions import APIException
-from datacrunch.instances.instances import InstancesService, Instance
-from datacrunch.constants import Actions, ErrorCodes, Locations
-
-INVALID_REQUEST = ErrorCodes.INVALID_REQUEST
-INVALID_REQUEST_MESSAGE = 'Your existence is invalid'
-
-INSTANCE_ID = 'deadc0de-a5d2-4972-ae4e-d429115d055b'
-SSH_KEY_ID = '12345dc1-a5d2-4972-ae4e-d429115d055b'
-OS_VOLUME_ID = '46fc0247-8f65-4d8a-ad73-852a8b3dc1d3'
-
-INSTANCE_TYPE = "1V100.6V"
-INSTANCE_IMAGE = "ubuntu-24.04-cuda-12.8-open-docker"
-INSTANCE_HOSTNAME = "I'll be your host for today"
-INSTANCE_DESCRIPTION = "hope you enjoy your GPU"
-INSTANCE_STATUS = 'running'
-INSTANCE_PRICE_PER_HOUR = 0.60
-INSTANCE_LOCATION = Locations.FIN_01
-INSTANCE_IP = '1.2.3.4'
-INSTANCE_CREATED_AT = "whatchalookingatboy?"
-INSTANCE_OS_VOLUME = {"name": "os volume", "size": 50}
-
-PAYLOAD = [
-    {
-        "created_at": INSTANCE_CREATED_AT,
-        "status": INSTANCE_STATUS,
-        "ip": INSTANCE_IP,
-        "cpu": {
-            "description": "super-duper-cpu",
-            "number_of_cores": 6
-        },
-        "gpu": {
-            "description": "super-duper-gpu",
-            "number_of_gpus": 1
-        },
-        "memory": {
-            "description": "super-duper-memory",
-            "size_in_gigabytes": 32
-        },
-        "gpu_memory": {
-            "description": "super-duper-memory",
-            "size_in_gigabytes": 20
-        },
-        "storage": {
-            "description": "super-duper-storage",
-            "size_in_gigabytes": 320
-        },
-        "hostname": INSTANCE_HOSTNAME,
-        "description": INSTANCE_DESCRIPTION,
-        "location": INSTANCE_LOCATION,
-        "price_per_hour": INSTANCE_PRICE_PER_HOUR,
-        "instance_type": INSTANCE_TYPE,
-        "image": INSTANCE_IMAGE,
-        "id": INSTANCE_ID,
-        "ssh_key_ids": [SSH_KEY_ID],
-        "os_volume_id": OS_VOLUME_ID
-    }
-]
-
-PAYLOAD_SPOT = PAYLOAD
-PAYLOAD_SPOT[0]["is_spot"] = True
-
-
-class TestInstancesService:
-    @pytest.fixture
-    def instances_service(self, http_client):
-        return InstancesService(http_client)
-
-    @pytest.fixture
-    def endpoint(self, http_client):
-        return http_client._base_url + "/instances"
-
-    def test_get_instances(self, instances_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.GET,
-            endpoint,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        instances = instances_service.get()
-        instance = instances[0]
-
-        # assert
-        assert type(instances) == list
-        assert len(instances) == 1
-        assert type(instance) == Instance
-        assert type(instance.ssh_key_ids) == list
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_get_instances_by_status_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + "?status=running"
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        instances = instances_service.get(status='running')
-        instance = instances[0]
-
-        # assert
-        assert type(instances) == list
-        assert len(instances) == 1
-        assert type(instance) == Instance
-        assert type(instance.ssh_key_ids) == list
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_get_instances_by_status_failed(self, instances_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + "?status=blabbering"
-        responses.add(
-            responses.GET,
-            url,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            instances_service.get(status='blabbering')
-
-        # 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_get_instance_by_id_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/' + INSTANCE_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD[0],
-            status=200
-        )
-
-        # act
-        instance = instances_service.get_by_id(INSTANCE_ID)
-
-        # assert
-        assert type(instance) == Instance
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_get_instance_by_id_failed(self, instances_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:
-            instances_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_instance_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        # create instance
-        responses.add(
-            responses.POST,
-            endpoint,
-            body=INSTANCE_ID,
-            status=200
-        )
-        # get instance by id
-        url = endpoint + '/' + INSTANCE_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD[0],
-            status=200
-        )
-
-        # act
-        instance = instances_service.create(
-            instance_type=INSTANCE_TYPE,
-            image=INSTANCE_IMAGE,
-            ssh_key_ids=[SSH_KEY_ID],
-            hostname=INSTANCE_HOSTNAME,
-            description=INSTANCE_DESCRIPTION,
-            os_volume=INSTANCE_OS_VOLUME
-        )
-
-        # assert
-        assert type(instance) == Instance
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert instance.os_volume_id == OS_VOLUME_ID
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.gpu_memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(endpoint, 1) is True
-        assert responses.assert_call_count(url, 1) is True
-        assert type(instance.__str__()) == str
-
-    def test_create_spot_instance_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        # add response mock for the create instance endpoint
-        responses.add(
-            responses.POST,
-            endpoint,
-            body=INSTANCE_ID,
-            status=200
-        )
-        # add response mock for the get instance by id endpoint
-        url = endpoint + '/' + INSTANCE_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD_SPOT[0],
-            status=200
-        )
-
-        # act
-        instance = instances_service.create(
-            instance_type=INSTANCE_TYPE,
-            image=INSTANCE_IMAGE,
-            ssh_key_ids=[SSH_KEY_ID],
-            hostname=INSTANCE_HOSTNAME,
-            description=INSTANCE_DESCRIPTION,
-            os_volume=INSTANCE_OS_VOLUME
-        )
-
-        # assert
-        assert type(instance) == Instance
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert instance.os_volume_id == OS_VOLUME_ID
-        assert instance.is_spot == True
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.gpu_memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(endpoint, 1) is True
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_create_instance_attached_os_volume_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        # create instance
-        responses.add(
-            responses.POST,
-            endpoint,
-            body=INSTANCE_ID,
-            status=200
-        )
-        # get instance by id
-        url = endpoint + '/' + INSTANCE_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD[0],
-            status=200
-        )
-
-        # act
-        instance = instances_service.create(
-            instance_type=INSTANCE_TYPE,
-            image=OS_VOLUME_ID,
-            hostname=INSTANCE_HOSTNAME,
-            description=INSTANCE_DESCRIPTION,
-        )
-
-        # assert
-        assert type(instance) == Instance
-        assert instance.id == INSTANCE_ID
-        assert instance.ssh_key_ids == [SSH_KEY_ID]
-        assert instance.status == INSTANCE_STATUS
-        assert instance.image == INSTANCE_IMAGE
-        assert instance.instance_type == INSTANCE_TYPE
-        assert instance.price_per_hour == INSTANCE_PRICE_PER_HOUR
-        assert instance.location == INSTANCE_LOCATION
-        assert instance.description == INSTANCE_DESCRIPTION
-        assert instance.hostname == INSTANCE_HOSTNAME
-        assert instance.ip == INSTANCE_IP
-        assert instance.created_at == INSTANCE_CREATED_AT
-        assert instance.os_volume_id == OS_VOLUME_ID
-        assert type(instance.cpu) == dict
-        assert type(instance.gpu) == dict
-        assert type(instance.memory) == dict
-        assert type(instance.gpu_memory) == dict
-        assert type(instance.storage) == dict
-        assert responses.assert_call_count(endpoint, 1) is True
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_create_instance_failed(self, instances_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:
-            instances_service.create(
-                instance_type=INSTANCE_TYPE,
-                image=INSTANCE_IMAGE,
-                ssh_key_ids=[SSH_KEY_ID],
-                hostname=INSTANCE_HOSTNAME,
-                description=INSTANCE_DESCRIPTION,
-            )
-
-        # 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_action_successful(self, instances_service, endpoint):
-        # arrange - add response mock
-        url = endpoint
-        responses.add(
-            responses.PUT,
-            url,
-            status=202
-        )
-
-        # act
-        result = instances_service.action(
-            id_list=[INSTANCE_ID], action=Actions.SHUTDOWN)
-
-        # assert
-        assert result is None
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_action_failed(self, instances_service, endpoint):
-        # arrange - add response mock
-        url = endpoint
-        responses.add(
-            responses.PUT,
-            url,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            instances_service.action(
-                id_list=[INSTANCE_ID], action="fluxturcate")
-
-        # 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_is_available_successful(self, instances_service):
-        # arrange - add response mock
-        url = instances_service._http_client._base_url + \
-            '/instance-availability/' + INSTANCE_TYPE + "?isSpot=false"
-        responses.add(
-            responses.GET,
-            url,
-            json=True,
-            status=200
-        )
-
-        # act
-        is_available = instances_service.is_available(INSTANCE_TYPE)
-
-        # assert
-        assert is_available is True
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_is_spot_available_successful(self, instances_service):
-        # arrange - add response mock
-        url = instances_service._http_client._base_url + \
-            '/instance-availability/' + INSTANCE_TYPE + '?isSpot=true'
-        responses.add(
-            responses.GET,
-            url,
-            json=True,
-            status=200
-        )
-
-        # act
-        is_available = instances_service.is_available(
-            INSTANCE_TYPE, is_spot=True)
-
-        # assert
-        assert is_available is True
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_is_available_failed(self, instances_service):
-        # arrange - add response mock
-        url = instances_service._http_client._base_url + \
-            '/instance-availability/x' + "?isSpot=false"
-        responses.add(
-            responses.GET,
-            url,
-            json={"code": INVALID_REQUEST, "message": INVALID_REQUEST_MESSAGE},
-            status=400
-        )
-
-        # act
-        with pytest.raises(APIException) as excinfo:
-            instances_service.is_available('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/ssh_keys/test_ssh_keys.py

@@ -1,198 +0,0 @@
-import pytest
-import responses # https://github.com/getsentry/responses
-
-from datacrunch.exceptions import APIException
-from datacrunch.ssh_keys.ssh_keys import SSHKeysService, SSHKey
-
-INVALID_REQUEST = 'invalid_request'
-INVALID_REQUEST_MESSAGE = 'Your existence is invalid'
-
-KEY_ID = '01cf5dc1-a5d2-4972-ae4e-d429115d055b'
-KEY_NAME = 'key to your heart'
-KEY_VALUE = 'asdf'
-
-KEY_ID_2 = '12345dc1-a5d2-4972-ae4e-d429115d055b'
-
-PAYLOAD = [
-    {
-        'id': KEY_ID,
-        'name': KEY_NAME,
-        'key': KEY_VALUE
-    }
-]
-
-
-class TestSSHKeys:
-
-    @pytest.fixture
-    def ssh_key_service(self, http_client):
-        return SSHKeysService(http_client)
-
-    @pytest.fixture
-    def endpoint(self, http_client):
-        return http_client._base_url + "/sshkeys"
-
-    def test_get_keys(self, ssh_key_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.GET,
-            endpoint,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        keys = ssh_key_service.get()
-        
-        # assert
-        assert type(keys) == list
-        assert len(keys) == 1
-        assert type(keys[0]) == SSHKey
-        assert keys[0].id == KEY_ID
-        assert keys[0].name == KEY_NAME
-        assert keys[0].public_key == KEY_VALUE
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_get_key_by_id_successful(self, ssh_key_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/' + KEY_ID
-        responses.add(
-            responses.GET,
-            url,
-            json=PAYLOAD,
-            status=200
-        )
-
-        # act
-        key = ssh_key_service.get_by_id(KEY_ID)
-        
-        # assert
-        assert type(key) == SSHKey
-        assert key.id == KEY_ID
-        assert key.name == KEY_NAME
-        assert key.public_key == KEY_VALUE
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_get_key_by_id_failed(self, ssh_key_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:
-            ssh_key_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_key_successful(self, ssh_key_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.POST,
-            endpoint,
-            body=KEY_ID,
-            status=201
-        )
-
-        # act
-        key = ssh_key_service.create(KEY_NAME, KEY_VALUE)
-        
-        # assert
-        assert type(key) == SSHKey
-        assert type(key.id) == str
-        assert key.id == KEY_ID 
-        assert responses.assert_call_count(endpoint, 1) is True
-
-    def test_create_key_failed(self, ssh_key_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:
-            ssh_key_service.create(KEY_NAME, KEY_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_keys_successful(self, ssh_key_service, endpoint):
-        # arrange - add response mock
-        responses.add(
-            responses.DELETE,
-            endpoint,
-            status=200
-        )
-
-        # act
-        result = ssh_key_service.delete([KEY_ID, KEY_ID_2])
-
-        # assert
-        assert result is None
-        assert responses.assert_call_count(endpoint, 1) is True
-        
-
-    def test_delete_keys_failed(self, ssh_key_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:
-            ssh_key_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_key_by_id_successful(self, ssh_key_service, endpoint):
-        # arrange - add response mock
-        url = endpoint + '/' + KEY_ID
-        responses.add(
-            responses.DELETE,
-            url,
-            status=200
-        )
-
-        # act
-        result = ssh_key_service.delete_by_id(KEY_ID)
-
-        # assert
-        assert result == None
-        assert responses.assert_call_count(url, 1) is True
-
-    def test_delete_key_by_id_failed(self, ssh_key_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:
-            ssh_key_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