From 14367ab4d76034f6d9f6cc3a4fb03725804a12ac Mon Sep 17 00:00:00 2001 From: Sander Roosendaal Date: Thu, 17 Feb 2022 09:52:14 +0100 Subject: [PATCH] fixing 2 tests --- rowers/tests/test_imports.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/rowers/tests/test_imports.py b/rowers/tests/test_imports.py index b7cf96b1..6ef6c6e0 100644 --- a/rowers/tests/test_imports.py +++ b/rowers/tests/test_imports.py @@ -455,7 +455,8 @@ class C2Objects(DjangoTestCase): @patch('rowers.c2stuff.requests.get', side_effect=mocked_requests) @patch('rowers.dataprep.create_engine') - def test_c2_import_tz(self, mock_get, mocked_sqlalchemy): + @patch('rowers.tasks.requests.session', side_effect=mocked_session) + def test_c2_import_tz(self, mock_get, mocked_sqlalchemy, mock_session): response = self.c.get('/rowers/workout/c2import/22/',follow=True) expected_url = '/rowers/workout/c2list/' @@ -493,7 +494,8 @@ class C2Objects(DjangoTestCase): @patch('rowers.tasks.requests.get', side_effect=mocked_requests) @patch('rowers.dataprep.create_engine') - def test_c2_import_tz2(self, mock_get, mocked_sqlalchemy): + @patch('rowers.c2stuff.requests.session', side_effect=mocked_requests) + def test_c2_import_tz2(self, mock_get, mocked_sqlalchemy, MockSession): response = self.c.get('/rowers/workout/c2import/31/',follow=True) expected_url = '/rowers/workout/c2list/'