# -*- coding: utf-8 -*-
'''
    :synopsis: Unit Tests for Windows SNMP Module 'module.win_snmp'
    :platform: Windows
    :maturity: develop
    versionadded:: Carbon
'''

# Import Python Libs
from __future__ import absolute_import

# Import Salt Libs
from salt.modules import win_snmp

# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.helpers import ensure_in_syspath
from salttesting.mock import (
    MagicMock,
    patch,
    NO_MOCK,
    NO_MOCK_REASON,
)

ensure_in_syspath('../../')

# Globals
win_snmp.__salt__ = {}

# Make sure this module runs on Windows system
HAS_SNMP = win_snmp.__virtual__()

COMMUNITY_NAMES = {'TestCommunity': 'Read Create'}


@skipIf(not HAS_SNMP, 'This test case runs only on Windows systems')
@skipIf(NO_MOCK, NO_MOCK_REASON)
class WinSnmpTestCase(TestCase):
    '''
    Test cases for salt.modules.win_snmp
    '''

    def test_get_agent_service_types(self):
        '''
        Test - Get the sysServices types that can be configured.
        '''
        with patch.dict(win_snmp.__salt__):
            self.assertIsInstance(win_snmp.get_agent_service_types(), list)

    def test_get_permission_types(self):
        '''
        Test - Get the permission types that can be configured for communities.
        '''
        with patch.dict(win_snmp.__salt__):
            self.assertIsInstance(win_snmp.get_permission_types(), list)

    def test_get_auth_traps_enabled(self):
        '''
        Test - Determine whether the host is configured to send authentication traps.
        '''
        mock_value = MagicMock(return_value={'vdata': 1})
        with patch.dict(win_snmp.__salt__, {'reg.read_value': mock_value}):
            self.assertTrue(win_snmp.get_auth_traps_enabled())

    @patch('salt.modules.win_snmp.get_auth_traps_enabled',
           MagicMock(return_value=True))
    def test_set_auth_traps_enabled(self):
        '''
        Test - Manage the sending of authentication traps.
        '''
        mock_value = MagicMock(return_value=True)
        kwargs = {'status': True}
        with patch.dict(win_snmp.__salt__, {'reg.set_value': mock_value}):
            self.assertTrue(win_snmp.set_auth_traps_enabled(**kwargs))

    def test_get_community_names(self):
        '''
        Test - Get the current accepted SNMP community names and their permissions.
        '''
        mock_value = MagicMock(return_value=[{'vdata': 16,
                                              'vname': 'TestCommunity'}])
        with patch.dict(win_snmp.__salt__, {'reg.list_values': mock_value}):
            self.assertEqual(win_snmp.get_community_names(),
                             COMMUNITY_NAMES)

    @patch('salt.modules.win_snmp.get_community_names',
           MagicMock(return_value=COMMUNITY_NAMES))
    def test_set_community_names(self):
        '''
        Test - Manage the SNMP accepted community names and their permissions.
        '''
        mock_value = MagicMock(return_value=True)
        kwargs = {'communities': COMMUNITY_NAMES}
        with patch.dict(win_snmp.__salt__, {'reg.set_value': mock_value}):
            self.assertTrue(win_snmp.set_community_names(**kwargs))


if __name__ == '__main__':
    from integration import run_tests  # pylint: disable=import-error
    run_tests(WinSnmpTestCase, needs_daemon=False)

Add initial unit tests for win_snmp