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