Added missing tests for extension class.