Add a test for the update action