diff --git a/rowers/tests/test_emails.py b/rowers/tests/test_emails.py index 7ba6bff8..b2d6bb87 100644 --- a/rowers/tests/test_emails.py +++ b/rowers/tests/test_emails.py @@ -104,13 +104,13 @@ workout bikeerg a.save() def tearDown(self): - for filename in os.listdir('media/mailbox_attachments'): - path = os.path.join('media/mailbox_attachments/',filename) - if not os.path.isdir(path): - try: - os.remove(path) - except (IOError,FileNotFoundError,OSError): - pass + filename = 'colin3.csv' + path = os.path.join('media/mailbox_attachments/',filename) + if not os.path.isdir(path): + try: + os.remove(path) + except (IOError,FileNotFoundError,OSError): + pass @patch('rowers.dataprep.create_engine') @patch('rowers.polarstuff.get_polar_notifications') @@ -159,19 +159,19 @@ class EmailBikeUpload(TestCase): workout bike """) m.save() - a2 = 'media/mailbox_attachments/colin3.csv' + a2 = 'media/mailbox_attachments/colin4.csv' copy('rowers/tests/testdata/emails/colin.csv',a2) a = MessageAttachment(message=m,document=a2[6:]) a.save() def tearDown(self): - for filename in os.listdir('media/mailbox_attachments'): - path = os.path.join('media/mailbox_attachments/',filename) - if not os.path.isdir(path): - try: - os.remove(path) - except (IOError,FileNotFoundError,OSError): - pass + filename = 'colin4.csv' + path = os.path.join('media/mailbox_attachments/',filename) + if not os.path.isdir(path): + try: + os.remove(path) + except (IOError,FileNotFoundError,OSError): + pass @patch('rowers.dataprep.create_engine') @patch('rowers.polarstuff.get_polar_notifications')