Add tests for expected extension behaviour