adding some tests

rowers/tests/mocks.py

@@ -350,6 +350,11 @@ def mocked_getsmallrowdata_db_water(*args, **kwargs):
 
     return df
 
+def mocked_getsmallrowdata_db_wps(*args, **kwargs):
+    df = pd.read_csv('rowers/tests/testdata/driveenergies.csv')
+
+    return df
+
 def mocked_getpowerdata_db(*args, **kwargs):
     df = pd.read_csv('rowers/tests/testdata/fake_powerdata.csv')
 

rowers/tests/test_async_tasks.py

@@ -510,7 +510,7 @@ class AsyncTaskTests(TestCase):
         self.assertEqual(res,1)
 
     @patch('rowers.dataprepnodjango.getsmallrowdata_db')
-    def test_handle_setcp(self,mocked_getsmallrowdata_db_db_setcp):
+    def test_handle_setcp(self,mocked_getsmallrowdata_db_setcp):
         strokesdf = pd.read_csv('rowers/tests/testdata/uhfull.csv')
         filename = 'rowers/tests/testdata/temp/pq.gz'
         workoutids = 1
@@ -520,3 +520,11 @@ class AsyncTaskTests(TestCase):
             os.remove(filename)
         except FileNotFoundError:
             pass
+
+    @patch('rowers.dataprepnodjango.getsmallrowdata_db')
+    def test_handle_update_wps(self,mocked_getsmallrowdata_db_wps):
+      ids = [1,2,3]
+      result = tasks.handle_update_wps(self.r.id,['water'],ids,mode='water')
+      self.assertEqual(result,110)
+      result = tasks.handle_update_wps(self.r.id,['water'],ids,mode='erg')
+      self.assertEqual(result,110)