summaryrefslogtreecommitdiff
path: root/tests/unittests/test_features.py
diff options
context:
space:
mode:
authorJames Falcon <james.falcon@canonical.com>2021-12-15 20:16:38 -0600
committerGitHub <noreply@github.com>2021-12-15 19:16:38 -0700
commitbae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf (patch)
tree1fbb3269fc87e39832e3286ef42eefd2b23fcd44 /tests/unittests/test_features.py
parent2bcf4fa972fde686c2e3141c58e640640b44dd00 (diff)
downloadvyos-cloud-init-bae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf.tar.gz
vyos-cloud-init-bae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf.zip
Adopt Black and isort (SC-700) (#1157)
Applied Black and isort, fixed any linting issues, updated tox.ini and CI.
Diffstat (limited to 'tests/unittests/test_features.py')
-rw-r--r--tests/unittests/test_features.py36
1 files changed, 22 insertions, 14 deletions
diff --git a/tests/unittests/test_features.py b/tests/unittests/test_features.py
index d7a7226d..141de55b 100644
--- a/tests/unittests/test_features.py
+++ b/tests/unittests/test_features.py
@@ -4,10 +4,11 @@
This file is for testing the feature flag functionality itself,
NOT for testing any individual feature flag
"""
-import pytest
import sys
from pathlib import Path
+import pytest
+
import cloudinit
@@ -24,37 +25,44 @@ def create_override(request):
features and feature_overrides modules to how they were before
the test started
"""
- override_path = Path(cloudinit.__file__).parent / 'feature_overrides.py'
+ override_path = Path(cloudinit.__file__).parent / "feature_overrides.py"
if override_path.exists():
- raise Exception("feature_overrides.py unexpectedly exists! "
- "Remove it to run this test.")
- with override_path.open('w') as f:
+ raise Exception(
+ "feature_overrides.py unexpectedly exists! "
+ "Remove it to run this test."
+ )
+ with override_path.open("w") as f:
for key, value in request.param.items():
- f.write('{} = {}\n'.format(key, value))
+ f.write("{} = {}\n".format(key, value))
- sys.modules.pop('cloudinit.features', None)
+ sys.modules.pop("cloudinit.features", None)
yield
override_path.unlink()
- sys.modules.pop('cloudinit.feature_overrides', None)
+ sys.modules.pop("cloudinit.feature_overrides", None)
class TestFeatures:
def test_feature_without_override(self):
from cloudinit.features import ERROR_ON_USER_DATA_FAILURE
+
assert ERROR_ON_USER_DATA_FAILURE is True
- @pytest.mark.parametrize('create_override',
- [{'ERROR_ON_USER_DATA_FAILURE': False}],
- indirect=True)
+ @pytest.mark.parametrize(
+ "create_override",
+ [{"ERROR_ON_USER_DATA_FAILURE": False}],
+ indirect=True,
+ )
def test_feature_with_override(self, create_override):
from cloudinit.features import ERROR_ON_USER_DATA_FAILURE
+
assert ERROR_ON_USER_DATA_FAILURE is False
- @pytest.mark.parametrize('create_override',
- [{'SPAM': True}],
- indirect=True)
+ @pytest.mark.parametrize(
+ "create_override", [{"SPAM": True}], indirect=True
+ )
def test_feature_only_in_override(self, create_override):
from cloudinit.features import SPAM
+
assert SPAM is True