From 67607e02b85bdf47210b1f3dc6006715ea7f95a1 Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Fri, 1 Nov 2019 17:37:36 +0100
Subject: [PATCH] adding test for NK SpeedCoach Empower

---
 rowers/tests/test_units.py | 40 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/rowers/tests/test_units.py b/rowers/tests/test_units.py
index efb165ae..7a7a55b5 100644
--- a/rowers/tests/test_units.py
+++ b/rowers/tests/test_units.py
@@ -39,7 +39,7 @@ class ForceUnits(TestCase):
         login = self.c.login(username=self.u.username, password=self.password)
         self.assertTrue(login)
 
-        filename = 'rowers/tests/testdata/testdata.csv'
+        filename = 'rowers/tests/testdata/PainsledForce.csv'
         f = open(filename,'rb')
         file_data = {'file': f}
         form_data = {
@@ -68,4 +68,40 @@ class ForceUnits(TestCase):
 
         df = dataprep.getsmallrowdata_db(['averageforce'],ids=[1])
         average_N = int(df['averageforce'].mean())
-        self.assertEqual(average_N,398)
+        self.assertEqual(average_N,399)
+
+
+    def test_upload_speedcoach_N(self):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        filename = 'rowers/tests/testdata/EmpowerSpeedCoachForce.csv'
+        f = open(filename,'rb')
+        file_data = {'file': f}
+        form_data = {
+        'title':'test',
+        'workouttype':'rower',
+        'boattype':'1x',
+        'notes':'aap noot mies',
+        'make_plot':False,
+        'upload_to_c2':False,
+        'plottype':'timeplot',
+        'file': f,
+        }
+
+        form = DocumentsForm(form_data,file_data)
+        response = self.c.post('/rowers/workout/upload/', form_data, follow=True)
+
+        self.assertRedirects(response, expected_url='/rowers/workout/'+encoded1+'/edit/',
+                             status_code=302,target_status_code=200)
+
+        self.assertEqual(response.status_code, 200)
+
+        f.close()
+
+        w = Workout.objects.get(id=1)
+        self.assertEqual(w.forceunit,'lbs')
+
+        df = dataprep.getsmallrowdata_db(['averageforce'],ids=[1])
+        average_N = int(df['averageforce'].mean())
+        self.assertEqual(average_N,267)