Add tests for joining associations