def setUp(self):
super(ManagerListTest, self).setUp()
@contextlib.contextmanager
def completion_cache_mock(*arg, **kwargs):
yield
self.manager = base.Manager()
self.manager.api = Mock()
self.manager.api.client = Mock()
self.response_key = "response_key"
self.data_p = ["p1", "p2"]
self.body_p = {self.response_key: self.data_p}
self.url_p = "test_url_post"
self.manager.api.client.post = Mock(return_value=(self.url_p,
self.body_p))
self.data_g = ["g1", "g2", "g3"]
self.body_g = {self.response_key: self.data_g}
self.url_g = "test_url_get"
self.manager.api.client.get = Mock(return_value=(self.url_g,
self.body_g))
mock = Mock()
mock.side_effect = completion_cache_mock
self.manager.completion_cache = mock
def test__create(self):
manager = base.Manager()
manager.api = Mock()
manager.api.client = Mock()
response_key = "response_key"
data_ = "test-data"
body_ = {response_key: data_}
url_ = "test_url_post"
manager.api.client.post = Mock(return_value=(url_, body_))
return_raw = True
r = manager._create(url_, body_, response_key, return_raw)
self.assertEqual(data_, r)
return_raw = False
@contextlib.contextmanager
def completion_cache_mock(*arg, **kwargs):
yield
mock = Mock()
mock.side_effect = completion_cache_mock
manager.completion_cache = mock
manager.resource_class = Mock(return_value="test-class")
r = manager._create(url_, body_, response_key, return_raw)
self.assertEqual("test-class", r)
def _get_mock(self):
manager = base.Manager()
manager.api = Mock()
manager.api.client = Mock()
def side_effect_func(self, body, loaded=True):
return body
manager.resource_class = Mock(side_effect=side_effect_func)
return manager
def test_write_to_completion_cache(self):
manager = base.Manager()
# no cache object, nothing should happen
manager.write_to_completion_cache("non-exist", "val")
def side_effect_func(val):
return val
manager._mock_cache = Mock()
manager._mock_cache.write = Mock(return_value=None)
manager.write_to_completion_cache("mock", "val")
self.assertEqual(1, manager._mock_cache.write.call_count)
def test_completion_cache(self):
manager = base.Manager()
# handling exceptions
mode = "w"
cache_type = "unittest"
obj_class = Mock
with manager.completion_cache(cache_type, obj_class, mode):
pass
os.makedirs = Mock(side_effect=OSError)
with manager.completion_cache(cache_type, obj_class, mode):
pass
def test___init__(self):
api = Mock()
base.Manager.__init__ = self.orig__init
manager = base.Manager(api)
self.assertEqual(api, manager.api)
def get_mock_mng_api_client(self):
manager = base.Manager()
manager.api = Mock()
manager.api.client = Mock()
return manager