test: add comprehensive unit tests for all managers

This commit is contained in:
Fredrick Amnehagen 2026-02-06 00:46:16 +01:00
parent 2d9e1bc06e
commit 2382e17830
3 changed files with 137 additions and 0 deletions

View file

@ -0,0 +1,55 @@
import pytest
from unittest.mock import MagicMock, patch
from infra_cli.cloudflare import CloudflareManager
class MockConfig:
def get(self, key, default=None):
if key == 'cloudflare.token': return 'test-token'
return default
@patch('infra_cli.cloudflare.requests')
def test_cloudflare_zone_id_retrieval(mock_requests):
# Mock API response
mock_res = MagicMock()
mock_res.json.return_value = {
"success": True,
"result": [{"id": "zone-123"}]
}
mock_requests.get.return_value = mock_res
mgr = CloudflareManager(MockConfig())
zone_id = mgr.get_zone_id("example.com")
assert zone_id == "zone-123"
mock_requests.get.assert_called_with(
"https://api.cloudflare.com/client/v4/zones?name=example.com",
headers={"Authorization": "Bearer test-token", "Content-Type": "application/json"}
)
@patch('infra_cli.cloudflare.requests')
def test_cloudflare_update_logic(mock_requests):
mgr = CloudflareManager(MockConfig())
mgr.add_domain("example.com")
# Mock external IP check
mock_requests.get.return_value.text = "1.2.3.4"
# Mock Zone ID and DNS Records
with patch.object(CloudflareManager, 'get_zone_id', return_value="zone-123"):
# Mock finding existing A record
mock_get_records = MagicMock()
mock_get_records.json.return_value = {
"result": [{"id": "rec-456", "content": "9.9.9.9", "proxied": True}]
}
# Setup specific returns for consecutive calls
mock_requests.get.side_effect = [
MagicMock(text="1.2.3.4"), # External IP
mock_get_records # List records
]
# Mock successful update
mock_requests.put.return_value.json.return_value = {"success": True}
res = mgr.update_ddns()
assert "Updated to 1.2.3.4" in res