Add integration test. Create custom objects to store id mapping.