def test_new_zone(self):
'''
Test for Add a new zone
'''
with patch.object(firewalld, '__mgmt', return_value='success'):
mock = MagicMock(return_value='A')
with patch.object(firewalld, '__firewall_cmd', mock):
self.assertEqual(firewalld.new_zone('zone'), 'A')
with patch.object(firewalld, '__mgmt', return_value='A'):
self.assertEqual(firewalld.new_zone('zone'), 'A')
with patch.object(firewalld, '__mgmt', return_value='A'):
self.assertEqual(firewalld.new_zone('zone', False), 'A')
def test_new_zone(self):
"""
Test for Add a new zone
"""
with patch.object(firewalld, "__mgmt", return_value="success"):
mock = MagicMock(return_value="A")
with patch.object(firewalld, "__firewall_cmd", mock):
self.assertEqual(firewalld.new_zone("zone"), "A")
with patch.object(firewalld, "__mgmt", return_value="A"):
self.assertEqual(firewalld.new_zone("zone"), "A")
with patch.object(firewalld, "__mgmt", return_value="A"):
self.assertEqual(firewalld.new_zone("zone", False), "A")
def test_new_zone(self):
'''
Test for Add a new zone
'''
with patch.object(firewalld, '__mgmt', return_value='success'):
mock = MagicMock(return_value='A')
with patch.object(firewalld, '__firewall_cmd', mock):
self.assertEqual(firewalld.new_zone('zone'), 'A')
with patch.object(firewalld, '__mgmt', return_value='A'):
self.assertEqual(firewalld.new_zone('zone'), 'A')
with patch.object(firewalld, '__mgmt', return_value='A'):
self.assertEqual(firewalld.new_zone('zone', False), 'A')
