aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/test/py
diff options
context:
space:
mode:
Diffstat (limited to 'test/py')
-rw-r--r--test/py/conftest.py6
-rw-r--r--test/py/lib/test_repo.py30
2 files changed, 33 insertions, 3 deletions
diff --git a/test/py/conftest.py b/test/py/conftest.py
index 6102898..3a9c589 100644
--- a/test/py/conftest.py
+++ b/test/py/conftest.py
@@ -11,7 +11,7 @@ from pytest import fixture
from lib.db import Database
from lib.net import random_unused_port
from lib.process import CmdLine
-from lib.test_repo import TestRepoOutputSimple, TestRepoOutputEmpty
+from lib.test_repo import TestRepoOutputSimple, TestRepoOutputEmpty, TestRepoOutputLong
class Param:
@@ -188,8 +188,8 @@ def client(client_cmd):
return client_cmd
-@fixture(params=[TestRepoOutputSimple, TestRepoOutputEmpty],
- ids=['output_simple', 'output_empty'])
+@fixture(params=[TestRepoOutputSimple, TestRepoOutputEmpty, TestRepoOutputLong],
+ ids=['output_simple', 'output_empty', 'output_long'])
def test_repo(tmp_path, request):
return request.param(os.path.join(tmp_path, 'repo'))
diff --git a/test/py/lib/test_repo.py b/test/py/lib/test_repo.py
index 94b6f93..3234de5 100644
--- a/test/py/lib/test_repo.py
+++ b/test/py/lib/test_repo.py
@@ -4,8 +4,10 @@
# Distributed under the MIT License.
import abc
+import base64
import logging
import os
+import random
import shlex
import shutil
@@ -137,3 +139,31 @@ class TestRepoOutputEmpty(TestRepoOutput):
def run_output_matches(self, output):
return len(output) == 0
+
+
+# Making it a bash script introduces way too much overhead with all the
+# argument expansions; it slows things down considerably.
+OUTPUT_SCRIPT_LONG = R'''#!/usr/bin/env python3
+
+output = {output}
+
+import sys
+sys.stdout.write(output)
+'''
+
+
+class TestRepoOutputLong(TestRepoOutput):
+ __test__ = False
+
+ OUTPUT_LEN_KB = 300
+
+ def __init__(self, *args, **kwargs):
+ nb = TestRepoOutputLong.OUTPUT_LEN_KB * 1024
+ self.output = base64.encodebytes(random.randbytes(nb)).decode()
+ super().__init__(*args, **kwargs)
+
+ def _format_output_script(self):
+ return OUTPUT_SCRIPT_LONG.format(output=repr(self.output))
+
+ def run_output_matches(self, output):
+ return output.decode() == self.output