From 4577924d883f56c4fe13a60d1d8533b798375aad Mon Sep 17 00:00:00 2001 From: Cédric Ollivier Date: Wed, 24 Jan 2018 13:33:20 +0100 Subject: Merge RegexMatch into test_cli_env.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit It also deletes SubstrMatch which is not called (then uncovered). Change-Id: Ifde1e2491ab5e0e75652eb4b25b359e859c2aa95 Signed-off-by: Cédric Ollivier --- functest/tests/unit/cli/commands/test_cli_env.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) (limited to 'functest/tests/unit/cli') diff --git a/functest/tests/unit/cli/commands/test_cli_env.py b/functest/tests/unit/cli/commands/test_cli_env.py index 2708add20..b5c653770 100644 --- a/functest/tests/unit/cli/commands/test_cli_env.py +++ b/functest/tests/unit/cli/commands/test_cli_env.py @@ -8,13 +8,24 @@ # pylint: disable=missing-docstring import logging +import re import unittest import mock from functest.cli.commands import cli_env from functest.utils.constants import CONST -from functest.tests.unit import test_utils + + +class RegexMatch(object): + def __init__(self, msg): + self.msg = msg + + def __eq__(self, other): + match = re.search(self.msg, other) + if match: + return True + return False class CliEnvTesting(unittest.TestCase): @@ -45,8 +56,7 @@ class CliEnvTesting(unittest.TestCase): with mock.patch('functest.cli.commands.cli_env.click.echo') \ as mock_click_echo: self.cli_environ.show() - mock_click_echo.assert_called_with(test_utils. - RegexMatch(reg_string)) + mock_click_echo.assert_called_with(RegexMatch(reg_string)) def test_show_missing_ci_installer_type(self, *args): self._test_show_missing_env_var('INSTALLER_TYPE', *args) -- cgit 1.2.3-korg