mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
112 lines
3.8 KiB
Python
112 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Anthony Shaw <anthonyshaw@apache.org>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# 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_storage as libcloud_storage
|
|
|
|
from libcloud.storage.base import Container, BaseDriver, Object
|
|
|
|
|
|
class MockStorageDriver(BaseDriver):
|
|
def __init__(self):
|
|
self._TEST_CONTAINER = Container(name='test_container', extra={}, driver=self)
|
|
self._TEST_OBJECT = Object(name='test_obj',
|
|
size=1234,
|
|
hash='123sdfsdf',
|
|
extra={},
|
|
meta_data={'key': 'value'},
|
|
container=self._TEST_CONTAINER,
|
|
driver=self)
|
|
|
|
def list_containers(self):
|
|
return [self._TEST_CONTAINER]
|
|
|
|
def get_container(self, container_name):
|
|
assert container_name == 'test_container'
|
|
return self._TEST_CONTAINER
|
|
|
|
def list_container_objects(self, container):
|
|
assert container.name == 'test_container'
|
|
return [self._TEST_OBJECT]
|
|
|
|
def create_container(self, container_name):
|
|
assert container_name == 'new_test_container'
|
|
return self._TEST_CONTAINER
|
|
|
|
def get_container_object(self, container_name, object_name):
|
|
assert container_name == 'test_container'
|
|
assert object_name == 'test_obj'
|
|
return self._TEST_OBJECT
|
|
|
|
|
|
def get_mock_driver():
|
|
return MockStorageDriver()
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
@patch('salt.modules.libcloud_storage._get_driver',
|
|
MagicMock(return_value=MockStorageDriver()))
|
|
class LibcloudStorageModuleTestCase(TestCase, LoaderModuleMockMixin):
|
|
|
|
def setup_loader_modules(self):
|
|
module_globals = {
|
|
'__salt__': {
|
|
'config.option': MagicMock(return_value={
|
|
'test': {
|
|
'driver': 'test',
|
|
'key': '2orgk34kgk34g'
|
|
}
|
|
})
|
|
}
|
|
}
|
|
if libcloud_storage.HAS_LIBCLOUD is False:
|
|
module_globals['sys.modules'] = {'libcloud': MagicMock()}
|
|
|
|
return {libcloud_storage: module_globals}
|
|
|
|
def test_module_creation(self):
|
|
client = libcloud_storage._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_storage.__init__(None)
|
|
dunder.assert_called_with('salt.modules.libcloud_storage')
|
|
|
|
def test_list_containers(self):
|
|
containers = libcloud_storage.list_containers('test')
|
|
self.assertEqual(len(containers), 1)
|
|
self.assertEqual(containers[0]['name'], 'test_container')
|
|
|
|
def test_list_container_objects(self):
|
|
objects = libcloud_storage.list_container_objects('test_container', 'test')
|
|
self.assertEqual(len(objects), 1)
|
|
self.assertEqual(objects[0]['name'], 'test_obj')
|
|
self.assertEqual(objects[0]['size'], 1234)
|
|
|
|
def test_create_container(self):
|
|
container = libcloud_storage.create_container('new_test_container', 'test')
|
|
self.assertEqual(container['name'], 'test_container')
|
|
|
|
def test_get_container(self):
|
|
container = libcloud_storage.get_container('test_container', 'test')
|
|
self.assertEqual(container['name'], 'test_container')
|
|
|
|
def test_get_container_object(self):
|
|
obj = libcloud_storage.get_container_object('test_container', 'test_obj', 'test')
|
|
self.assertEqual(obj['name'], 'test_obj')
|
|
self.assertEqual(obj['size'], 1234)
|