1 files changed, 49 insertions, 0 deletions
diff --git a/tests/pyanaconda_test/firewall_test.py b/tests/pyanaconda_test/firewall_test.py
new file mode 100644
index 000000000..a2d141159
--- /dev/null
+++ b/tests/pyanaconda_test/firewall_test.py
@@ -0,0 +1,49 @@
+import mock
+
+class ImportTest(mock.TestCase):
+    def setUp(self):
+        self.setupModules(["_isys", "block", "ConfigParser"])
+        
+        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")
+