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/'