salt/tests/unit/utils/test_win_osinfo.py

# -*- coding: utf-8 -*-
'''
:codeauthor: Shane Lee <slee@saltstack.com>
'''
# Import Python Libs
from __future__ import absolute_import, unicode_literals, print_function
import sys

# Import 3rd Party Libs
from salt.ext import six

# Import Salt Testing Libs
from tests.support.unit import TestCase, skipIf

# Import Salt Libs
import salt.utils.win_osinfo as win_osinfo
import salt.utils.platform


@skipIf(not salt.utils.platform.is_windows(), 'Requires Windows')
class WinOsInfo(TestCase):
    '''
    Test cases for salt/utils/win_osinfo.py
    '''
    def test_get_os_version_info(self):
        sys_info = sys.getwindowsversion()
        get_info = win_osinfo.get_os_version_info()
        self.assertEqual(sys_info.major, int(get_info['MajorVersion']))
        self.assertEqual(sys_info.minor, int(get_info['MinorVersion']))
        self.assertEqual(sys_info.platform, int(get_info['PlatformID']))
        self.assertEqual(sys_info.build, int(get_info['BuildNumber']))
        # Platform ID is the reason for this function
        # Since we can't get the actual value another way, we will just check
        # that it exists and is a number
        self.assertIn('PlatformID', get_info)
        self.assertTrue(isinstance(get_info['BuildNumber'], six.integer_types))

    def test_get_join_info(self):
        join_info = win_osinfo.get_join_info()
        self.assertIn('Domain', join_info)
        self.assertIn('DomainType', join_info)
        valid_types = ['Unknown', 'Unjoined', 'Workgroup', 'Domain']
        self.assertIn(join_info['DomainType'], valid_types)