community.general/tests/unit/plugins/modules/test_redis_data_info.py

#
# 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 pytest
import json
from redis import __version__

from ansible_collections.community.general.plugins.modules import (
    redis_data_info)
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import set_module_args


HAS_REDIS_USERNAME_OPTION = True
if tuple(map(int, __version__.split('.'))) < (3, 4, 0):
    HAS_REDIS_USERNAME_OPTION = False


def test_redis_data_info_without_arguments(capfd):
    with set_module_args({}):
        with pytest.raises(SystemExit):
            redis_data_info.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_info_existing_key(capfd, mocker):
    with set_module_args({
        'login_host': 'localhost',
        'login_user': 'root',
        'login_password': 'secret',
        'key': 'foo',
        '_ansible_check_mode': False
    }):
        mocker.patch('redis.Redis.get', return_value='bar')
        with pytest.raises(SystemExit):
            redis_data_info.main()
    out, err = capfd.readouterr()
    print(out)
    assert not err
    assert json.loads(out)['exists']
    assert json.loads(out)['value'] == 'bar'


@pytest.mark.skipif(not HAS_REDIS_USERNAME_OPTION, reason="Redis version < 3.4.0")
def test_redis_data_info_absent_key(capfd, mocker):
    with set_module_args({
        'login_host': 'localhost',
        'login_user': 'root',
        'login_password': 'secret',
        'key': 'foo',
        '_ansible_check_mode': False
    }):
        mocker.patch('redis.Redis.get', return_value=None)
        with pytest.raises(SystemExit):
            redis_data_info.main()
    out, err = capfd.readouterr()
    print(out)
    assert not err
    assert not json.loads(out)['exists']
    assert 'value' not in json.loads(out)


@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',
        '_ansible_check_mode': False
    }):
        with pytest.raises(SystemExit):
            redis_data_info.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.'


@pytest.mark.skipif(HAS_REDIS_USERNAME_OPTION, reason="Redis version > 3.4.0")
def test_redis_data_info_absent_key_no_username(capfd, mocker):
    with set_module_args({
        'login_host': 'localhost',
        'login_password': 'secret',
        'key': 'foo',
        '_ansible_check_mode': False
    }):
        mocker.patch('redis.Redis.get', return_value=None)
        with pytest.raises(SystemExit):
            redis_data_info.main()
    out, err = capfd.readouterr()
    print(out)
    assert not err
    assert not json.loads(out)['exists']
    assert 'value' not in json.loads(out)


@pytest.mark.skipif(HAS_REDIS_USERNAME_OPTION, reason="Redis version > 3.4.0")
def test_redis_data_info_existing_key_no_username(capfd, mocker):
    with set_module_args({
        'login_host': 'localhost',
        'login_password': 'secret',
        'key': 'foo',
        '_ansible_check_mode': False
    }):
        mocker.patch('redis.Redis.get', return_value='bar')
        with pytest.raises(SystemExit):
            redis_data_info.main()
    out, err = capfd.readouterr()
    print(out)
    assert not err
    assert json.loads(out)['exists']
    assert json.loads(out)['value'] == 'bar'