selfprivacy-rest-api/tests/test_rest_endpoints/services/test_mailserver.py

82 lines
2.3 KiB
Python
Raw Normal View History

import base64
import json
import pytest
###############################################################################
2021-11-30 21:53:39 +00:00
class ProcessMock:
"""Mock subprocess.Popen"""
2021-11-30 21:53:39 +00:00
def __init__(self, args, **kwargs):
self.args = args
self.kwargs = kwargs
def communicate():
return (b"I am a DKIM key", None)
2021-11-30 21:53:39 +00:00
class NoFileMock(ProcessMock):
def communicate():
return (b"", None)
@pytest.fixture
def mock_subproccess_popen(mocker):
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=ProcessMock)
2021-11-30 21:53:39 +00:00
mocker.patch(
"selfprivacy_api.rest.services.get_domain",
2021-11-30 21:53:39 +00:00
autospec=True,
return_value="example.com",
)
mocker.patch("os.path.exists", autospec=True, return_value=True)
2021-11-30 21:53:39 +00:00
return mock
@pytest.fixture
def mock_no_file(mocker):
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=NoFileMock)
2021-11-30 21:53:39 +00:00
mocker.patch(
"selfprivacy_api.rest.services.get_domain",
2021-11-30 21:53:39 +00:00
autospec=True,
return_value="example.com",
)
mocker.patch("os.path.exists", autospec=True, return_value=False)
return mock
2021-11-30 21:53:39 +00:00
###############################################################################
2021-11-30 21:53:39 +00:00
def test_unauthorized(client, mock_subproccess_popen):
"""Test unauthorized"""
response = client.get("/services/mailserver/dkim")
assert response.status_code == 401
2021-11-30 21:53:39 +00:00
def test_illegal_methods(authorized_client, mock_subproccess_popen):
response = authorized_client.post("/services/mailserver/dkim")
assert response.status_code == 405
response = authorized_client.put("/services/mailserver/dkim")
assert response.status_code == 405
response = authorized_client.delete("/services/mailserver/dkim")
assert response.status_code == 405
2021-11-30 21:53:39 +00:00
def test_dkim_key(authorized_client, mock_subproccess_popen):
"""Test DKIM key"""
response = authorized_client.get("/services/mailserver/dkim")
assert response.status_code == 200
assert base64.b64decode(response.text) == b"I am a DKIM key"
2021-11-30 21:53:39 +00:00
assert mock_subproccess_popen.call_args[0][0] == [
"cat",
"/var/dkim/example.com.selector.txt",
]
def test_no_dkim_key(authorized_client, mock_no_file):
"""Test no DKIM key"""
response = authorized_client.get("/services/mailserver/dkim")
assert response.status_code == 404
2021-11-30 21:53:39 +00:00
assert mock_no_file.called == False