From 04ee0c05b801d134dc927466aa3a220e3e95dcb6 Mon Sep 17 00:00:00 2001 From: Cesar E Garza Date: Mon, 30 Mar 2026 01:41:57 -0500 Subject: [PATCH] persist regenerated config tokens --- src/splatnet3_scraper/query/config/config.py | 2 ++ tests/query/configuration/test_config.py | 26 ++++++++++++++++++++ 2 files changed, 28 insertions(+) diff --git a/src/splatnet3_scraper/query/config/config.py b/src/splatnet3_scraper/query/config/config.py index 9091cbc..4b0c6c7 100644 --- a/src/splatnet3_scraper/query/config/config.py +++ b/src/splatnet3_scraper/query/config/config.py @@ -108,6 +108,8 @@ def regenerate_tokens(self) -> None: token, self.token_manager.get_token(token).value, ) + if self._output_file_path is not None: + self.save_to_file() @property def session_token(self) -> str: diff --git a/tests/query/configuration/test_config.py b/tests/query/configuration/test_config.py index eff3255..45c2638 100644 --- a/tests/query/configuration/test_config.py +++ b/tests/query/configuration/test_config.py @@ -48,6 +48,32 @@ def test_regenerate_tokens(self) -> None: mock_token_manager.regenerate_tokens.assert_called_once_with() assert mock_handler.set_value.call_count == 3 + def test_regenerate_tokens_saves_file_backed_config(self) -> None: + mock_token_manager = MagicMock() + mock_handler = MagicMock() + config = Config( + mock_handler, + token_manager=mock_token_manager, + output_file_path="test.ini", + ) + config.save_to_file = MagicMock() + + config.regenerate_tokens() + + mock_token_manager.regenerate_tokens.assert_called_once_with() + assert mock_handler.set_value.call_count == 3 + config.save_to_file.assert_called_once_with() + + def test_regenerate_tokens_does_not_save_without_file_path(self) -> None: + mock_token_manager = MagicMock() + mock_handler = MagicMock() + config = Config(mock_handler, token_manager=mock_token_manager) + config.save_to_file = MagicMock() + + config.regenerate_tokens() + + config.save_to_file.assert_not_called() + @pytest.mark.parametrize( "token", [