Add a bunch of tests for the UpdateViewModel