test: fix auth callback coverage test and temp credentials path

Author: DiogoRibeiro7

tests/cli/test_cli_auth.py

@@ -1,5 +1,6 @@
 import io
 import json
+import tempfile
 import unittest
 from pathlib import Path
 from unittest.mock import MagicMock, patch
@@ -183,16 +184,16 @@ def test_get_access_token_refresh_failure_deletes_credentials(
         self, mock_load, mock_validate, mock_refresh
     ):
         original_credentials_file = auth._CREDENTIALS_FILE
-        temp_credentials = Path("test_credentials.json")
-        try:
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            temp_credentials = Path(tmp_dir) / "test_credentials.json"
             temp_credentials.write_text("{}")
-            auth._CREDENTIALS_FILE = temp_credentials
-            with self.assertRaises(ValueError):
-                auth.get_access_token()
-            self.assertFalse(temp_credentials.exists())
-        finally:
-            auth._CREDENTIALS_FILE = original_credentials_file
-            temp_credentials.unlink(missing_ok=True)
+            try:
+                auth._CREDENTIALS_FILE = temp_credentials
+                with self.assertRaises(ValueError):
+                    auth.get_access_token()
+                self.assertFalse(temp_credentials.exists())
+            finally:
+                auth._CREDENTIALS_FILE = original_credentials_file
 
     @patch("codecarbon.cli.auth._load_credentials")
     def test_get_id_token(self, mock_load):
@@ -263,13 +264,21 @@ def test_authorize_raises_on_callback_error(
             "abc",
         )
         mock_session_cls.return_value = mock_session
-        mock_server_cls.return_value = MagicMock()
+        mock_server = MagicMock()
+        mock_server.handle_request.side_effect = lambda: setattr(
+            auth._CallbackHandler,
+            "error",
+            "access_denied",
+        )
+        mock_server_cls.return_value = mock_server
 
         auth._CallbackHandler.callback_url = None
-        auth._CallbackHandler.error = "access_denied"
+        auth._CallbackHandler.error = None
 
         with self.assertRaises(ValueError):
             auth.authorize()
+        mock_server.handle_request.assert_called_once()
+        mock_server.server_close.assert_called_once()
 
     @patch("codecarbon.cli.auth.HTTPServer")
     @patch("codecarbon.cli.auth.OAuth2Session")

tests/cli/test_cli_utils.py

@@ -1,5 +1,4 @@
 import configparser
-from pathlib import Path
 
 import pytest
 
@@ -80,7 +79,9 @@ def test_create_new_config_file_creates_parent_and_file(monkeypatch, tmp_path):
     target = tmp_path / "nested" / ".codecarbon.config"
     prompts = iter([str(target)])
 
-    monkeypatch.setattr(cli_utils.typer, "prompt", lambda *args, **kwargs: next(prompts))
+    monkeypatch.setattr(
+        cli_utils.typer, "prompt", lambda *args, **kwargs: next(prompts)
+    )
     monkeypatch.setattr(cli_utils.Confirm, "ask", lambda *args, **kwargs: True)
 
     created_path = cli_utils.create_new_config_file()
@@ -96,7 +97,9 @@ def test_create_new_config_file_expands_home(monkeypatch, tmp_path):
     target = home / ".codecarbon.config"
 
     monkeypatch.setattr(cli_utils.Path, "home", lambda: home)
-    monkeypatch.setattr(cli_utils.typer, "prompt", lambda *args, **kwargs: "~/.codecarbon.config")
+    monkeypatch.setattr(
+        cli_utils.typer, "prompt", lambda *args, **kwargs: "~/.codecarbon.config"
+    )
 
     created_path = cli_utils.create_new_config_file()
 

tests/cli/test_monitor.py

@@ -22,9 +22,7 @@ def stop(self):
 
 def test_run_and_monitor_requires_command(monkeypatch):
     monkeypatch.setattr(monitor_module, "EmissionsTracker", FakeTracker)
-    monkeypatch.setattr(
-        monitor_module, "print", lambda *args, **kwargs: None
-    )
+    monkeypatch.setattr(monitor_module, "print", lambda *args, **kwargs: None)
 
     with pytest.raises(typer.Exit) as exc_info:
         monitor_module.run_and_monitor(SimpleNamespace(args=[]))
@@ -39,9 +37,7 @@ def __init__(self, command, text=True):
 
     monkeypatch.setattr(monitor_module, "EmissionsTracker", FakeTracker)
     monkeypatch.setattr(monitor_module.subprocess, "Popen", FakePopen)
-    monkeypatch.setattr(
-        monitor_module, "print", lambda *args, **kwargs: None
-    )
+    monkeypatch.setattr(monitor_module, "print", lambda *args, **kwargs: None)
 
     with pytest.raises(typer.Exit) as exc_info:
         monitor_module.run_and_monitor(SimpleNamespace(args=["missing-command"]))
@@ -56,9 +52,7 @@ def __init__(self, command, text=True):
 
     monkeypatch.setattr(monitor_module, "EmissionsTracker", FakeTracker)
     monkeypatch.setattr(monitor_module.subprocess, "Popen", FakePopen)
