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_deregistration_exception(self):
mock_vm = MagicMock()
dm = DockerMonitor()
dm.deregister(mock_vm)
self.assertRaises(Exception, dm.deregister(mock_vm))
