blob: 6210d58b5781dbfc1c088dc48000fa9c3a171814 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
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")
|