Add tests for collection hydration