coverage

rowers/tests/mocks.py

@@ -349,9 +349,16 @@ def mocked_cpraw(*args, **kwargs):
 
     return df
 
-def StravaActivity():
+class StravalibResponse():
+    def __init__(self,*args, **kwargs):
+        self.id = 1
+
+class StravaActivity():
     def wait(*args, **kwargs):
-        return 1
+        return StravalibResponse()
+
+    def __init__(*args, **kwargs):
+        pass
 
 
 
@@ -362,6 +369,9 @@ class MockStravalibClient():
     def update_activity(*args, **kwargs):
         return StravaActivity()
 
+    def __init__(*args,**kwargs):
+        pass
+
 # Mocked Braintree
 
 ## Higher level - unfortunately didn't succeed in mocking the gateway