[Backport] Fixing delete project dump asynch task
parent
15d8793fe3
commit
bff7fe580f
|
@ -63,7 +63,7 @@ class ProjectExporterViewSet(mixins.ImportThrottlingPolicyMixin, GenericViewSet)
|
||||||
|
|
||||||
if settings.CELERY_ENABLED:
|
if settings.CELERY_ENABLED:
|
||||||
task = tasks.dump_project.delay(request.user, project)
|
task = tasks.dump_project.delay(request.user, project)
|
||||||
tasks.delete_project_dump.apply_async((project.pk, project.slug),
|
tasks.delete_project_dump.apply_async((project.pk, project.slug, task.id),
|
||||||
countdown=settings.EXPORTS_TTL)
|
countdown=settings.EXPORTS_TTL)
|
||||||
return response.Accepted({"export_id": task.id})
|
return response.Accepted({"export_id": task.id})
|
||||||
|
|
||||||
|
|
|
@ -16,6 +16,8 @@
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
from unittest import mock
|
||||||
|
|
||||||
from django.core.urlresolvers import reverse
|
from django.core.urlresolvers import reverse
|
||||||
|
|
||||||
from .. import factories as f
|
from .. import factories as f
|
||||||
|
@ -61,10 +63,16 @@ def test_valid_project_export_with_celery_enabled(client, settings):
|
||||||
|
|
||||||
url = reverse("exporter-detail", args=[project.pk])
|
url = reverse("exporter-detail", args=[project.pk])
|
||||||
|
|
||||||
response = client.get(url, content_type="application/json")
|
#delete_project_dump task should have been launched
|
||||||
assert response.status_code == 202
|
with mock.patch('taiga.export_import.tasks.delete_project_dump') as delete_project_dump_mock:
|
||||||
response_data = response.data
|
response = client.get(url, content_type="application/json")
|
||||||
assert "export_id" in response_data
|
assert response.status_code == 202
|
||||||
|
response_data = response.data
|
||||||
|
assert "export_id" in response_data
|
||||||
|
|
||||||
|
args = (project.id, project.slug, response_data["export_id"],)
|
||||||
|
kwargs = {"countdown": settings.EXPORTS_TTL}
|
||||||
|
delete_project_dump_mock.apply_async.assert_called_once_with(args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
def test_valid_project_with_throttling(client, settings):
|
def test_valid_project_with_throttling(client, settings):
|
||||||
|
|
Loading…
Reference in New Issue