61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import pytest
|
|
|
|
import os
|
|
from os import path
|
|
from tests.conftest import global_data_dir
|
|
|
|
from selfprivacy_api.utils import get_dkim_key, get_domain
|
|
|
|
###############################################################################
|
|
|
|
DKIM_FILE_CONTENT = b'selector._domainkey\tIN\tTXT\t( "v=DKIM1; k=rsa; "\n\t "p=MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDNn/IhEz1SxgHxxxI8vlPYC2dNueiLe1GC4SYz8uHimC8SDkMvAwm7rqi2SimbFgGB5nccCNOqCkrIqJTCB9vufqBnVKAjshHqpOr5hk4JJ1T/AGQKWinstmDbfTLPYTbU8ijZrwwGeqQLlnXR5nSN0GB9GazheA9zaPsT6PV+aQIDAQAB" ) ; ----- DKIM key selector for test-domain.tld\n'
|
|
|
|
|
|
@pytest.fixture
|
|
def dkim_file(mocker, domain_file, tmpdir):
|
|
domain = get_domain()
|
|
assert domain is not None
|
|
assert domain != ""
|
|
|
|
filename = domain + ".selector.txt"
|
|
dkim_path = path.join(tmpdir, filename)
|
|
|
|
with open(dkim_path, "wb") as file:
|
|
file.write(DKIM_FILE_CONTENT)
|
|
|
|
mocker.patch("selfprivacy_api.utils.DKIM_DIR", tmpdir)
|
|
return dkim_path
|
|
|
|
|
|
@pytest.fixture
|
|
def domain_file(mocker):
|
|
# TODO: move to conftest. Challenge: it does not behave with "/" like pytest datadir does
|
|
domain_path = path.join(global_data_dir(), "domain")
|
|
mocker.patch("selfprivacy_api.utils.DOMAIN_FILE", domain_path)
|
|
return domain_path
|
|
|
|
|
|
@pytest.fixture
|
|
def no_dkim_file(dkim_file):
|
|
os.remove(dkim_file)
|
|
assert path.exists(dkim_file) is False
|
|
return dkim_file
|
|
|
|
|
|
###############################################################################
|
|
|
|
|
|
def test_get_dkim_key(domain_file, dkim_file):
|
|
"""Test DKIM key"""
|
|
dkim_key = get_dkim_key("test-domain.tld")
|
|
assert (
|
|
dkim_key
|
|
== "v=DKIM1; k=rsa; p=MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDNn/IhEz1SxgHxxxI8vlPYC2dNueiLe1GC4SYz8uHimC8SDkMvAwm7rqi2SimbFgGB5nccCNOqCkrIqJTCB9vufqBnVKAjshHqpOr5hk4JJ1T/AGQKWinstmDbfTLPYTbU8ijZrwwGeqQLlnXR5nSN0GB9GazheA9zaPsT6PV+aQIDAQAB"
|
|
)
|
|
|
|
|
|
def test_no_dkim_key(domain_file, no_dkim_file):
|
|
"""Test no DKIM key"""
|
|
dkim_key = get_dkim_key("test-domain.tld")
|
|
assert dkim_key is None
|