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)