mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-06 10:02:57 +08:00
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
|
import os
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from hermes_cli.web_server import _save_anthropic_oauth_creds
|
||
|
|
|
||
|
|
|
||
|
|
class _DummyPool:
|
||
|
|
def entries(self):
|
||
|
|
return []
|
||
|
|
|
||
|
|
def remove_entry(self, _id):
|
||
|
|
return None
|
||
|
|
|
||
|
|
def add_entry(self, _entry):
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def oauth_file(monkeypatch, tmp_path):
|
||
|
|
target = tmp_path / '.anthropic_oauth.json'
|
||
|
|
monkeypatch.setattr('agent.anthropic_adapter._HERMES_OAUTH_FILE', target)
|
||
|
|
monkeypatch.setattr('agent.credential_pool.load_pool', lambda _provider: _DummyPool())
|
||
|
|
return target
|
||
|
|
|
||
|
|
|
||
|
|
def test_dashboard_oauth_write_uses_owner_only_permissions(oauth_file):
|
||
|
|
old_umask = os.umask(0o022)
|
||
|
|
try:
|
||
|
|
_save_anthropic_oauth_creds('access-token', 'refresh-token', 123456)
|
||
|
|
finally:
|
||
|
|
os.umask(old_umask)
|
||
|
|
|
||
|
|
assert oauth_file.exists()
|
||
|
|
mode = oauth_file.stat().st_mode & 0o777
|
||
|
|
assert mode == 0o600
|
||
|
|
|
||
|
|
|
||
|
|
def test_dashboard_oauth_write_uses_atomic_replace_and_cleans_temp_files(oauth_file, monkeypatch):
|
||
|
|
replace_calls = []
|
||
|
|
|
||
|
|
def flaky_replace(src, dst):
|
||
|
|
replace_calls.append((src, dst))
|
||
|
|
raise OSError('simulated replace failure')
|
||
|
|
|
||
|
|
monkeypatch.setattr('hermes_cli.web_server.os.replace', flaky_replace)
|
||
|
|
|
||
|
|
with pytest.raises(OSError, match='simulated replace failure'):
|
||
|
|
_save_anthropic_oauth_creds('access-token', 'refresh-token', 123456)
|
||
|
|
|
||
|
|
assert replace_calls, 'helper should attempt atomic os.replace()'
|
||
|
|
assert not oauth_file.exists()
|
||
|
|
assert not list(oauth_file.parent.glob(f'{oauth_file.name}.tmp*'))
|