diff --git a/tests/test_graphql/test_repository/test_tokens_repository.py b/tests/test_graphql/test_repository/test_tokens_repository.py index 8b8b089..ee1b9e0 100644 --- a/tests/test_graphql/test_repository/test_tokens_repository.py +++ b/tests/test_graphql/test_repository/test_tokens_repository.py @@ -44,6 +44,8 @@ EMPTY_KEYS_JSON = """ } """ +EMPTY_TOKENS_JSON = ' {"tokens": []}' + def mnemonic_from_hex(hexkey): return Mnemonic(language="english").to_mnemonic(bytes.fromhex(hexkey)) @@ -65,6 +67,16 @@ def empty_keys(mocker, tmpdir): return tmpdir +@pytest.fixture +def empty_tokens(mocker, tmpdir): + tokens_file = tmpdir / "empty_tokens.json" + with open(tokens_file, "w") as file: + file.write(EMPTY_TOKENS_JSON) + mocker.patch("selfprivacy_api.utils.TOKENS_FILE", new=tokens_file) + assert read_json(tokens_file)["tokens"] == [] + return tmpdir + + @pytest.fixture def mock_new_device_key_generate(mocker): mock = mocker.patch( @@ -153,7 +165,7 @@ def mock_recovery_key_generate(mocker): @pytest.fixture -def empty_json_repo(empty_keys): +def empty_json_repo(empty_tokens): repo = JsonTokensRepository() for token in repo.get_tokens(): repo.delete_token(token)