Update test cases for mock endpoint