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")