From de5cd462cc3f04275e22d9267ecb8e6c2b23dfda Mon Sep 17 00:00:00 2001
From: drebs <drebs@leap.se>
Date: Mon, 25 Jul 2016 21:34:23 -0300
Subject: [test] allow passing number of docs on command line on perf tests

---
 testing/tests/perf/conftest.py  | 6 ++++++
 testing/tests/perf/test_sync.py | 4 ++--
 2 files changed, 8 insertions(+), 2 deletions(-)

(limited to 'testing/tests')

diff --git a/testing/tests/perf/conftest.py b/testing/tests/perf/conftest.py
index 85a48059..463c791a 100644
--- a/testing/tests/perf/conftest.py
+++ b/testing/tests/perf/conftest.py
@@ -26,6 +26,12 @@ def pytest_addoption(parser):
         help="the url for the couch server to be used during tests")
 
 
+def pytest_addoption(parser):
+    parser.addoption(
+        "--num-docs", type="int", default=100,
+        help="the number of documents to use in performance tests")
+
+
 #
 # default options for all tests
 #
diff --git a/testing/tests/perf/test_sync.py b/testing/tests/perf/test_sync.py
index 9de733fb..45af9a91 100644
--- a/testing/tests/perf/test_sync.py
+++ b/testing/tests/perf/test_sync.py
@@ -12,7 +12,7 @@ content = ' ' * 10000
 @pytest.inlineCallbacks
 def test_upload(soledad_client, request):
     # create a bunch of local documents
-    uploads = 100
+    uploads = request.config.option.num_docs
     deferreds = []
     for i in xrange(uploads):
         d = soledad_client.create_doc({'upload': True, 'content': content})
@@ -32,7 +32,7 @@ def test_upload(soledad_client, request):
 @pytest.inlineCallbacks
 def test_download(soledad_client, request):
     # create a bunch of remote documents
-    downloads = 100
+    downloads = request.config.option.num_docs
     url = request.config.getoption('--couch-url')
     remote = CouchDatabase(url, 'user-0')
     for i in xrange(downloads):
-- 
cgit v1.2.3