community.docker/tests/unit/plugins/module_utils/test__scramble.py
Felix Fontein e8ec22d3b1
Python code modernization, 3/n (#1157)
* Remove __metaclass__ = type.

for i in $(grep -REl '__metaclass__ = type' plugins/ tests/); do
  sed -e '/^__metaclass__ = type/d' -i $i;
done

* Remove super arguments, and stop inheriting from object.
2025-10-10 08:11:58 +02:00

30 lines
854 B
Python

# Copyright 2022 Red Hat | Ansible
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later
from __future__ import annotations
import pytest
from ansible_collections.community.docker.plugins.module_utils._scramble import (
scramble,
unscramble,
)
@pytest.mark.parametrize(
"plaintext, key, scrambled",
[
("", b"0", "=S="),
("hello", b"\x00", "=S=aGVsbG8="),
("hello", b"\x01", "=S=aWRtbW4="),
],
)
def test_scramble_unscramble(plaintext, key, scrambled):
scrambled_ = scramble(plaintext, key)
print(f"{scrambled_!r} == {scrambled!r}")
assert scrambled_ == scrambled
plaintext_ = unscramble(scrambled, key)
print(f"{plaintext_!r} == {plaintext!r}")
assert plaintext_ == plaintext