salt/tests/integration/modules/test_win_dns_client.py

# -*- coding: utf-8 -*-

# Import Python libs
from __future__ import absolute_import

import pytest

# Import Salt Testing libs
from tests.support.case import ModuleCase
from tests.support.unit import skipIf

# Import Salt libs
import salt.utils.platform


@skipIf(not salt.utils.platform.is_windows(), 'windows test only')
@pytest.mark.windows_whitelisted
class WinDNSTest(ModuleCase):
    '''
    Test for salt.modules.win_dns_client
    '''
    @pytest.mark.destructive_test
    def test_add_remove_dns(self):
        '''
        Test add and removing a dns server
        '''
        # Get a list of interfaces on the system
        interfaces = self.run_function('network.interfaces_names')
        skipIf(interfaces.count == 0, 'This test requires a network interface')

        interface = interfaces[0]
        dns = '8.8.8.8'
        # add dns server
        self.assertTrue(self.run_function('win_dns_client.add_dns', [dns, interface], index=42))

        srvs = self.run_function('win_dns_client.get_dns_servers', interface=interface)
        self.assertIn(dns, srvs)

        # remove dns server
        self.assertTrue(self.run_function('win_dns_client.rm_dns', [dns], interface=interface))

        srvs = self.run_function('win_dns_client.get_dns_servers', interface=interface)
        self.assertNotIn(dns, srvs)