def test_get_ipaddress_attribute_exists(self):
"""Should not save ipaddress when attribute exists"""
class MockSudoReturnValue(str):
succeeded = True
node = {'ipaddress': '1.1.1.1'}
with patch.object(chef, 'sudo') as mock_method:
mocked_ohai_response = '["{0}"]'.format("1.1.1.2")
mock_method.return_value = MockSudoReturnValue(mocked_ohai_response)
response = chef._get_ipaddress(node)
self.assertFalse(response)
self.assertEqual(node['ipaddress'], '1.1.1.1')
def test_get_ipaddress_attribute_exists(self):
"""Should not save ipaddress when attribute exists"""
class MockSudoReturnValue(str):
succeeded = True
node = {'ipaddress': '1.1.1.1'}
with patch.object(chef, 'sudo') as mock_method:
mocked_ohai_response = '["{0}"]'.format("1.1.1.2")
mock_method.return_value = MockSudoReturnValue(
mocked_ohai_response)
response = chef._get_ipaddress(node)
self.assertFalse(response)
self.assertEqual(node['ipaddress'], '1.1.1.1')
def test_get_ipaddress(self):
"""Should add ipaddress attribute when ohai returns correct IP address
"""
class MockSudoReturnValue(str):
succeeded = True
node = {}
fake_ip = "1.1.1.2"
with patch.object(chef, 'sudo') as mock_method:
mocked_ohai_response = '["{0}"]'.format(fake_ip)
mock_method.return_value = MockSudoReturnValue(mocked_ohai_response)
response = chef._get_ipaddress(node)
self.assertTrue(response)
self.assertEqual(node['ipaddress'], fake_ip)
def test_get_ipaddress(self):
"""Should add ipaddress attribute when ohai returns correct IP address
"""
class MockSudoReturnValue(str):
succeeded = True
node = {}
fake_ip = "1.1.1.2"
with patch.object(chef, 'sudo') as mock_method:
mocked_ohai_response = '["{0}"]'.format(fake_ip)
mock_method.return_value = MockSudoReturnValue(
mocked_ohai_response)
response = chef._get_ipaddress(node)
self.assertTrue(response)
self.assertEqual(node['ipaddress'], fake_ip)
