update test cases with new api