Add tests for relationship model