add tests for the new API