trove-dashboard/trove_dashboard/test/helpers.py
David Lyle bac61fbfd2 Completing the plugin
Change-Id: I410aaac24b76e63a82d6c50364e46d54ead03b05
2015-12-18 15:53:31 -07:00

58 lines
1.7 KiB
Python

#
# 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.
from troveclient import client as trove_client
from openstack_dashboard.test import helpers
from trove_dashboard import api
from trove_dashboard.test.test_data import utils
def create_stubs(stubs_to_create={}):
return helpers.create_stubs(stubs_to_create)
class TroveTestsMixin(object):
def _setup_test_data(self):
super(TroveTestsMixin, self)._setup_test_data()
utils.load_test_data(self)
class TestCase(TroveTestsMixin, helpers.TestCase):
pass
class BaseAdminViewTests(TroveTestsMixin, helpers.TestCase):
pass
class TroveAPITestCase(helpers.APITestCase):
def setUp(self):
super(TroveAPITestCase, self).setUp()
self._original_troveclient = api.trove.client
api.trove.client = lambda request: self.stub_troveclient()
def tearDown(self):
super(TroveAPITestCase, self).tearDown()
api.trove.client = self._original_troveclient
def stub_troveclient(self):
if not hasattr(self, "troveclient"):
self.mox.StubOutWithMock(trove_client, 'Client')
self.troveclient = self.mox.CreateMock(trove_client.Client)
return self.troveclient