@@ -38,6 +38,9 @@ def setUp(self):
3838        self .output_file_name  =  'fake_boa_script_results.txt' 
3939        self .job_id  =  '1a2b3c4d5e6f7g8' 
4040
41+         from  conftest  import  start_mock_send_grid 
42+         self .mock_send_grid  =  start_mock_send_grid (self )
43+ 
4144    def  tearDown (self ):
4245        super ().tearDown ()
4346
@@ -52,9 +55,10 @@ def test_boa_error_code(self):
5255        assert  BoaErrorCode .FILE_TOO_LARGE_ERROR  ==  6 
5356        assert  BoaErrorCode .JOB_TIME_OUT_ERROR  ==  7 
5457
58+     @mock .patch ('website.mails.settings.USE_EMAIL' , True ) 
59+     @mock .patch ('website.mails.settings.USE_CELERY' , False ) 
5560    def  test_handle_boa_error (self ):
56-         with  mock .patch ('addons.boa.tasks.send_mail' , return_value = None ) as  mock_send_mail , \
57-                 mock .patch ('addons.boa.tasks.sentry.log_message' , return_value = None ) as  mock_sentry_log_message , \
61+         with  mock .patch ('addons.boa.tasks.sentry.log_message' , return_value = None ) as  mock_sentry_log_message , \
5862                mock .patch ('addons.boa.tasks.logger.error' , return_value = None ) as  mock_logger_error :
5963            return_value  =  handle_boa_error (
6064                self .error_message ,
@@ -68,24 +72,7 @@ def test_handle_boa_error(self):
6872                output_file_name = self .output_file_name ,
6973                job_id = self .job_id 
7074            )
71-             mock_send_mail .assert_called_with (
72-                 to_addr = self .user_username ,
73-                 mail = ADDONS_BOA_JOB_FAILURE ,
74-                 fullname = self .user_fullname ,
75-                 code = BoaErrorCode .UNKNOWN ,
76-                 message = self .error_message ,
77-                 query_file_name = self .query_file_name ,
78-                 file_size = self .file_size ,
79-                 max_file_size = boa_settings .MAX_SUBMISSION_SIZE ,
80-                 query_file_full_path = self .file_full_path ,
81-                 output_file_name = self .output_file_name ,
82-                 job_id = self .job_id ,
83-                 max_job_wait_hours = self .max_job_wait_hours ,
84-                 project_url = self .project_url ,
85-                 boa_job_list_url = boa_settings .BOA_JOB_LIST_URL ,
86-                 boa_support_email = boa_settings .BOA_SUPPORT_EMAIL ,
87-                 osf_support_email = osf_settings .OSF_SUPPORT_EMAIL ,
88-             )
75+             self .mock_send_grid .assert_called ()
8976            mock_sentry_log_message .assert_called_with (self .error_message , skip_session = True )
9077            mock_logger_error .assert_called_with (self .error_message )
9178            assert  return_value  ==  BoaErrorCode .UNKNOWN 
@@ -167,9 +154,14 @@ def setUp(self):
167154        boa_settings .REFRESH_JOB_INTERVAL  =  DEFAULT_REFRESH_JOB_INTERVAL 
168155        boa_settings .MAX_JOB_WAITING_TIME  =  DEFAULT_MAX_JOB_WAITING_TIME 
169156
157+         from  conftest  import  start_mock_send_grid 
158+         self .mock_send_grid  =  start_mock_send_grid (self )
159+ 
170160    def  tearDown (self ):
171161        super ().tearDown ()
172162
163+     @mock .patch ('website.mails.settings.USE_EMAIL' , True ) 
164+     @mock .patch ('website.mails.settings.USE_CELERY' , False ) 
173165    async  def  test_submit_success (self ):
174166        with  mock .patch ('osf.models.user.OSFUser.objects.get' , return_value = self .user ), \
175167                mock .patch ('osf.models.user.OSFUser.get_or_create_cookie' , return_value = self .user_cookie ), \
@@ -179,7 +171,6 @@ async def test_submit_success(self):
179171                mock .patch ('boaapi.boa_client.BoaClient.query' , return_value = self .mock_job ), \
180172                mock .patch ('boaapi.boa_client.BoaClient.close' , return_value = None ) as  mock_close , \
181173                mock .patch ('asyncio.sleep' , new_callable = AsyncMock , return_value = None ) as  mock_async_sleep , \
182-                 mock .patch ('addons.boa.tasks.send_mail' , return_value = None ) as  mock_send_mail , \
183174                mock .patch ('addons.boa.tasks.handle_boa_error' , return_value = None ) as  mock_handle_boa_error :
184175            return_value  =  await  submit_to_boa_async (
185176                self .host ,
@@ -199,19 +190,7 @@ async def test_submit_success(self):
199190            assert  self .mock_job .refresh .call_count  ==  4 
200191            assert  mock_async_sleep .call_count  ==  4 
201192            mock_close .assert_called ()
202-             mock_send_mail .assert_called_with (
203-                 to_addr = self .user .username ,
204-                 mail = ADDONS_BOA_JOB_COMPLETE ,
205-                 fullname = self .user .fullname ,
206-                 query_file_name = self .query_file_name ,
207-                 query_file_full_path = self .file_full_path ,
208-                 output_file_name = self .output_file_name ,
209-                 job_id = self .mock_job .id ,
210-                 project_url = self .project_url ,
211-                 boa_job_list_url = boa_settings .BOA_JOB_LIST_URL ,
212-                 boa_support_email = boa_settings .BOA_SUPPORT_EMAIL ,
213-                 osf_support_email = osf_settings .OSF_SUPPORT_EMAIL ,
214-             )
193+             self .mock_send_grid .assert_called ()
215194            mock_handle_boa_error .assert_not_called ()
216195
217196    async  def  test_download_error (self ):
0 commit comments