Initial configuration clusters list tests

Author: MudlaffP

tests/example_responses.py

@@ -6140,3 +6140,51 @@
     },
     "tagFilter": []
 }
+
+EXAMPLE_CLUSTERS_LIST_RESPONSE = [
+    {
+        "id": "cluster_id_1",
+        "name": "cluster name 1",
+        "type": "Job Cluster",
+        "region": "Private",
+        "cloud": "private",
+        "teamId": "team_id",
+        "isDefault": True,
+        "dtCreated": "2019-07-05T23:28:17.416Z",
+        "dtModified": "2019-07-05T23:28:17.416Z",
+        "clusterId": 91,
+        "isPrivate": True,
+        "modelName": "team",
+        "modelId": 1170
+    },
+    {
+        "id": "cluster_id_2",
+        "name": "cluster name 2",
+        "type": "Kubernetes Processing Site",
+        "region": "Private",
+        "cloud": "aws",
+        "teamId": "team_id",
+        "isDefault": False,
+        "dtCreated": "2019-07-22T14:50:10.170Z",
+        "dtModified": "2019-11-21T18:12:27.723Z",
+        "clusterId": 92,
+        "isPrivate": True,
+        "modelName": "team",
+        "modelId": 1170
+    },
+    {
+        "id": "cluster_id_3",
+        "name": "cluster name 3",
+        "type": "Job Cluster",
+        "region": "Private",
+        "cloud": "gcp",
+        "teamId": "team_id",
+        "isDefault": False,
+        "dtCreated": "2019-10-29T18:42:50.985Z",
+        "dtModified": "2019-11-21T18:12:27.723Z",
+        "clusterId": 100,
+        "isPrivate": True,
+        "modelName": "team",
+        "modelId": 1170
+    }
+]

tests/functional/test_clusters.py

@@ -0,0 +1,46 @@
+import mock
+from click.testing import CliRunner
+
+from gradient.api_sdk.clients import http_client
+from gradient.cli import cli
+from tests import example_responses, MockResponse
+
+
+class TestClustersList(object):
+    URL = "https://api.paperspace.io/clusters/getClusters"
+
+    COMMAND = ["clusters", "list"]
+    COMMAND_WITH_OPTIONS_FILE = ["clusters", "list", "--optionsFile", ]  # path added in test
+
+    LIST_CLUSTERS = example_responses.EXAMPLE_CLUSTERS_LIST_RESPONSE
+
+    DEFAULT_PARAMS = {
+        "filter": '{"limit": 20, "offset": 0, "where": {"isPrivate": true}}'
+    }
+
+    COMMAND_WITH_API_KEY = ["clusters", "list", "--apiKey", "some_key"]
+    EXPECTED_HEADERS = http_client.default_headers.copy()
+    EXPECTED_HEADERS_WITH_CHANGED_API_KEY = http_client.default_headers.copy()
+    EXPECTED_HEADERS_WITH_CHANGED_API_KEY["X-API-Key"] = "some_key"
+
+    LIST_STDOUT = """+--------------+----------------+----------------------------+
+| ID           | Name           | Type                       |
++--------------+----------------+----------------------------+
+| cluster_id_1 | cluster name 1 | Job Cluster                |
+| cluster_id_2 | cluster name 2 | Kubernetes Processing Site |
+| cluster_id_3 | cluster name 3 | Job Cluster                |
++--------------+----------------+----------------------------+
+"""
+
+    @mock.patch("gradient.api_sdk.clients.http_client.requests.get")
+    def test_should_send_get_request_and_print_list_of_clusters(self, get_patched):
+        get_patched.return_value = MockResponse(self.LIST_CLUSTERS, 200, "fake content")
+
+        runner = CliRunner()
+        result = runner.invoke(cli.cli, self.COMMAND)
+
+        assert self.LIST_STDOUT in result.output, result.exc_info
+        get_patched.assert_called_once_with(self.URL,
+                                            headers=self.EXPECTED_HEADERS,
+                                            json=None,
+                                            params=self.DEFAULT_PARAMS)