2b8282bb1e
Now that we no longer support py27, we can use the standard library unittest.mock module instead of the third party mock lib. Change-Id: Ib83e626f51e6e01898bb7931c1660a1a548500f3 Signed-off-by: Sean McGinnis <sean.mcginnis@gmail.com>
258 lines
9.4 KiB
Python
258 lines
9.4 KiB
Python
# Copyright 2011 OpenStack Foundation
|
|
# Copyright 2014 Rackspace Hosting
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import json
|
|
import testtools
|
|
from unittest import mock
|
|
|
|
from troveclient import base
|
|
from troveclient.v1 import configurations
|
|
from troveclient.v1 import management
|
|
|
|
"""
|
|
Unit tests for configurations.py
|
|
"""
|
|
|
|
|
|
class ConfigurationTest(testtools.TestCase):
|
|
def setUp(self):
|
|
super(ConfigurationTest, self).setUp()
|
|
self.orig__init = configurations.Configuration.__init__
|
|
configurations.Configuration.__init__ = mock.Mock(return_value=None)
|
|
self.configuration = configurations.Configuration()
|
|
|
|
def tearDown(self):
|
|
super(ConfigurationTest, self).tearDown()
|
|
configurations.Configuration.__init__ = self.orig__init
|
|
|
|
def test___repr__(self):
|
|
self.configuration.name = "config-1"
|
|
self.assertEqual('<Configuration: config-1>',
|
|
self.configuration.__repr__())
|
|
|
|
|
|
class ConfigurationsTest(testtools.TestCase):
|
|
def setUp(self):
|
|
super(ConfigurationsTest, self).setUp()
|
|
self.orig__init = configurations.Configurations.__init__
|
|
configurations.Configurations.__init__ = mock.Mock(return_value=None)
|
|
self.configurations = configurations.Configurations()
|
|
self.configurations.api = mock.Mock()
|
|
self.configurations.api.client = mock.Mock()
|
|
|
|
self.orig_base_getid = base.getid
|
|
base.getid = mock.Mock(return_value="configuration1")
|
|
|
|
def tearDown(self):
|
|
super(ConfigurationsTest, self).tearDown()
|
|
configurations.Configurations.__init__ = self.orig__init
|
|
base.getid = self.orig_base_getid
|
|
|
|
def _get_mock_method(self):
|
|
self._resp = mock.Mock()
|
|
self._body = None
|
|
self._url = None
|
|
|
|
def side_effect_func(url, body=None):
|
|
self._body = body
|
|
self._url = url
|
|
return (self._resp, body)
|
|
|
|
return mock.Mock(side_effect=side_effect_func)
|
|
|
|
def test_create(self):
|
|
self.configurations.api.client.post = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
config = '{"test":12}'
|
|
|
|
self.configurations.create('config1', config, 'test description')
|
|
self.assertEqual('/configurations', self._url)
|
|
expected = {
|
|
'description': 'test description',
|
|
'name': 'config1',
|
|
'values': json.loads(config)
|
|
}
|
|
self.assertEqual({"configuration": expected}, self._body)
|
|
|
|
def test_delete(self):
|
|
self.configurations.api.client.delete = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
self.configurations.delete(27)
|
|
self.assertEqual('/configurations/configuration1', self._url)
|
|
self._resp.status_code = 500
|
|
self.assertRaises(Exception, self.configurations.delete, 34)
|
|
|
|
def test_list(self):
|
|
page_mock = mock.Mock()
|
|
self.configurations._paginated = page_mock
|
|
limit = "test-limit"
|
|
marker = "test-marker"
|
|
self.configurations.list(limit, marker)
|
|
page_mock.assert_called_with("/configurations", "configurations",
|
|
limit, marker)
|
|
|
|
def test_get(self):
|
|
def side_effect_func(path, config):
|
|
return path, config
|
|
|
|
self.configurations._get = mock.Mock(side_effect=side_effect_func)
|
|
self.assertEqual(('/configurations/configuration1', 'configuration'),
|
|
self.configurations.get(123))
|
|
|
|
def test_instances(self):
|
|
page_mock = mock.Mock()
|
|
self.configurations._paginated = page_mock
|
|
limit = "test-limit"
|
|
marker = "test-marker"
|
|
configuration = "configuration1"
|
|
self.configurations.instances(configuration, limit, marker)
|
|
page_mock.assert_called_with(
|
|
"/configurations/" + configuration + "/instances",
|
|
"instances", limit, marker)
|
|
|
|
def test_update(self):
|
|
self.configurations.api.client.put = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
config = '{"test":12}'
|
|
self.configurations.update(27, config)
|
|
self.assertEqual('/configurations/configuration1', self._url)
|
|
self._resp.status_code = 500
|
|
self.assertRaises(Exception, self.configurations.update, 34)
|
|
|
|
def test_edit(self):
|
|
self.configurations.api.client.patch = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
config = '{"test":12}'
|
|
self.configurations.edit(27, config)
|
|
self.assertEqual('/configurations/configuration1', self._url)
|
|
self._resp.status_code = 500
|
|
self.assertRaises(Exception, self.configurations.edit, 34)
|
|
|
|
|
|
class ConfigurationParametersTest(testtools.TestCase):
|
|
def setUp(self):
|
|
super(ConfigurationParametersTest, self).setUp()
|
|
self.orig__init = configurations.ConfigurationParameters.__init__
|
|
configurations.ConfigurationParameters.__init__ = mock.Mock(
|
|
return_value=None)
|
|
self.config_params = configurations.ConfigurationParameters()
|
|
self.config_params.api = mock.Mock()
|
|
self.config_params.api.client = mock.Mock()
|
|
|
|
def tearDown(self):
|
|
super(ConfigurationParametersTest, self).tearDown()
|
|
configurations.ConfigurationParameters.__init__ = self.orig__init
|
|
|
|
def test_list_parameters(self):
|
|
def side_effect_func(path, config):
|
|
return path
|
|
|
|
self.config_params._list = mock.Mock(side_effect=side_effect_func)
|
|
self.assertEqual('/datastores/datastore/versions/version/parameters',
|
|
self.config_params.parameters('datastore', 'version'))
|
|
|
|
def test_get_parameter(self):
|
|
def side_effect_func(path):
|
|
return path
|
|
|
|
self.config_params._get = mock.Mock(side_effect=side_effect_func)
|
|
self.assertEqual(
|
|
'/datastores/datastore/versions/version/parameters/key',
|
|
self.config_params.get_parameter('datastore', 'version',
|
|
'key')
|
|
)
|
|
|
|
def test_list_parameters_by_version(self):
|
|
def side_effect_func(path, config):
|
|
return path
|
|
|
|
self.config_params._list = mock.Mock(side_effect=side_effect_func)
|
|
self.assertEqual('/datastores/versions/version/parameters',
|
|
self.config_params.parameters_by_version('version'))
|
|
|
|
def test_get_parameter_by_version(self):
|
|
def side_effect_func(path):
|
|
return path
|
|
|
|
self.config_params._get = mock.Mock(side_effect=side_effect_func)
|
|
self.assertEqual('/datastores/versions/version/parameters/key',
|
|
self.config_params.get_parameter_by_version(
|
|
'version', 'key'))
|
|
|
|
|
|
class MgmtConfigurationParametersTest(testtools.TestCase):
|
|
|
|
def setUp(self):
|
|
super(MgmtConfigurationParametersTest, self).setUp()
|
|
self.orig__init = management.MgmtConfigurationParameters.__init__
|
|
management.MgmtConfigurationParameters.__init__ = mock.Mock(
|
|
return_value=None)
|
|
self.config_params = management.MgmtConfigurationParameters()
|
|
self.config_params.api = mock.Mock()
|
|
self.config_params.api.client = mock.Mock()
|
|
|
|
def tearDown(self):
|
|
super(MgmtConfigurationParametersTest, self).tearDown()
|
|
management.MgmtConfigurationParameters.__init__ = self.orig__init
|
|
|
|
def _get_mock_method(self):
|
|
self._resp = mock.Mock()
|
|
self._body = None
|
|
self._url = None
|
|
|
|
def side_effect_func(url, body=None):
|
|
self._body = body
|
|
self._url = url
|
|
return (self._resp, body)
|
|
|
|
return mock.Mock(side_effect=side_effect_func)
|
|
|
|
def test_create(self):
|
|
self.config_params.api.client.post = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
|
|
self.config_params.create('id', 'config_name', 1, 'string')
|
|
self.assertEqual('/mgmt/datastores/versions/id/parameters', self._url)
|
|
expected = {
|
|
"name": "config_name",
|
|
"data_type": "string",
|
|
"restart_required": 1
|
|
}
|
|
self.assertEqual({"configuration-parameter": expected}, self._body)
|
|
|
|
def test_modify(self):
|
|
self.config_params.api.client.put = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
|
|
self.config_params.modify('id', 'config_name', '1', 'string')
|
|
self.assertEqual('/mgmt/datastores/versions/id/parameters/config_name',
|
|
self._url)
|
|
expected = {
|
|
"name": "config_name",
|
|
"data_type": "string",
|
|
"restart_required": 1
|
|
}
|
|
self.assertEqual({"configuration-parameter": expected}, self._body)
|
|
|
|
def test_delete(self):
|
|
self.config_params.api.client.delete = self._get_mock_method()
|
|
self._resp.status_code = 200
|
|
|
|
self.config_params.delete('id', 'param_id')
|
|
self.assertEqual('/mgmt/datastores/versions/id/parameters/param_id',
|
|
self._url)
|
|
self.assertIsNone(self._body)
|