Add integration test for collection.set_schema