Method test_authorize_success

tests/cli/test_cli_auth.py:240–273

(
        self,
        mock_discover,
        mock_session_cls,
        mock_server_cls,
        mock_browser_open,
        mock_save_credentials,
    )

        },
    )
    def test_authorize_success(
        self,
        mock_discover,
        mock_session_cls,
        mock_server_cls,
        mock_browser_open,
        mock_save_credentials,
    ):
        mock_session = MagicMock()
        mock_session.create_authorization_url.return_value = (
            "https://auth.example/authorize?state=abc",
            "abc",
        )
        mock_session.fetch_token.return_value = {"access_token": "token"}
        mock_session_cls.return_value = mock_session

        mock_server = MagicMock()
        mock_server.handle_request.side_effect = lambda: setattr(
            auth._CallbackHandler,
            "callback_url",
            "http://localhost:8090/callback?code=123",
        )
        mock_server_cls.return_value = mock_server

        auth._CallbackHandler.callback_url = None
        auth._CallbackHandler.error = None

        result = auth.authorize()

        self.assertEqual(result, {"access_token": "token"})
        mock_browser_open.assert_called_once()
        mock_server.handle_request.assert_called_once()
        mock_server.server_close.assert_called_once()
        mock_save_credentials.assert_called_once_with({"access_token": "token"})

    @patch("codecarbon.cli.auth.HTTPServer")
    @patch("codecarbon.cli.auth.OAuth2Session")