-    monkeypatch.setattr(
-        monitor_module, "print", lambda *args, **kwargs: None
-    )
+    monkeypatch.setattr(monitor_module, "print", lambda *args, **kwargs: None)
 
     with pytest.raises(typer.Exit) as exc_info:
         monitor_module.run_and_monitor(SimpleNamespace(args=["python"]))
@@ -86,9 +80,7 @@ def kill(self):
 
     monkeypatch.setattr(monitor_module, "EmissionsTracker", FakeTracker)
     monkeypatch.setattr(monitor_module.subprocess, "Popen", FakePopen)
-    monkeypatch.setattr(
-        monitor_module, "print", lambda *args, **kwargs: None
-    )
+    monkeypatch.setattr(monitor_module, "print", lambda *args, **kwargs: None)
 
     with pytest.raises(typer.Exit) as exc_info:
         monitor_module.run_and_monitor(SimpleNamespace(args=["python"]))

tests/test_api_call.py

@@ -260,7 +260,11 @@ def test_create_run_returns_none_on_unsuccessful_status(self):
 
     def test_list_experiments_from_project_returns_empty_list_on_error(self):
         with requests_mock.Mocker() as m:
-            m.get("http://test.com/projects/proj-1/experiments", text="bad", status_code=500)
+            m.get(
+                "http://test.com/projects/proj-1/experiments",
+                text="bad",
+                status_code=500,
+            )
             api = ApiClient(
                 endpoint_url="http://test.com",
                 create_run_automatically=False,

tests/test_cpu.py

@@ -10,18 +10,18 @@
 from codecarbon.core.config import normalize_gpu_ids
 from codecarbon.core.cpu import (
     DEFAULT_POWER_PER_CORE,
+    TDP,
+    IntelPowerGadget,
+    IntelRAPL,
     _check_energy_file,
     _get_candidate_bases,
     _is_main_domain,
     _scan_base_for_rapl,
     _scan_direct_entries,
     _scan_domain_directories,
-    TDP,
-    IntelPowerGadget,
-    IntelRAPL,
     is_powergadget_available,
-    is_rapl_available,
     is_psutil_available,
+    is_rapl_available,
 )
 from codecarbon.core.resource_tracker import ResourceTracker
 from codecarbon.core.units import Energy, Power, Time
@@ -32,7 +32,9 @@
 
 class TestCPU(unittest.TestCase):
     @mock.patch("codecarbon.core.cpu.IntelPowerGadget", side_effect=Exception("boom"))
-    def test_is_powergadget_available_returns_false_on_exception(self, mock_powergadget):
+    def test_is_powergadget_available_returns_false_on_exception(
+        self, mock_powergadget
+    ):
         self.assertFalse(is_powergadget_available())
 
     @mock.patch("psutil.cpu_times")
@@ -141,19 +143,21 @@ def test_scan_base_for_rapl_checks_direct_entries_fallback(self):
         with (
             mock.patch("codecarbon.core.cpu.os.listdir", return_value=["intel-rapl:0"]),
             mock.patch("codecarbon.core.cpu.os.path.isdir", return_value=False),
-            mock.patch("codecarbon.core.cpu._scan_domain_directories", return_value=False),
+            mock.patch(
+                "codecarbon.core.cpu._scan_domain_directories", return_value=False
+            ),
             mock.patch("codecarbon.core.cpu._scan_direct_entries", return_value=True),
         ):
             assert _scan_base_for_rapl("/tmp/base", lambda _: None) is True
 
     @mock.patch("codecarbon.core.cpu._scan_base_for_rapl", side_effect=[False, True])
     @mock.patch("codecarbon.core.cpu._get_candidate_bases", return_value=["a", "b"])
-    def test_is_rapl_available_scans_candidate_bases(
-        self, mock_candidates, mock_scan
-    ):
+    def test_is_rapl_available_scans_candidate_bases(self, mock_candidates, mock_scan):
         assert is_rapl_available("/tmp/custom") is True
 
-    @mock.patch("codecarbon.core.cpu._scan_base_for_rapl", side_effect=Exception("boom"))
+    @mock.patch(
+        "codecarbon.core.cpu._scan_base_for_rapl", side_effect=Exception("boom")
+    )
     @mock.patch("codecarbon.core.cpu._get_candidate_bases", return_value=["a"])
     def test_is_rapl_available_returns_false_on_unexpected_error(
         self, mock_candidates, mock_scan
@@ -221,9 +225,7 @@ def test_setup_cli_uses_windows_backup_when_primary_missing(self):
             ),
             mock.patch(
                 "codecarbon.core.cpu.shutil.which",
-                side_effect=lambda path: None
-                if path == "PowerLog3.0.exe"
-                else path,
+                side_effect=lambda path: None if path == "PowerLog3.0.exe" else path,
             ),
         ):
             gadget = IntelPowerGadget()
