passes tests but doesn't clean up all temp

rowers/tests/test_newusers.py

@@ -1,49 +1,7 @@
 #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
-from rowers.views import checkworkoutuser,c2_open
-from rowers.models import Workout, User, Rower, WorkoutForm,RowerForm,GraphImage
-from rowers.forms import DocumentsForm,CNsummaryForm,RegistrationFormUniqueEmail
-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 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 redis import StrictRedis
-redis_connection = StrictRedis()
+from statements import *
 
+nu = datetime.datetime.now()
 
 #@pytest.mark.django_db        
 class NewUserRegistrationTest(TestCase):
@@ -51,7 +9,14 @@ class NewUserRegistrationTest(TestCase):
 	self.c = Client()
         redis_connection.publish('tasks','KILL')
 
-    def test_newuser(self):
+    def tearDown(self):
+        try:
+            os.remove('rowsandall_workouts_2018-01-01_2019_2019-01-01.csv')
+        except:
+            pass
+            
+    @patch('rowers.dataprep.workout_summary_to_df',side_effect=mock_workout_summaries)
+    def test_newuser(self,mock_workout_summaries):
         form_data = {
 	    'first_name':'Jan',
 	    'last_name':'Roeiert',
@@ -75,3 +40,44 @@ class NewUserRegistrationTest(TestCase):
 	self.assertRedirects(response,
                              expected_url='/rowers/me/gdpr-optin/?next=/rowers/list-workouts/',
 			     status_code=302,target_status_code=200)
+
+
+        url = '/rowers/me/gdpr-optin-confirm/?next=/rowers/list-workouts/'
+
+        response = self.c.get(url)
+
+        self.assertRedirects(response,
+                             expected_url='/rowers/list-workouts/',
+                             status_code=302,target_status_code=200,
+                             )
+
+        url = '/rowers/exportallworkouts/'
+
+        response = self.c.get(url,follow=True)
+
+
+        self.assertTrue(response.status_code,200)
+
+
+        form_data = {
+            'startdate':'2018-01-01',
+            'enddate':'2019-01-01'
+        }
+
+        response = self.c.post(url,form_data)
+
+        self.assertTrue(response.status_code,200)
+        
+        url = '/rowers/me/delete/'
+
+        form_data = {
+            'delete_user':True,
+            }
+
+        response = self.c.post(url,form_data,follow=True)
+        self.assertRedirects(response,
+                            expected_url='/login/',
+                            status_code=302,target_status_code=200)
+
+
+