# -*- 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'")