salt/tests/unit/grains/test_napalm.py

# -*- coding: utf-8 -*-
'''
    :codeauthor: :email:`Anthony Shaw <anthonyshaw@apache.org>`
'''

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

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

import tests.support.napalm as napalm_test_support
import salt.grains.napalm as napalm_grains  # NOQA
import salt.proxy.napalm as napalm_proxy  # NOQA
napalm_grains.salt.utils.napalm.is_proxy = MagicMock(return_value=True)

TEST_DEVICE_CACHE = {
    'DRIVER': napalm_test_support.MockNapalmDevice(),
    'DRIVER_NAME': 'cisco',
    'OS_VERSION': '1.2.3',
    'HOSTNAME': 'test-device.com',
    'USERNAME': 'admin'
}

TEST_CACHE = {
    'result': True,
    'out': napalm_test_support.TEST_FACTS
}


@patch('salt.grains.napalm.DEVICE_CACHE', TEST_DEVICE_CACHE)
@patch('salt.grains.napalm.GRAINS_CACHE', TEST_CACHE)
class NapalmGrainsTestCase(TestCase, LoaderModuleMockMixin):

    def setup_loader_modules(self):
        module_globals = {
            '__salt__': {
                'config.option': MagicMock(return_value={
                    'test': {
                        'driver': 'test',
                        'key': '2orgk34kgk34g'
                    }
                }),
                'file.file_exists': napalm_test_support.true,
                'file.join': napalm_test_support.join,
                'file.get_managed': napalm_test_support.get_managed_file,
                'random.hash': napalm_test_support.random_hash,
            }
        }

        return {napalm_grains: module_globals}

    def test_os(self):
        ret = napalm_grains.getos(proxy=napalm_proxy)
        assert ret['os'] == 'cisco'

    def test_os_version(self):
        ret = napalm_grains.version(proxy=napalm_proxy)
        assert ret['version'] == '1.2.3'

    def test_model(self):
        ret = napalm_grains.model(proxy=napalm_proxy)
        assert ret['model'] == 'test_model'

    def test_serial(self):
        ret = napalm_grains.serial(proxy=napalm_proxy)
        assert ret['serial'] == '123456'

    def test_vendor(self):
        ret = napalm_grains.vendor(proxy=napalm_proxy)
        assert ret['vendor'] == 'cisco'

    def test_uptime(self):
        ret = napalm_grains.uptime(proxy=napalm_proxy)
        assert ret['uptime'] == 'Forever'

    def test_interfaces(self):
        ret = napalm_grains.interfaces(proxy=napalm_proxy)
        assert ret['interfaces'] == napalm_test_support.TEST_INTERFACES

    def test_username(self):
        ret = napalm_grains.username(proxy=napalm_proxy)
        assert ret['username'] == 'admin'

    def test_hostname(self):
        ret = napalm_grains.hostname(proxy=napalm_proxy)
        assert ret['hostname'] == 'test-device.com'

    def test_host(self):
        ret = napalm_grains.host(proxy=napalm_proxy)
        assert ret['host'] == 'test-device.com'