Add tests for updater