FM / tests

/snap/core20/2866/usr/lib/python3/dist-packages/probert/tests/

/snap/core20/2866/usr/lib/python3/dist-packages/probert/tests

Name	Size	Mode	Actions
data/	-	0755	rm
__pycache__/	-	0755	rm
fakes.py	201	0644	edit dl rm
helpers.py	1600	0644	edit dl rm
test_dasd.py	7227	0644	edit dl rm
test_lvm.py	12482	0644	edit dl rm
test_multipath.py	3648	0644	edit dl rm
test_prober.py	1401	0644	edit dl rm
test_storage.py	1935	0644	edit dl rm
test_utils.py	2268	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /snap/core20/2866/usr/lib/python3/dist-packages/probert/tests/helpers.py (1600B)

import contextlib
import imp
import importlib
import mock
import random
import string

def builtin_module_name():
    options = ('builtins', '__builtin__')
    for name in options:
        try:
            imp.find_module(name)
        except ImportError:
            continue
        else:
            print('importing and returning: %s' % name)
            importlib.import_module(name)
            return name

@contextlib.contextmanager
def simple_mocked_open(content=None):
    if not content:
        content = ''
    m_open = mock.mock_open(read_data=content)
    mod_name = builtin_module_name()
    m_patch = '{}.open'.format(mod_name)
    with mock.patch(m_patch, m_open, create=True):
        yield m_open

def random_string(length=8):
    """ return a random lowercase string with default length of 8"""
    return ''.join(
        random.choice(string.ascii_lowercase) for _ in range(length))

# Copyright 2019 Canonical, Ltd. # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as # published by the Free Software Foundation, either version 3 of the # License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . import contextlib import imp import importlib import mock import random import string def builtin_module_name(): options = ('builtins', '__builtin__') for name in options: try: imp.find_module(name) except ImportError: continue else: print('importing and returning: %s' % name) importlib.import_module(name) return name @contextlib.contextmanager def simple_mocked_open(content=None): if not content: content = '' m_open = mock.mock_open(read_data=content) mod_name = builtin_module_name() m_patch = '{}.open'.format(mod_name) with mock.patch(m_patch, m_open, create=True): yield m_open def random_string(length=8): """ return a random lowercase string with default length of 8""" return ''.join( random.choice(string.ascii_lowercase) for _ in range(length))