salt/tests/unit/modules/test_ret.py

# -*- coding: utf-8 -*-
"""
    :codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
"""

# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals

import salt.loader

# Import Salt Libs
import salt.modules.ret as ret

# Import Salt Testing Libs
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.mock import MagicMock, patch
from tests.support.unit import TestCase


class RetTestCase(TestCase, LoaderModuleMockMixin):
    """
    Test cases for salt.modules.ret
    """

    def setup_loader_modules(self):
        return {ret: {}}

    # 'get_jid' function tests: 1

    def test_get_jid(self):
        """
        Test if it return the information for a specified job id
        """
        mock_ret = MagicMock(return_value="DB")
        with patch.object(
            salt.loader,
            "returners",
            MagicMock(return_value={"redis.get_jid": mock_ret}),
        ):
            self.assertEqual(ret.get_jid("redis", "net"), "DB")

    # 'get_fun' function tests: 1

    def test_get_fun(self):
        """
        Test if it return info about last time fun was called on each minion
        """
        mock_ret = MagicMock(return_value="DB")
        with patch.object(
            salt.loader,
            "returners",
            MagicMock(return_value={"mysql.get_fun": mock_ret}),
        ):
            self.assertEqual(ret.get_fun("mysql", "net"), "DB")

    # 'get_jids' function tests: 1

    def test_get_jids(self):
        """
        Test if it return a list of all job ids
        """
        mock_ret = MagicMock(return_value="DB")
        with patch.object(
            salt.loader,
            "returners",
            MagicMock(return_value={"mysql.get_jids": mock_ret}),
        ):
            self.assertEqual(ret.get_jids("mysql"), "DB")

    # 'get_minions' function tests: 1

    def test_get_minions(self):
        """
        Test if it return a list of all minions
        """
        mock_ret = MagicMock(return_value="DB")
        with patch.object(
            salt.loader,
            "returners",
            MagicMock(return_value={"mysql.get_minions": mock_ret}),
        ):
            self.assertEqual(ret.get_minions("mysql"), "DB")