small refactor step 1
This commit is contained in:
@@ -623,6 +623,12 @@ def mocked_requests(*args, **kwargs):
|
||||
with open('rowers/tests/testdata/nk_list.json','r') as infile:
|
||||
nkworkoutlist = json.load(infile)
|
||||
|
||||
with open('rowers/tests/testdata/example-session-strokes-with-impeller-data.json','r') as infile:
|
||||
nkimpellerstrokedata = json.load(infile)
|
||||
|
||||
with open('rowers/tests/testdata/example-session-with-impeller-data.json','r') as infile:
|
||||
nkimpellerworkoutlist = json.load(infile)
|
||||
|
||||
polar_json = {
|
||||
'available-user-data': []
|
||||
}
|
||||
@@ -960,10 +966,14 @@ def mocked_requests(*args, **kwargs):
|
||||
}
|
||||
return MockResponse(json_data,200)
|
||||
if nkstrokestester.match(args[0]):
|
||||
|
||||
params = kwargs.pop('params',{})
|
||||
if 'sessionIds' in params and params['sessionIds'] == '404':
|
||||
return MockResponse(nkimpellerstrokedata, 200)
|
||||
return MockResponse(nkstrokedata,200)
|
||||
if nkworkoutlisttester.match(args[0]):
|
||||
|
||||
params = kwargs.pop('params',{})
|
||||
if 'startTime' in params and params['startTime'] == 1267136372998:
|
||||
return MockResponse(nkimpellerworkoutlist,200)
|
||||
return MockResponse(nkworkoutlist,200)
|
||||
|
||||
|
||||
|
||||
@@ -424,9 +424,25 @@ class NKObjects(DjangoTestCase):
|
||||
|
||||
self.assertEqual(response.status_code,200)
|
||||
|
||||
|
||||
@patch('rowers.nkstuff.requests.get', side_effect=mocked_requests)
|
||||
@patch('rowers.nkstuff.requests.post', side_effect=mocked_requests)
|
||||
@patch('rowers.nkstuff.requests.session', side_effect=mocked_session)
|
||||
def test_nk_get_workouts(self, mock_get, mockpost):
|
||||
result = rowers.nkstuff.nk_open(self.u)
|
||||
self.assertEqual(result,"TA3n1vrNjuQJWw0TdCDHnjSmrjIPULhTlejMIWqq")
|
||||
response = self.c.get('/rowers/workout/nkimport/all/',follow=True)
|
||||
|
||||
expected = reverse('workouts_view')
|
||||
|
||||
self.assertRedirects(response,
|
||||
expected_url=expected,
|
||||
status_code=302,target_status_code=200)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
@patch('rowers.nkstuff.requests.get', side_effect=mocked_requests)
|
||||
@patch('rowers.nkstuff.requests.post', side_effect=mocked_requests)
|
||||
@patch('rowers.nkimportutils.requests.session', side_effect=mocked_session)
|
||||
@patch('rowers.dataprep.getsmallrowdata_db', side_effect=mocked_getsmallrowdata_db)
|
||||
def test_nk_import(self, mock_get, mock_post,
|
||||
mocked_session,
|
||||
@@ -441,6 +457,27 @@ class NKObjects(DjangoTestCase):
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
@patch('rowers.nkstuff.requests.get', side_effect=mocked_requests)
|
||||
@patch('rowers.nkstuff.requests.post', side_effect=mocked_requests)
|
||||
@patch('rowers.nkimportutils.requests.session', side_effect=mocked_session)
|
||||
@patch('rowers.dataprep.getsmallrowdata_db', side_effect=mocked_getsmallrowdata_db)
|
||||
def test_nk_import_impeller(self, mock_get, mock_post,
|
||||
mocked_session,
|
||||
mocked_getsmallrowdata_db):
|
||||
|
||||
result = rowers.nkstuff.rower_nk_token_refresh(self.u)
|
||||
response = self.c.get('/rowers/workout/nkimport/404',follow=True)
|
||||
|
||||
self.assertRedirects(response,
|
||||
expected_url='/rowers/workout/'+encoded1+'/edit/',
|
||||
status_code=301,target_status_code=200)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
response = self.c.get('/rowers/workout/'+encoded1+'/otwuseimpeller/', follow=True)
|
||||
self.assertRedirects(response,
|
||||
expected_url='/rowers/workout/'+encoded1+'/edit/',
|
||||
status_code=302,target_status_code=200)
|
||||
|
||||
|
||||
#@pytest.mark.django_db
|
||||
|
||||
Reference in New Issue
Block a user