mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-02-04 07:51:50 +00:00
313 lines
10 KiB
Python
313 lines
10 KiB
Python
#
|
|
# Copyright (c) 2021, Andreas Botzner <andreas at botzner dot com>
|
|
# 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 json
|
|
|
|
import pytest
|
|
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import set_module_args
|
|
from redis import __version__
|
|
|
|
from ansible_collections.community.general.plugins.modules import redis_data
|
|
|
|
HAS_REDIS_USERNAME_OPTION = True
|
|
if tuple(map(int, __version__.split("."))) < (3, 4, 0):
|
|
HAS_REDIS_USERNAME_OPTION = False
|
|
|
|
|
|
def test_redis_data_without_arguments(capfd):
|
|
with set_module_args({}):
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
assert not err
|
|
assert json.loads(out)["failed"]
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
|
|
def test_redis_data_key(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_user": "root",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"value": "baz",
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
mocker.patch("redis.Redis.set", return_value=True)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["old_value"] == "bar"
|
|
assert json.loads(out)["value"] == "baz"
|
|
assert json.loads(out)["msg"] == "Set key: foo"
|
|
assert json.loads(out)["changed"] is True
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
|
|
def test_redis_data_existing_key_nx(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_user": "root",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"value": "baz",
|
|
"non_existing": True,
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
mocker.patch("redis.Redis.set", return_value=None)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["old_value"] == "bar"
|
|
assert "value" not in json.loads(out)
|
|
assert json.loads(out)["msg"] == "Could not set key: foo. Key already present."
|
|
assert json.loads(out)["changed"] is False
|
|
assert json.loads(out)["failed"] is True
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
|
|
def test_redis_data_non_existing_key_xx(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_user": "root",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"value": "baz",
|
|
"existing": True,
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value=None)
|
|
mocker.patch("redis.Redis.set", return_value=None)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["old_value"] is None
|
|
assert "value" not in json.loads(out)
|
|
assert json.loads(out)["msg"] == "Could not set key: foo. Key not present."
|
|
assert json.loads(out)["changed"] is False
|
|
assert json.loads(out)["failed"] is True
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
|
|
def test_redis_data_delete_present_key(capfd, mocker):
|
|
with set_module_args(
|
|
{"login_host": "localhost", "login_user": "root", "login_password": "secret", "key": "foo", "state": "absent"}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
mocker.patch("redis.Redis.delete", return_value=1)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Deleted key: foo"
|
|
assert json.loads(out)["changed"] is True
|
|
|
|
|
|
@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
|
|
def test_redis_data_delete_absent_key(capfd, mocker):
|
|
with set_module_args(
|
|
{"login_host": "localhost", "login_user": "root", "login_password": "secret", "key": "foo", "state": "absent"}
|
|
):
|
|
mocker.patch("redis.Redis.delete", return_value=0)
|
|
mocker.patch("redis.Redis.get", return_value=None)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Key: foo not present"
|
|
assert json.loads(out)["changed"] is False
|
|
|
|
|
|
@pytest.mark.skipif(HAS_REDIS_USERNAME_OPTION, reason="Redis version > 3.4.0")
|
|
def test_redis_data_fail_username(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_user": "root",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"value": "baz",
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["failed"]
|
|
assert json.loads(out)["msg"] == "The option `username` in only supported with redis >= 3.4.0."
|
|
|
|
|
|
def test_redis_data_key_no_username(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"value": "baz",
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
mocker.patch("redis.Redis.set", return_value=True)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["old_value"] == "bar"
|
|
assert json.loads(out)["value"] == "baz"
|
|
assert json.loads(out)["msg"] == "Set key: foo"
|
|
assert json.loads(out)["changed"] is True
|
|
|
|
|
|
def test_redis_delete_key_no_username(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "absent",
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
mocker.patch("redis.Redis.delete", return_value=1)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Deleted key: foo"
|
|
assert json.loads(out)["changed"] is True
|
|
|
|
|
|
def test_redis_delete_key_non_existent_key(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "absent",
|
|
"_ansible_check_mode": False,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value=None)
|
|
mocker.patch("redis.Redis.delete", return_value=0)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Key: foo not present"
|
|
assert json.loads(out)["changed"] is False
|
|
|
|
|
|
def test_redis_set_key_check_mode_nochange(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "present",
|
|
"value": "bar",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Key foo already has desired value"
|
|
assert json.loads(out)["value"] == "bar"
|
|
assert not json.loads(out)["changed"]
|
|
assert json.loads(out)["old_value"] == "bar"
|
|
|
|
|
|
def test_redis_set_key_check_mode_delete_nx(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "present",
|
|
"value": "baz",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value=None)
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Set key: foo"
|
|
assert json.loads(out)["value"] == "baz"
|
|
assert json.loads(out)["old_value"] is None
|
|
|
|
|
|
def test_redis_set_key_check_mode_delete(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "present",
|
|
"value": "baz",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Set key: foo"
|
|
assert json.loads(out)["value"] == "baz"
|
|
assert json.loads(out)["old_value"] == "bar"
|
|
|
|
|
|
def test_redis_set_key_check_mode(capfd, mocker):
|
|
with set_module_args(
|
|
{
|
|
"login_host": "localhost",
|
|
"login_password": "secret",
|
|
"key": "foo",
|
|
"state": "present",
|
|
"value": "baz",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
mocker.patch("redis.Redis.get", return_value="bar")
|
|
with pytest.raises(SystemExit):
|
|
redis_data.main()
|
|
out, err = capfd.readouterr()
|
|
print(out)
|
|
assert not err
|
|
assert json.loads(out)["msg"] == "Set key: foo"
|
|
assert json.loads(out)["value"] == "baz"
|
|
assert json.loads(out)["old_value"] == "bar"
|