Add tests for address model