update test for new-extension api.