Add tests for friendship model