def test_get_registered(self):
dm = DockerMonitor()
n = 5
for i in range(n):
dm.register(MagicMock())
registered = dm.get_registered()
self.assertEqual(len(registered), n)
def test_register_vm(self):
mock_vm = MagicMock()
dm = DockerMonitor()
dm.register(mock_vm)
registered = dm.get_registered()
# Asserts
self.assertEqual(len(registered), 1)
self.assertEqual(registered.pop(), mock_vm)
def test_start_monitoring(self, mstart_new_thread, mxenapiclient,
log_info):
mvm_ref = MagicMock()
dm = DockerMonitor(host=MagicMock())
dm.start_monitoring(mvm_ref)
registered = dm.get_registered()[0]
mstart_new_thread.assert_called_with(registered._monitoring_loop,
tuple())
def test_load_one_vm(self, mprocess_vmrecord):
host = MagicMock()
mvm_rec = MagicMock()
host.client.get_all_vm_records.return_value = {'test_rec': mvm_rec}
dm = DockerMonitor(host)
dm.refresh()
host.client.get_all_vm_records.assert_called_once_with()
mprocess_vmrecord.assert_called_with('test_rec', mvm_rec)
def test_load_all_vm(self, mprocess_vmrecord):
host = MagicMock()
# Mock out the vm_ref: vm_rec dictionary object
mvm_recs = {}
n = 10
for i in range(n):
mvm_recs["rec_%d" % i] = MagicMock()
host.client.get_all_vm_records.return_value = mvm_recs
dm = DockerMonitor(host)
dm.refresh()
host.client.get_all_vm_records.assert_called_once_with()
# Assert we process every record.
self.assertEqual(mprocess_vmrecord.call_count, n)
def test_unregister_vm(self):
mock_vm = MagicMock()
dm = DockerMonitor()
dm.register(mock_vm)
dm.deregister(mock_vm)
registered = dm.get_registered()
self.assertEqual(registered, [])
def test_stop_monitoring(self, log_info):
mvm_ref = MagicMock()
dm = DockerMonitor()
thevm = MonitoredVM(MagicMock(), ref=mvm_ref)
dm.register(thevm)
dm.stop_monitoring(mvm_ref)
self.assertEqual(thevm._stop_monitoring_request, ["stop"])
def test_deregistration_exception(self):
mock_vm = MagicMock()
dm = DockerMonitor()
dm.deregister(mock_vm)
self.assertRaises(Exception, dm.deregister(mock_vm))
def test_is_registered(self):
mock_vm = MagicMock()
dm = DockerMonitor()
self.assertEqual(dm.is_registered(mock_vm), False)
dm.register(mock_vm)
self.assertEqual(dm.is_registered(mock_vm), True)