nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-06-10 11:00:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

bind: Convert tests to pytest style

Browse Source 
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org>
Reviewed-by: Joseph Nuthalapati <njoseph@thoughtworks.com>
This commit is contained in:
Sunil Mohan Adapa 2019-05-01 16:07:11 -07:00 committed by Joseph Nuthalapati
parent 2d832ace36
commit 1644b30605
No known key found for this signature in database
GPG Key ID: 5398F00A2FA43C35
1 changed files with 28 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
plinth/modules/bind/tests/test_bind.py
View File

@ -18,36 +18,41 @@
Test actions for configuring bind
"""
import tempfile
import unittest
import pytest
from plinth.modules import bind
class TestBind(unittest.TestCase):
"""Test actions for configuring bind."""
@pytest.fixture(name='configuration_file')
def fixture_configuration_file(tmp_path):
"""Setup the a bind configuration file temporary directory."""
conf_file = tmp_path / 'named.conf.options'
conf_file.write_text(bind.DEFAULT_CONFIG)
old_config_file = bind.CONFIG_FILE
bind.CONFIG_FILE = str(conf_file)
yield
bind.CONFIG_FILE = old_config_file
def setUp(self):
self.conf_file = tempfile.NamedTemporaryFile()
with open(self.conf_file.name, 'w') as conf:
conf.write(bind.DEFAULT_CONFIG)
bind.CONFIG_FILE = self.conf_file.name
@pytest.mark.usefixtures('configuration_file')
def test_set_forwarders():
"""Test that setting forwarders works."""
bind.set_forwarders('8.8.8.8 8.8.4.4')
conf = bind.get_config()
assert conf['forwarders'] == '8.8.8.8 8.8.4.4'
def test_set_forwarders(self):
bind.set_forwarders('8.8.8.8 8.8.4.4')
conf = bind.get_config()
self.assertEqual(conf['forwarders'], '8.8.8.8 8.8.4.4')
bind.set_forwarders('')
conf = bind.get_config()
assert conf['forwarders'] == ''
bind.set_forwarders('')
conf = bind.get_config()
self.assertEqual(conf['forwarders'], '')
def test_enable_dnssec(self):
bind.set_dnssec('enable')
conf = bind.get_config()
self.assertEqual(conf['enable_dnssec'], True)
@pytest.mark.usefixtures('configuration_file')
def test_enable_dnssec():
"""Test that enabling DNSSEC works."""
bind.set_dnssec('enable')
conf = bind.get_config()
assert conf['enable_dnssec']
bind.set_dnssec('disable')
conf = bind.get_config()
self.assertEqual(conf['enable_dnssec'], False)
bind.set_dnssec('disable')
conf = bind.get_config()
assert not conf['enable_dnssec']