diff --git a/bulkredditdownloader/tests/test_oauth2.py b/bulkredditdownloader/tests/test_oauth2.py index c5a63b7..ced7605 100644 --- a/bulkredditdownloader/tests/test_oauth2.py +++ b/bulkredditdownloader/tests/test_oauth2.py @@ -2,6 +2,7 @@ # coding=utf-8 import configparser +from pathlib import Path from unittest.mock import MagicMock import pytest @@ -42,10 +43,10 @@ def test_split_scopes(test_scopes: str, expected: set[str]): @pytest.mark.online @pytest.mark.parametrize('test_scopes', ( - ('random',), - ('scope', 'another_scope'), + {'random', }, + {'scope', 'another_scope'}, )) -def test_check_scopes_bad(test_scopes: list[str]): +def test_check_scopes_bad(test_scopes: set[str]): with pytest.raises(BulkDownloaderException): OAuth2Authenticator._check_scopes(test_scopes) @@ -53,14 +54,18 @@ def test_check_scopes_bad(test_scopes: list[str]): def test_token_manager_read(example_config: configparser.ConfigParser): mock_authoriser = MagicMock() mock_authoriser.refresh_token = None - test_manager = OAuth2TokenManager(example_config) + test_manager = OAuth2TokenManager(example_config, None) test_manager.pre_refresh_callback(mock_authoriser) assert mock_authoriser.refresh_token == example_config.get('DEFAULT', 'user_token') -def test_token_manager_write(example_config: configparser.ConfigParser): +def test_token_manager_write(example_config: configparser.ConfigParser, tmp_path: Path): + test_path = tmp_path / 'test.cfg' mock_authoriser = MagicMock() mock_authoriser.refresh_token = 'changed_token' - test_manager = OAuth2TokenManager(example_config) + test_manager = OAuth2TokenManager(example_config, test_path) test_manager.post_refresh_callback(mock_authoriser) assert example_config.get('DEFAULT', 'user_token') == 'changed_token' + with open(test_path, 'r') as file: + file_contents = file.read() + assert 'user_token = changed_token' in file_contents