44
55from click .testing import CliRunner
66
7- from datacustomcode .cli import init , deploy
7+ from datacustomcode .cli import deploy , init
88
99
1010class TestInit :
@@ -80,7 +80,10 @@ def test_deploy_command_success(self, mock_credentials, mock_deploy_full):
8080 def test_deploy_command_credentials_error (self , mock_credentials ):
8181 """Test deploy command when credentials are not available."""
8282 # Mock credentials to raise ValueError
83- mock_credentials .side_effect = ValueError ("Credentials not found in env or ini file. Run `datacustomcode configure` to create a credentials file." )
83+ mock_credentials .side_effect = ValueError (
84+ "Credentials not found in env or ini file. "
85+ "Run `datacustomcode configure` to create a credentials file."
86+ )
8487
8588 runner = CliRunner ()
8689 with runner .isolated_filesystem ():
@@ -101,7 +104,9 @@ def test_deploy_command_custom_path(self, mock_credentials, mock_deploy_full):
101104 # Create test directory
102105 os .makedirs ("custom_path" , exist_ok = True )
103106
104- result = runner .invoke (deploy , ["--path" , "custom_path" , "--name" , "test-job" ])
107+ result = runner .invoke (
108+ deploy , ["--path" , "custom_path" , "--name" , "test-job" ]
109+ )
105110
106111 assert result .exit_code == 0
107112 mock_deploy_full .assert_called_once ()
@@ -112,14 +117,18 @@ def test_deploy_command_custom_path(self, mock_credentials, mock_deploy_full):
112117
113118 @patch ("datacustomcode.deploy.deploy_full" )
114119 @patch ("datacustomcode.credentials.Credentials.from_available" )
115- def test_deploy_command_custom_description (self , mock_credentials , mock_deploy_full ):
120+ def test_deploy_command_custom_description (
121+ self , mock_credentials , mock_deploy_full
122+ ):
116123 """Test deploy command with custom description."""
117124 runner = CliRunner ()
118125 with runner .isolated_filesystem ():
119126 # Create test payload directory
120127 os .makedirs ("payload" , exist_ok = True )
121128
122- result = runner .invoke (deploy , ["--name" , "test-job" , "--description" , "Custom description" ])
129+ result = runner .invoke (
130+ deploy , ["--name" , "test-job" , "--description" , "Custom description" ]
131+ )
123132
124133 assert result .exit_code == 0
125134 mock_deploy_full .assert_called_once ()
0 commit comments