Add more test cases for update_all