fix: tests

examples/api/get_endpoints.py

@@ -0,0 +1,8 @@
+""" Get all endpoints from the API """
+
+
+import runpod
+
+endpoints = runpod.get_endpoints()
+
+print(endpoints)

runpod/cli/groups/project/helpers.py

@@ -10,6 +10,7 @@
 from runpod import get_pods, create_pod, get_endpoints
 from runpod import error as rp_error
 
+
 def validate_project_name(name):
     '''
     Validate the project name.
@@ -19,6 +20,7 @@ def validate_project_name(name):
         raise click.BadParameter(f"Project name contains an invalid character: '{match.group()}'.")
     return name
 
+
 def get_project_pod(project_id: str):
     """Check if a project pod exists.
     Return the pod_id if it exists, else return None.
@@ -29,16 +31,18 @@ def get_project_pod(project_id: str):
 
     return None
 
+
 def get_project_endpoint(project_id: str):
     """Check if a project endpoint exists.
-    Return the endpoint_id if it exists, else return None.
+    Return the endpoint if it exists, else return None.
     """
     for endpoint in get_endpoints():
         if project_id in endpoint['name']:
             return endpoint
 
     return None
 
+
 def copy_template_files(template_dir, destination):
     """Copy the template files to the destination directory."""
     for item in os.listdir(template_dir):
@@ -49,6 +53,7 @@ def copy_template_files(template_dir, destination):
         else:
             shutil.copy2(source_item, destination_item)
 
+
 def attempt_pod_launch(config, environment_variables):
     """Attempt to launch a pod with the given configuration."""
     for gpu_type in config['project'].get('gpu_types', []):
@@ -72,6 +77,7 @@ def attempt_pod_launch(config, environment_variables):
             print("Unavailable.")
     return None
 
+
 def load_project_config():
     """Load the project config file."""
     project_config_file = os.path.join(os.getcwd(), 'runpod.toml')

tests/test_cli/test_cli_groups/test_project_functions.py

@@ -256,7 +256,7 @@ def test_create_project_endpoint(self, mock_get_project_endpoint, mock_get_proje
 
         self.assertEqual(result, 'test_endpoint_id')
         mock_create_template.assert_called_once_with(
-            name='test_project-endpoint | 123456| 123456',
+            name='test_project-endpoint | 123456 | 123456',
             image_name='test_image',
             container_disk_in_gb=10,
             docker_start_cmd='bash -c ". /runpod-volume/123456/prod/venv/bin/activate && python -u /runpod-volume/123456/prod/test_project/handler.py"',  # pylint: disable=line-too-long

tests/test_cli/test_cli_groups/test_project_helpers.py

@@ -50,7 +50,7 @@ def test_get_project_endpoint_exists(self, mock_get_endpoints):
         """Test the get_project_endpoint function when the project endpoint exists."""
         mock_get_endpoints.return_value = [{"name": "test-1234", "id": "endpoint_id"}]
         result = get_project_endpoint("1234")
-        self.assertEqual(result, "endpoint_id")
+        self.assertEqual(result, {"name": "test-1234", "id": "endpoint_id"})
 
     @patch("os.listdir")
     @patch("os.path.isdir", return_value=False)