salt/tests/unit/test_transport.py

# -*- coding: utf-8 -*-

# Import python libs
from __future__ import absolute_import
import logging

from salt.transport import MessageClientPool

# Import Salt Testing libs
from tests.support.unit import TestCase

log = logging.getLogger(__name__)


class MessageClientPoolTest(TestCase):

    class MockClass(object):
        def __init__(self, *args, **kwargs):
            self.args = args
            self.kwargs = kwargs

    def test_init(self):
        opts = {'sock_pool_size': 10}
        args = (0,)
        kwargs = {'kwarg': 1}
        message_client_pool = MessageClientPool(self.MockClass, opts, args=args, kwargs=kwargs)
        self.assertEqual(opts['sock_pool_size'], len(message_client_pool.message_clients))
        for message_client in message_client_pool.message_clients:
            self.assertEqual(message_client.args, args)
            self.assertEqual(message_client.kwargs, kwargs)

    def test_init_without_config(self):
        opts = {}
        args = (0,)
        kwargs = {'kwarg': 1}
        message_client_pool = MessageClientPool(self.MockClass, opts, args=args, kwargs=kwargs)
        # The size of pool is set as 1 by the MessageClientPool init method.
        self.assertEqual(1, len(message_client_pool.message_clients))
        for message_client in message_client_pool.message_clients:
            self.assertEqual(message_client.args, args)
            self.assertEqual(message_client.kwargs, kwargs)

    def test_init_less_than_one(self):
        opts = {'sock_pool_size': -1}
        args = (0,)
        kwargs = {'kwarg': 1}
        message_client_pool = MessageClientPool(self.MockClass, opts, args=args, kwargs=kwargs)
        # The size of pool is set as 1 by the MessageClientPool init method.
        self.assertEqual(1, len(message_client_pool.message_clients))
        for message_client in message_client_pool.message_clients:
            self.assertEqual(message_client.args, args)
            self.assertEqual(message_client.kwargs, kwargs)