Add tests for updater cleanup