Fix broadcast mock

Author: ericholscher

readthedocs/rtd_tests/tests/test_celery.py

@@ -54,15 +54,13 @@ def test_clear_artifacts(self):
         directory = self.project.get_production_media_path(type_='pdf', version_slug=version.slug)
         os.makedirs(directory)
         self.assertTrue(exists(directory))
-        result = tasks.clear_artifacts(version_pk=version.pk)
-        self.assertTrue(result.successful())
+        tasks.clear_artifacts(version_pk=version.pk)
         self.assertFalse(exists(directory))
 
         directory = version.project.rtd_build_path(version=version.slug)
         os.makedirs(directory)
         self.assertTrue(exists(directory))
-        result = tasks.clear_artifacts(version_pk=version.pk)
-        self.assertTrue(result.successful())
+        tasks.clear_artifacts(version_pk=version.pk)
         self.assertFalse(exists(directory))
 
     @patch('readthedocs.projects.tasks.UpdateDocsTask.build_docs',

readthedocs/rtd_tests/tests/test_project_views.py

@@ -21,6 +21,7 @@
 from readthedocs.projects.models import Project, Domain
 from readthedocs.projects.views.private import ImportWizardView
 from readthedocs.projects.views.mixins import ProjectRelationMixin
+from readthedocs.projects import tasks
 
 
 @patch('readthedocs.projects.views.private.trigger_build', lambda x, basic: None)
@@ -371,13 +372,13 @@ def test_delete_project(self):
         response = self.client.get('/dashboard/pip/delete/')
         self.assertEqual(response.status_code, 200)
 
-        patcher = patch('readthedocs.projects.tasks.broadcast')
-        with patcher as broadcast:
+        with patch('readthedocs.projects.views.private.broadcast') as broadcast:
             response = self.client.post('/dashboard/pip/delete/')
             self.assertEqual(response.status_code, 302)
             self.assertFalse(Project.objects.filter(slug='pip').exists())
             broadcast.assert_called_with(
                 type='app',
+                task=tasks.remove_dir,
                 args=[project.doc_path])