84ad99b98b
Some checks failed
CI / skinny-install (aco) (push) Successful in 1m10s
CI / skinny-install (api) (push) Successful in 32s
CI / skinny-install (bcda) (push) Successful in 37s
CI / skinny-install (bib) (push) Successful in 35s
CI / skinny-install (bls) (push) Successful in 27s
CI / skinny-install (ccw) (push) Successful in 27s
CI / skinny-install (cli) (push) Successful in 39s
CI / skinny-install (cms) (push) Successful in 30s
CI / skinny-install (conf) (push) Successful in 27s
CI / skinny-install (opps) (push) Successful in 31s
CI / skinny-install (perf) (push) Successful in 40s
CI / skinny-install (pfs) (push) Successful in 36s
CI / skinny-install (rex) (push) Successful in 36s
CI / lint-test (push) Failing after 10m34s
Deploy / build-scan-report (push) Failing after 5m27s
Covers iom, oig, cli/{mail,prisma,zot}, mail/{cloudflare,postmark,
droplet,resend}, prisma/{vpn,screen,llm,project,export,eligibility,
extract,flow,ingest,tools}. Import + smoke tests with mocked deps.
Tracks #353.
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""Tests for mail.cloudflare — CF DNS records."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from mail.cloudflare import (
|
|
_upsert_record,
|
|
_zone_id,
|
|
)
|
|
|
|
|
|
class TestUpsertRecord:
|
|
def test_creates_new(self):
|
|
client = MagicMock()
|
|
client.get.return_value = MagicMock(
|
|
status_code=200, json=MagicMock(return_value={"result": []})
|
|
)
|
|
client.post.return_value = MagicMock(
|
|
status_code=201, json=MagicMock(return_value={"success": True})
|
|
)
|
|
with patch(
|
|
"mail.cloudflare._headers", return_value={"Authorization": "Bearer x"}
|
|
):
|
|
_upsert_record(
|
|
client, "zone1", type_="A", name="mail.test.io", content="1.2.3.4"
|
|
)
|
|
|
|
|
|
class TestZoneId:
|
|
def test_finds_zone(self):
|
|
client = MagicMock()
|
|
client.get.return_value = MagicMock(
|
|
status_code=200, json=MagicMock(return_value={"result": [{"id": "z1"}]})
|
|
)
|
|
client.get.return_value.raise_for_status = MagicMock()
|
|
with patch(
|
|
"mail.cloudflare._headers", return_value={"Authorization": "Bearer x"}
|
|
):
|
|
assert _zone_id(client, "test.io") == "z1"
|