adding tests to increase coverage

rowers/tests/test_user.py

@@ -0,0 +1,89 @@
+#from __future__ import print_function
+import pytest
+
+pytestmark = pytest.mark.django_db
+
+from bs4 import BeautifulSoup
+import re
+from nose_parameterized import parameterized
+from django.test import TestCase, Client,override_settings
+from django.core.management import call_command
+from django.utils.six import StringIO
+from django.test.client import RequestFactory
+import datetime
+from django.utils import timezone
+
+import json
+import numpy as np
+import rowers.dataprep as dataprep
+
+from redis import StrictRedis
+redis_connection = StrictRedis()
+
+from rowers.models import User,Rower, Workout
+
+class UserMiddleWareTest(TestCase):
+    def setUp(self):
+	u = User.objects.create_user('john',
+				     'sander@ds.ds',
+				     'koeinsloot')
+	r = Rower.objects.create(user=u)
+        
+        self.c = Client()
+        self.c.login(username='john',password='koeinsloot')
+	nu = datetime.datetime.now()
+
+        response = dataprep.new_workout_from_file(r,'/media/testdata.csv.gz',
+                                                  title='sample')
+
+        newworkoutid = response[0]
+        self.w = Workout.objects.get(id=newworkoutid)
+        
+        
+    def test_middleware(self):
+        response = self.c.get('/rowers/list-workouts',follow=True)
+        self.assertEqual(response.status_code,200)
+        
+class UserExpireTest(TestCase):
+    def setUp(self):
+        redis_connection.publish('tasks','KILL')
+	self.u = User.objects.create_user('john',
+				     'sander@ds.ds',
+				     'koeinsloot')
+
+        yesterday_dt = timezone.now()-datetime.timedelta(days=1)
+        yesterday = yesterday_dt.date()
+        
+	r = Rower.objects.create(user=self.u,planexpires = yesterday,rowerplan='pro',
+                                 paymenttype='single')
+        
+        self.c = Client()
+        self.c.login(username='john',password='koeinsloot')
+	nu = datetime.datetime.now()
+
+        response = dataprep.new_workout_from_file(r,'/media/testdata.csv.gz',
+                                                  title='sample')
+
+        newworkoutid = response[0]
+        self.w = Workout.objects.get(id=newworkoutid)
+        
+        
+    def test_expire(self):
+        response = self.c.get('/rowers/list-workouts',follow=True)
+        self.assertEqual(response.status_code,200)
+        self.assertEqual(self.u.rower.rowerplan,'basic')
+        
+class UserTestsNoRower(TestCase):
+    def setUp(self):
+	u = User.objects.create_user('john',
+				     'sander@ds.ds',
+				     'koeinsloot')
+
+        self.c = Client()
+        self.c.login(username='john',password='koeinsloot')
+
+        
+    def test_user(self):
+        response = self.c.get('/rowers/list-workouts',follow=True)
+        self.assertEqual(response.status_code,200)
+