fix: missing test

Author: justinmerrell

runpod/cli/groups/project/commands.py

@@ -3,6 +3,7 @@
 '''
 
 import os
+import sys
 import click
 from InquirerPy import prompt as cli_select
 
@@ -30,6 +31,7 @@ def new_project_wizard(project_name, model_type, model_name, init_current_dir):
     if len(network_volumes) == 0:
         click.echo("You do not have any network volumes.")
         click.echo("Please create a network volume (https://runpod.io/console/user/storage) and try again.") # pylint: disable=line-too-long
+        sys.exit(1)
 
     click.echo("Creating a new project...")
 

tests/test_cli/test_cli_groups/test_project_commands.py

@@ -16,6 +16,17 @@ class TestProjectCLI(unittest.TestCase):
     def setUp(self):
         self.runner = CliRunner()
 
+    def test_new_project_wizard_no_network_volumes(self):
+        '''
+        Tests the new_project_wizard command with no network volumes.
+        '''
+        with patch('runpod.cli.groups.project.commands.get_user') as mock_get_user:
+            mock_get_user.return_value = {'networkVolumes':[]}
+
+            result = self.runner.invoke(new_project_wizard)
+
+        self.assertEqual(result.exit_code, 1)
+        self.assertIn("You do not have any network volumes.", result.output)
 
     def test_new_project_wizard_success(self):
         '''