@@ -269,7 +271,9 @@ def test_log_values_warns_on_nonzero_returncode_windows(self):
         gadget._log_file_path = "intel.csv"
 
         with (
-            mock.patch("codecarbon.core.cpu.subprocess.call", return_value=1) as mock_call,
+            mock.patch(
+                "codecarbon.core.cpu.subprocess.call", return_value=1
+            ) as mock_call,
             mock.patch("codecarbon.core.cpu.logger.warning") as mock_warning,
         ):
             gadget._log_values()

tests/test_powermetrics.py

@@ -73,7 +73,9 @@ def test_has_powermetrics_sudo_returns_false_on_password_prompt(self):
             ),
             mock.patch(
                 "codecarbon.core.powermetrics.subprocess.Popen",
-                return_value=FakeProcess(stderr="[sudo] password for user:", returncode=0),
+                return_value=FakeProcess(
+                    stderr="[sudo] password for user:", returncode=0
+                ),
             ),
         ):
             assert powermetrics_module._has_powermetrics_sudo() is False
@@ -106,15 +108,19 @@ def test_has_powermetrics_sudo_returns_true_on_success(self):
             assert powermetrics_module._has_powermetrics_sudo() is True
 
     def test_setup_cli_raises_on_unsupported_platform(self):
-        with mock.patch.object(ApplePowermetrics, "_setup_cli", ApplePowermetrics._setup_cli):
+        with mock.patch.object(
+            ApplePowermetrics, "_setup_cli", ApplePowermetrics._setup_cli
+        ):
             with mock.patch("codecarbon.core.powermetrics.sys.platform", "win32"):
                 with pytest.raises(SystemError):
                     ApplePowermetrics()
 
     def test_setup_cli_raises_when_binary_missing_on_apple_silicon(self):
         with (
             mock.patch("codecarbon.core.powermetrics.sys.platform", "darwin"),
-            mock.patch("codecarbon.core.powermetrics.detect_cpu_model", return_value="Apple M4"),
+            mock.patch(
+                "codecarbon.core.powermetrics.detect_cpu_model", return_value="Apple M4"
+            ),
             mock.patch("codecarbon.core.powermetrics.shutil.which", return_value=None),
         ):
             with pytest.raises(FileNotFoundError):
@@ -134,7 +140,9 @@ def test_log_values_warns_on_nonzero_returncode(self):
         powermetrics._log_file_path = "powermetrics_log.txt"
 
         with (
-            mock.patch("codecarbon.core.powermetrics.subprocess.call", return_value=1) as mock_call,
+            mock.patch(
+                "codecarbon.core.powermetrics.subprocess.call", return_value=1
+            ) as mock_call,
             mock.patch("codecarbon.core.powermetrics.logger.warning") as mock_warning,
         ):
             powermetrics._log_values()

tests/test_ram.py

@@ -148,9 +148,7 @@ def test_parse_scontrol_falls_back_to_machine_total_when_multiple_matches(
         mock_virtual_memory.return_value = mock.Mock(total=32 * 1024**3)
         ram = RAM(tracking_mode="slurm")
 
-        result = ram._parse_scontrol(
-            "AllocTRES=cpu=1,mem=4G AllocTRES=cpu=1,mem=8G"
-        )
+        result = ram._parse_scontrol("AllocTRES=cpu=1,mem=4G AllocTRES=cpu=1,mem=8G")
 
         self.assertEqual(result, mock_virtual_memory.return_value.total / (1024**3))
 
@@ -201,7 +199,9 @@ def test_process_memory_gb_includes_children(self, mock_process):
         mock_process.return_value = process
         ram = RAM(pid=123, tracking_mode="process")
 
-        with mock.patch.object(ram, "_get_children_memories", return_value=[1024**3, 0]):
+        with mock.patch.object(
+            ram, "_get_children_memories", return_value=[1024**3, 0]
+        ):
             result = ram.process_memory_GB
 
         self.assertEqual(result, 4.0)
@@ -222,8 +222,15 @@ def test_total_power_uses_process_memory_in_process_mode(self):
         ram = RAM(tracking_mode="process")
 
         with (
-            mock.patch.object(type(ram), "process_memory_GB", new_callable=mock.PropertyMock, return_value=12.0),
-            mock.patch.object(ram, "_calculate_ram_power", return_value=18.0) as mock_calc,
+            mock.patch.object(
+                type(ram),
+                "process_memory_GB",
+                new_callable=mock.PropertyMock,
+                return_value=12.0,
+            ),
+            mock.patch.object(
+                ram, "_calculate_ram_power", return_value=18.0
+            ) as mock_calc,
         ):
             result = ram.total_power()
 
@@ -234,7 +241,10 @@ def test_total_power_returns_zero_on_exception(self):
         ram = RAM(tracking_mode="machine")
 
         with mock.patch.object(
-            type(ram), "machine_memory_GB", new_callable=mock.PropertyMock, side_effect=RuntimeError("boom")
+            type(ram),
+            "machine_memory_GB",
+            new_callable=mock.PropertyMock,
+            side_effect=RuntimeError("boom"),
         ):
             result = ram.total_power()