wuttamess/tests/test_postfix.py

# -*- coding: utf-8; -*-

from unittest import TestCase
from unittest.mock import MagicMock

from wuttamess import postfix as mod


class TestSetConfig(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_config(c, "foo", "bar")
        c.run.assert_called_once_with("postconf -e 'foo=bar'")


class TestSetMyhostname(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_myhostname(c, "test.example.com")
        c.run.assert_called_once_with("postconf -e 'myhostname=test.example.com'")


class TestSetMyorigin(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_myorigin(c, "example.com")
        c.run.assert_called_once_with("postconf -e 'myorigin=example.com'")


class TestSetMydestination(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_mydestination(c, "example.com", "test.example.com", "localhost")
        c.run.assert_called_once_with(
            "postconf -e 'mydestination=example.com, test.example.com, localhost'"
        )


class TestSetMynetworks(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_mynetworks(c, "127.0.0.0/8", "[::1]/128")
        c.run.assert_called_once_with("postconf -e 'mynetworks=127.0.0.0/8 [::1]/128'")


class TestSetRelayhost(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.set_relayhost(c, "mail.example.com")
        c.run.assert_called_once_with("postconf -e 'relayhost=mail.example.com'")