diff options
author | Martin Sivak <msivak@redhat.com> | 2010-05-21 15:13:29 +0200 |
---|---|---|
committer | Martin Sivak <msivak@redhat.com> | 2010-05-31 14:21:50 +0200 |
commit | 9f529dd6dc8fe095709e87ce344d64b3dffe22d2 (patch) | |
tree | e6cce78fa8eeca053ee091d5b35e2e58d05bc0db /tests | |
parent | ac639fdfb97e629e05fa54db03c64b0c22be5d0e (diff) | |
download | anaconda-9f529dd6dc8fe095709e87ce344d64b3dffe22d2.tar.gz anaconda-9f529dd6dc8fe095709e87ce344d64b3dffe22d2.tar.xz anaconda-9f529dd6dc8fe095709e87ce344d64b3dffe22d2.zip |
Add simple firewall unit test
Diffstat (limited to 'tests')
-rw-r--r-- | tests/fw_test.py | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/tests/fw_test.py b/tests/fw_test.py new file mode 100644 index 000000000..6210d58b5 --- /dev/null +++ b/tests/fw_test.py @@ -0,0 +1,49 @@ +import mock + +class ImportTest(mock.TestCase): + def setUp(self): + self.setupModules([]) + + self.fs = mock.DiskIO() + + import pyanaconda.firewall + + self.modifiedModule("pyanaconda.firewall") + self.modifiedModule("os.path") + + pyanaconda.firewall.open = self.fs.open + pyanaconda.firewall.log = mock.Mock() + pyanaconda.firewall.iutil = mock.Mock() + pyanaconda.firewall.os = mock.Mock() + pyanaconda.firewall.os.path = mock.Mock() + pyanaconda.firewall.os.path.exists = self.fs.os_path_exists + + def default_write_test(self): + import pyanaconda.firewall + fw = pyanaconda.firewall.Firewall() + fw.write("/mnt/sysimage") + + self.assertEqual(pyanaconda.firewall.iutil.method_calls, [ + ("execWithRedirect", + ( + "/usr/sbin/lokkit", + [ "--quiet", "--nostart", "-f", "--service=ssh" ] + ), + { + "root": "/mnt/sysimage", + "stdout": "/dev/null", + "stderr": "/dev/null", + } + ) + ]) + + def default_ks_test(self): + """Simulate writing default fw to kickstart""" + + import pyanaconda.firewall + fw = pyanaconda.firewall.Firewall() + f = self.fs.open("/tmp/fw.txt", "w") + fw.writeKS(f) + f.close() + self.assertEquals(self.fs["/tmp/fw.txt"], "firewall --service=ssh\n") + |