From 95368b8f52be5bf4e0b3708371ad147666c35fb3 Mon Sep 17 00:00:00 2001
From: Ruslan Kuprieiev <ruslan@iterative.ai>
Date: Fri, 8 Feb 2019 05:59:38 +0200
Subject: [PATCH] test: use "flaky" nose plugin for flaky tests

Signed-off-by: Ruslan Kuprieiev <ruslan@iterative.ai>
---
 test-requirements.txt | 1 +
 tests/__main__.py     | 2 +-
 tests/test_repro.py   | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/test-requirements.txt b/test-requirements.txt
index 6f1750217..5d6bb8e74 100644
--- a/test-requirements.txt
+++ b/test-requirements.txt
@@ -1,5 +1,6 @@
 pytest>=3.3.1
 nose>=1.3.7
+flaky>=3.5.3
 mock>=2.0.0
 coverage>=4.5.1
 codecov>=2.0.15
diff --git a/tests/__main__.py b/tests/__main__.py
index 4acdc549b..9612fa198 100644
--- a/tests/__main__.py
+++ b/tests/__main__.py
@@ -12,6 +12,6 @@ os.putenv("DVC_TEST", "true")
 
 cmd = (
     "nosetests -v --processes=-1 --process-timeout=500 --cover-inclusive "
-    "--cover-erase --cover-package=dvc --with-coverage --all-modules"
+    "--cover-erase --cover-package=dvc --with-coverage --all-modules --with-flaky"
 )
 check_call(cmd, shell=True)
diff --git a/tests/test_repro.py b/tests/test_repro.py
index 9e99f7136..001588be8 100644
--- a/tests/test_repro.py
+++ b/tests/test_repro.py
@@ -12,6 +12,7 @@ import boto3
 import uuid
 import paramiko
 from google.cloud import storage as gc
+from flaky.flaky_decorator import flaky
 
 from dvc.main import main
 from dvc.project import Project, ReproductionError
@@ -1026,6 +1027,7 @@ class TestReproExternalHDFS(TestReproExternalBase):
         self.assertEqual(p.returncode, 0)
 
 
+@flaky(max_runs=3, min_passes=1)
 class TestReproExternalSSH(TestReproExternalBase):
     _dir = None