Skip to content

Commit accbab4

Browse files
committed
fix: update test_cmd_auth patches for deferred imports
Imports in cmd_auth.py were moved into function bodies, so mock patches must target the source modules instead of cmd_auth's namespace.
1 parent 2e2e19f commit accbab4

1 file changed

Lines changed: 15 additions & 15 deletions

File tree

tests/test_cmd_auth.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99

1010

1111
class TestAuthLogin:
12-
@patch("codeflash.cli_cmds.cmd_auth.get_codeflash_api_key")
13-
@patch("codeflash.cli_cmds.cmd_auth.console")
12+
@patch("codeflash.code_utils.env_utils.get_codeflash_api_key")
13+
@patch("codeflash.cli_cmds.console.console")
1414
def test_existing_api_key_skips_oauth(self, mock_console: MagicMock, mock_get_key: MagicMock) -> None:
1515
mock_get_key.return_value = "cf-test1234abcd"
1616

@@ -21,30 +21,30 @@ def test_existing_api_key_skips_oauth(self, mock_console: MagicMock, mock_get_ke
2121
"To re-authenticate, unset [bold]CODEFLASH_API_KEY[/bold] and run this command again."
2222
)
2323

24-
@patch("codeflash.cli_cmds.cmd_auth.get_codeflash_api_key")
25-
@patch("codeflash.cli_cmds.cmd_auth.console")
24+
@patch("codeflash.code_utils.env_utils.get_codeflash_api_key")
25+
@patch("codeflash.cli_cmds.console.console")
2626
def test_existing_api_key_oserror_treated_as_missing(
2727
self, mock_console: MagicMock, mock_get_key: MagicMock
2828
) -> None:
2929
mock_get_key.side_effect = OSError("permission denied")
3030

3131
with pytest.raises(SystemExit):
32-
with patch("codeflash.cli_cmds.cmd_auth.perform_oauth_signin", return_value=None):
32+
with patch("codeflash.cli_cmds.oauth_handler.perform_oauth_signin", return_value=None):
3333
auth_login()
3434

35-
@patch("codeflash.cli_cmds.cmd_auth.perform_oauth_signin")
36-
@patch("codeflash.cli_cmds.cmd_auth.get_codeflash_api_key", return_value="")
35+
@patch("codeflash.cli_cmds.oauth_handler.perform_oauth_signin")
36+
@patch("codeflash.code_utils.env_utils.get_codeflash_api_key", return_value="")
3737
def test_oauth_failure_exits_with_code_1(self, mock_get_key: MagicMock, mock_oauth: MagicMock) -> None:
3838
mock_oauth.return_value = None
3939

4040
with pytest.raises(SystemExit, match="1"):
4141
auth_login()
4242

4343
@patch("codeflash.cli_cmds.cmd_auth.os")
44-
@patch("codeflash.cli_cmds.cmd_auth.save_api_key_to_rc")
45-
@patch("codeflash.cli_cmds.cmd_auth.perform_oauth_signin")
46-
@patch("codeflash.cli_cmds.cmd_auth.get_codeflash_api_key", return_value="")
47-
@patch("codeflash.cli_cmds.cmd_auth.console")
44+
@patch("codeflash.code_utils.shell_utils.save_api_key_to_rc")
45+
@patch("codeflash.cli_cmds.oauth_handler.perform_oauth_signin")
46+
@patch("codeflash.code_utils.env_utils.get_codeflash_api_key", return_value="")
47+
@patch("codeflash.cli_cmds.console.console")
4848
def test_successful_oauth_saves_key(
4949
self,
5050
mock_console: MagicMock,
@@ -63,10 +63,10 @@ def test_successful_oauth_saves_key(
6363
mock_console.print.assert_called_with("[green]Signed in successfully![/green]")
6464

6565
@patch("codeflash.cli_cmds.cmd_auth.os")
66-
@patch("codeflash.cli_cmds.cmd_auth.save_api_key_to_rc")
67-
@patch("codeflash.cli_cmds.cmd_auth.perform_oauth_signin")
68-
@patch("codeflash.cli_cmds.cmd_auth.get_codeflash_api_key", return_value="")
69-
@patch("codeflash.cli_cmds.cmd_auth.console")
66+
@patch("codeflash.code_utils.shell_utils.save_api_key_to_rc")
67+
@patch("codeflash.cli_cmds.oauth_handler.perform_oauth_signin")
68+
@patch("codeflash.code_utils.env_utils.get_codeflash_api_key", return_value="")
69+
@patch("codeflash.cli_cmds.console.console")
7070
def test_windows_oauth_saves_key(
7171
self,
7272
mock_console: MagicMock,

0 commit comments

Comments
 (0)