tests: Update tests to new APIs
2 files changed