Skip to content

Commit 3ae381e

Browse files
committed
Fix lint issues
1 parent cd2b139 commit 3ae381e

1 file changed

Lines changed: 14 additions & 5 deletions

File tree

tests/test_cli.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
from click.testing import CliRunner
66

7-
from datacustomcode.cli import init, deploy
7+
from datacustomcode.cli import deploy, init
88

99

1010
class 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

Comments
 (0)