Improve tests around OnlineSchemaChange