def test_run_os_command_exception(self, popen_mock):
def base_test():
return "base test"
def sub_test():
return "output 1", "error 1"
base_test.communicate = sub_test
base_test.returncode = 0
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
popen_mock.side_effect = None
popen_mock.return_value = base_test
self.assertTrue(firewall.check_iptables())
popen_mock.side_effect = OSError('File not found')
popen_mock.return_value = None
self.assertFalse(firewall.check_iptables())
def test_run_os_command_exception(self, popen_mock):
def base_test():
return "base test"
def sub_test():
return "output 1", "error 1"
base_test.communicate = sub_test
base_test.returncode = 0
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
popen_mock.side_effect = None
popen_mock.return_value = base_test
self.assertTrue(firewall.check_iptables())
popen_mock.side_effect = OSError('File not found')
popen_mock.return_value = None
self.assertFalse(firewall.check_iptables())
def test_IpTablesStopped(self, run_os_command_mock):
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
run_os_command_mock.return_value = firewall.get_stopped_result()
self.assertFalse(firewall.check_iptables())
def test_IpTablesRunning(self, run_os_command_mock):
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
run_os_command_mock.return_value = firewall.get_running_result()
self.assertTrue(firewall.check_iptables())
def test_IpTablesStopped(self, run_os_command_mock):
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
run_os_command_mock.return_value = firewall.get_stopped_result()
self.assertFalse(firewall.check_iptables())
def test_IpTablesRunning(self, run_os_command_mock):
hostInfo = HostInfo()
for firewallType in hostInfo.getFirewallObjectTypes():
firewall = firewallType()
run_os_command_mock.return_value = firewall.get_running_result()
self.assertTrue(firewall.check_iptables())
