Add a couple unit tests for update command/update task