summaryrefslogtreecommitdiff
path: root/tests/unit/mock
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/mock')
-rw-r--r--tests/unit/mock/loader.py1
-rw-r--r--tests/unit/mock/path.py2
-rw-r--r--tests/unit/mock/procenv.py2
-rw-r--r--tests/unit/mock/vault_helper.py1
-rw-r--r--tests/unit/mock/yaml_helper.py10
5 files changed, 14 insertions, 2 deletions
diff --git a/tests/unit/mock/loader.py b/tests/unit/mock/loader.py
index 2828f38..67b84c6 100644
--- a/tests/unit/mock/loader.py
+++ b/tests/unit/mock/loader.py
@@ -18,6 +18,7 @@
# Make coding more python3-ish
from __future__ import absolute_import, division, print_function
+
__metaclass__ = type
import os
diff --git a/tests/unit/mock/path.py b/tests/unit/mock/path.py
index 3962f24..1f1b7f0 100644
--- a/tests/unit/mock/path.py
+++ b/tests/unit/mock/path.py
@@ -1,8 +1,10 @@
from __future__ import absolute_import, division, print_function
+
__metaclass__ = type
from ansible.utils.path import unfrackpath
from ansible_collections.vyos.vyos.tests.unit.compat.mock import MagicMock
+
mock_unfrackpath_noop = MagicMock(spec_set=unfrackpath, side_effect=lambda x, *args, **kwargs: x)
diff --git a/tests/unit/mock/procenv.py b/tests/unit/mock/procenv.py
index 06987e3..d7f3dc9 100644
--- a/tests/unit/mock/procenv.py
+++ b/tests/unit/mock/procenv.py
@@ -19,10 +19,12 @@
# Make coding more python3-ish
from __future__ import absolute_import, division, print_function
+
__metaclass__ = type
import json
import sys
+
from contextlib import contextmanager
from io import BytesIO, StringIO
diff --git a/tests/unit/mock/vault_helper.py b/tests/unit/mock/vault_helper.py
index 4c6c890..6006ba9 100644
--- a/tests/unit/mock/vault_helper.py
+++ b/tests/unit/mock/vault_helper.py
@@ -14,6 +14,7 @@
# Make coding more python3-ish
from __future__ import absolute_import, division, print_function
+
__metaclass__ = type
from ansible.module_utils._text import to_bytes
diff --git a/tests/unit/mock/yaml_helper.py b/tests/unit/mock/yaml_helper.py
index 540055d..2e85759 100644
--- a/tests/unit/mock/yaml_helper.py
+++ b/tests/unit/mock/yaml_helper.py
@@ -1,9 +1,11 @@
from __future__ import absolute_import, division, print_function
+
__metaclass__ = type
import io
import yaml
+
from ansible.module_utils.six import PY3
from ansible.parsing.yaml.dumper import AnsibleDumper
from ansible.parsing.yaml.loader import AnsibleLoader
@@ -115,10 +117,14 @@ class YamlTestUtils(object):
yaml_string_obj_from_string = yaml.dump(obj_from_string, Dumper=AnsibleDumper)
else:
yaml_string_obj_from_stream = yaml.dump(
- obj_from_stream, Dumper=AnsibleDumper, encoding=None
+ obj_from_stream,
+ Dumper=AnsibleDumper,
+ encoding=None,
)
yaml_string_obj_from_string = yaml.dump(
- obj_from_string, Dumper=AnsibleDumper, encoding=None
+ obj_from_string,
+ Dumper=AnsibleDumper,
+ encoding=None,
)
assert yaml_string == yaml_string_obj_from_stream