salt/tests/unit/modules/test_napalm_snmp.py

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

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

import tests.support.napalm as napalm_test_support

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

import salt.modules.napalm_snmp as napalm_snmp  # NOQA
import salt.modules.napalm_network as napalm_network  # NOQA


class NapalmSnmpModuleTestCase(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,
                "net.load_template": napalm_network.load_template,
            }
        }

        return {napalm_snmp: module_globals, napalm_network: module_globals}

    def test_config(self):
        ret = napalm_snmp.config()
        assert ret["out"] == napalm_test_support.TEST_SNMP_INFO

    def test_remove_config(self):
        ret = napalm_snmp.remove_config("1.2.3.4")
        assert ret["result"] is False

    def test_update_config(self):
        ret = napalm_snmp.update_config("1.2.3.4")
        assert ret["result"] is False