def test_lookup_registration_id_for_mds(self):
registry = DeviceRegistry(self.file_name)
registry.save_device_registration("node1", "registration1")
registry.save_device_registration("node2", "registration2")
node_to_be_found = "node"
registry_key_to_return = "registry"
registry.save_device_registration(node_to_be_found,
registry_key_to_return)
self.assertEqual(
registry.lookup_registration_id_for_mds(node_to_be_found),
registry_key_to_return)
