diff options
Diffstat (limited to 'tests/integration_tests/util.py')
| -rw-r--r-- | tests/integration_tests/util.py | 15 | 
1 files changed, 14 insertions, 1 deletions
| diff --git a/tests/integration_tests/util.py b/tests/integration_tests/util.py index 8d726bb2..ce62ffc8 100644 --- a/tests/integration_tests/util.py +++ b/tests/integration_tests/util.py @@ -3,12 +3,15 @@ import multiprocessing  import os  import time  from contextlib import contextmanager +from collections import namedtuple  from pathlib import Path -log = logging.getLogger('integration_testing') +log = logging.getLogger('integration_testing') +key_pair = namedtuple('key_pair', 'public_key private_key')  ASSETS_DIR = Path('tests/integration_tests/assets') +KEY_PATH = ASSETS_DIR / 'keys'  def verify_ordered_items_in_text(to_verify: list, text: str): @@ -51,3 +54,13 @@ def emit_dots_on_travis():          yield      finally:          dot_process.terminate() + + +def get_test_rsa_keypair(key_name: str = 'test1') -> key_pair: +    private_key_path = KEY_PATH / 'id_rsa.{}'.format(key_name) +    public_key_path = KEY_PATH / 'id_rsa.{}.pub'.format(key_name) +    with public_key_path.open() as public_file: +        public_key = public_file.read() +    with private_key_path.open() as private_file: +        private_key = private_file.read() +    return key_pair(public_key, private_key) | 
