adding test for leaflet charts

rowers/tests/test_empower.py

@@ -0,0 +1,122 @@
+#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, RequestFactory
+
+from django.core.management import call_command
+from django.utils.six import StringIO
+from django.test.client import RequestFactory
+from rowers.views import checkworkoutuser,c2_open, multi_compare_view
+
+from rowers.forms import (
+    DocumentsForm,CNsummaryForm,RegistrationFormUniqueEmail,
+    ChartParamChoiceForm,WorkoutMultipleCompareForm,
+    BoxPlotChoiceForm)
+import rowers.plots as plots
+import rowers.interactiveplots as iplots
+import datetime
+from rowingdata import rowingdata as rdata
+from rowingdata import rower as rrower
+from django.utils import timezone
+from rowers.rows import handle_uploaded_file
+from django.core.files.uploadedfile import SimpleUploadedFile
+from time import strftime,strptime,mktime,time,daylight
+import os
+from rowers.tasks import handle_makeplot
+from rowers.utils import serialize_list,deserialize_list
+from rowers.utils import NoTokenError
+from rowers.plannedsessions import get_dates_timeperiod
+from shutil import copyfile
+from nose.tools import assert_true
+from mock import Mock, patch
+from minimocktest import MockTestCase
+import pandas as pd
+import rowers.c2stuff as c2stuff
+
+import json
+import numpy as np
+
+from rowers import urls
+from rowers.views import (
+    error500_view,error404_view,error400_view,error403_view
+    )
+
+from rowers.dataprep import delete_strokedata
+from django.contrib.sessions.middleware import SessionMiddleware
+
+from django.conf import settings
+from importlib import import_module
+
+from redis import StrictRedis
+redis_connection = StrictRedis()
+
+from django_mailbox.models import Mailbox,MessageAttachment,Message
+
+from rowers.tests.mocks import *
+
+from rowers.models import *
+from rowers.tests.mocks import *
+
+class EmpowerTest(TestCase):
+    def setUp(self):
+        redis_connection.publish('tasks','KILL')
+	u = User.objects.create_user('john',
+				     'sander@ds.ds',
+				     'koeinsloot',
+                                     )
+	r = Rower.objects.create(user=u,rowerplan='coach',gdproptin=True,
+                                 gdproptindate=timezone.now())
+	self.c = Client()
+        
+	self.nu = datetime.datetime.now()
+	filename = 'rowers/tests/testdata/testdata.csv'
+        otwfilename = 'rowers/tests/testdata/empower.csv'
+	self.wotw = Workout.objects.create(name='testworkout',
+					workouttype='water',
+					user=r,date=self.nu.strftime('%Y-%m-%d'),
+					starttime=self.nu.strftime('%H:%M:%S'),
+					duration="0:55:00",distance=8000,
+					csvfilename=filename)
+	
+	self.wote = Workout.objects.create(name='testworkout',
+					workouttype='Indoor Rower',
+					user=r,date=self.nu.strftime('%Y-%m-%d'),
+					starttime=self.nu.strftime('%H:%M:%S'),
+					duration="0:55:00",distance=8000,
+					csvfilename=otwfilename)
+	
+        powerperc = 100*np.array([r.pw_ut2,
+                                  r.pw_ut1,
+                                  r.pw_at,
+                                  r.pw_tr,r.pw_an])/r.ftp
+
+        
+	self.hrdata = {
+	    'hrmax':r.max,
+	    'hrut2':r.ut2,
+	    'hrut1':r.ut1,
+	    'hrat':r.at,
+	    'hrtr':r.tr,
+	    'hran':r.an,
+	    'ftp':r.ftp,
+            'powerperc':serialize_list(powerperc),
+            'powerzones':serialize_list(r.powerzones),
+	}
+
+    @patch('rowers.dataprep.create_engine')
+    def test_workflow_map_view(self, mocked_sqlalchemy):
+	login = self.c.login(username='john',password='koeinsloot')
+        self.assertTrue(login)
+
+        url = '/rowers/workout/{id}/workflow/'.format(
+            id = self.wote.id
+            )
+
+        response = self.c.get(url)
+
+        self.assertEqual(response.status_code, 200)