def test_refresh(self):
smart_machine_test_data = {'uuid': 'test', 'foo': 'bar'}
smart_machine = SmartMachine(datacenter=MagicMock(), data=smart_machine_test_data)
smart_machine.dc.request = MagicMock()
smart_machine.dc.request.return_value = ({'foo': 'not-bar'}, None)
smart_machine.refresh()
self.assertEqual(getattr(smart_machine, 'foo'), 'not-bar')
self.assertEqual(smart_machine.dc.request.call_count, 1)
def test_is_attached_to_network(self):
smart_machine_test_data = {'uuid': 'test', 'nics': [{'network_uuid': 'foo'}, {'network_uuid': 'bar'}]}
smart_machine = SmartMachine(datacenter=MagicMock(), data=smart_machine_test_data)
smart_machine.refresh = MagicMock()
self.assertTrue(smart_machine.is_attached_to_network('foo'))
self.assertTrue(smart_machine.is_attached_to_network('bar'))
self.assertFalse(smart_machine.is_attached_to_network('foobar'))