mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: Anthony Shaw <anthonyshaw@apache.org>
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import, unicode_literals, print_function
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
patch,
|
|
MagicMock,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
import salt.modules.libcloud_dns as libcloud_dns
|
|
|
|
|
|
class MockDNSDriver(object):
|
|
def __init__(self):
|
|
pass
|
|
|
|
|
|
def get_mock_driver():
|
|
return MockDNSDriver()
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
@skipIf(not libcloud_dns.HAS_LIBCLOUD, 'No libcloud installed')
|
|
class LibcloudDnsModuleTestCase(TestCase, LoaderModuleMockMixin):
|
|
|
|
def setup_loader_modules(self):
|
|
module_globals = {
|
|
'_get_driver': MagicMock(return_value=MockDNSDriver()),
|
|
'__salt__': {
|
|
'config.option': MagicMock(return_value={
|
|
'test': {
|
|
'driver': 'test',
|
|
'key': '2orgk34kgk34g'
|
|
}
|
|
})
|
|
}
|
|
}
|
|
if libcloud_dns.HAS_LIBCLOUD is False:
|
|
module_globals['sys.modules'] = {'libcloud': MagicMock()}
|
|
|
|
return {libcloud_dns: module_globals}
|
|
|
|
def test_module_creation(self):
|
|
client = libcloud_dns._get_driver('test')
|
|
self.assertFalse(client is None)
|
|
|
|
def test_init(self):
|
|
with patch('salt.utils.compat.pack_dunder', return_value=False) as dunder:
|
|
libcloud_dns.__init__(None)
|
|
dunder.assert_called_with('salt.modules.libcloud_dns')
|