Add unit tests for collection_update