salt/tests/unit/beacons/test_network_info.py

# coding: utf-8

# Python libs
from __future__ import absolute_import
from collections import namedtuple

# Salt testing libs
from tests.support.unit import TestCase
from tests.support.mock import patch, MagicMock
from tests.support.mixins import LoaderModuleMockMixin

# Salt libs
import salt.beacons.network_info as network_info

import logging
log = logging.getLogger(__name__)

STUB_NET_IO_COUNTERS = {'eth0': namedtuple('snetio',
                                           'bytes_sent bytes_recv \
                                           packets_sent packets_recv \
                                           errin errout \
                                           dropin \
                                           dropout')(93662618, 914626664,
                                                     465694, 903802, 0,
                                                     0, 0, 0)}


class NetworkInfoBeaconTestCase(TestCase, LoaderModuleMockMixin):
    '''
    Test case for salt.beacons.network_info
    '''

    def setup_loader_modules(self):
        return {
            network_info: {
                '__context__': {},
                '__salt__': {},
            }
        }

    def test_non_list_config(self):
        config = {}

        ret = network_info.validate(config)

        self.assertEqual(ret, (False, 'Configuration for network_info beacon'
                                      ' must be a list.'))

    def test_empty_config(self):
        config = [{}]

        ret = network_info.validate(config)

        self.assertEqual(ret, (True, 'Valid beacon configuration'))

    def test_network_info_equal(self):
        with patch('salt.utils.psutil_compat.net_io_counters',
                   MagicMock(return_value=STUB_NET_IO_COUNTERS)):
            config = [{'interfaces': {'eth0': {'type': 'equal',
                                               'bytes_sent': 914626664,
                                               'bytes_recv': 93662618,
                                               'packets_sent': 465694,
                                               'packets_recv': 903802,
                                               'errin': 0,
                                               'errout': 0,
                                               'dropin': 0,
                                               'dropout': 0}}}]

            ret = network_info.validate(config)

            self.assertEqual(ret, (True, 'Valid beacon configuration'))

            _expected_return = [{'interface': 'eth0',
                                 'network_info': {'bytes_recv': 914626664,
                                                  'bytes_sent': 93662618,
                                                  'dropin': 0,
                                                  'dropout': 0,
                                                  'errin': 0,
                                                  'errout': 0,
                                                  'packets_recv': 903802,
                                                  'packets_sent': 465694}}]

            ret = network_info.beacon(config)
            self.assertEqual(ret, _expected_return)

    def test_network_info_greater_than(self):
        with patch('salt.utils.psutil_compat.net_io_counters',
                   MagicMock(return_value=STUB_NET_IO_COUNTERS)):
            config = [{'interfaces': {'eth0': {'type': 'greater',
                                               'bytes_sent': 100000,
                                               'bytes_recv': 100000,
                                               'packets_sent': 100000,
                                               'packets_recv': 100000,
                                               'errin': 0,
                                               'errout': 0,
                                               'dropin': 0,
                                               'dropout': 0}}}]

            ret = network_info.validate(config)

            self.assertEqual(ret, (True, 'Valid beacon configuration'))

            _expected_return = [{'interface': 'eth0',
                                 'network_info': {'bytes_recv': 914626664,
                                                  'bytes_sent': 93662618,
                                                  'dropin': 0,
                                                  'dropout': 0,
                                                  'errin': 0,
                                                  'errout': 0,
                                                  'packets_recv': 903802,
                                                  'packets_sent': 465694}}]

            ret = network_info.beacon(config)
            self.assertEqual(ret, _expected_return)