Add some tests for the new custom extensions API