def setUp(self, data=None):
self.data = data or self.data
self.requests_patcher = patch('strava_client.service_clients.requests')
self.requests = self.requests_patcher.start()
self.user = Mock()
self.auth_model = Mock(extra_data={'access_token': 123})
def related_filter(provider=None):
return [self.auth_model]
self.user.social_auth.filter = related_filter
self.requests.get.return_value = Mock(json=lambda: self.data)
self.actual_data = StravaServiceClient.get_user_data(self.user)
def get(self, request):
strava_user = StravaServiceClient.get_user_data(request.user)
return Response(strava_user)
def test_get_user_data_calls_requests_with_correct_params(self):
self.actual_data = StravaServiceClient.get_user_data(self.user)
self.requests.get.assert_called_with(
'https://www.strava.com/api/v3/athlete',
params={'access_token': 123})
def test_calls_requests_get_function(self):
self.actual_data = StravaServiceClient.get_user_data(self.user)
self.assertEqual(self.actual_data, self.data)
