Add tests for extension model types and records