diff --git a/onadata/apps/logger/tests/test_briefcase_client.py b/onadata/apps/logger/tests/test_briefcase_client.py index 945aeba4e4..c3ddfe6d95 100644 --- a/onadata/apps/logger/tests/test_briefcase_client.py +++ b/onadata/apps/logger/tests/test_briefcase_client.py @@ -111,7 +111,9 @@ def submission_list(request, context): res.status_code, ";".join(media), request.url, - storage.exists(os.path.join(settings.PROJECT_ROOT, "test_media/", media[0])), + storage.exists( + os.path.join(settings.PROJECT_ROOT, "test_media/", media[0]) + ), ) response.encoding = res.get("content-type") return get_streaming_content(res) @@ -129,7 +131,7 @@ def get_streaming_content(res): return content -@flaky(max_runs=3) +@flaky() class TestBriefcaseClient(TestBase): """Test briefcase_client module.""" @@ -200,6 +202,7 @@ def test_download_xform_xml(self): ) self.assertTrue(storage.exists(media_path)) + @flaky(max_runs=3, min_passes=2) def test_push(self): """Test ODK briefcase client push function.""" xforms = XForm.objects.filter( diff --git a/onadata/apps/main/tests/test_process.py b/onadata/apps/main/tests/test_process.py index a01bbbbf4f..2ab8df2f26 100644 --- a/onadata/apps/main/tests/test_process.py +++ b/onadata/apps/main/tests/test_process.py @@ -21,6 +21,7 @@ import pytz import requests from django_digest.test import Client as DigestClient +from flaky import flaky from mock import patch from six import iteritems @@ -36,6 +37,7 @@ uuid_regex = re.compile(r'(.*uuid[^//]+="\')([^\']+)(\'".*)', re.DOTALL) +@flaky() class TestProcess(TestBase, SerializeMixin): """ Test form publishing processes. @@ -153,6 +155,7 @@ def test_google_url_upload(self, mock_requests): self.assertEqual(response.status_code, 200) self.assertEqual(XForm.objects.count(), pre_count + 1) + @flaky(max_runs=3, min_passes=2) @patch("onadata.apps.main.forms.requests") def test_url_upload(self, mock_requests): """Test uploading an XLSForm from a URL."""