sahara-dashboard/sahara_dashboard/test/helpers.py

66 lines
1.9 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.
import mock
from saharaclient import client as sahara_client
from openstack_dashboard.test import helpers
from sahara_dashboard import api
from sahara_dashboard.test.test_data import utils
def create_stubs(stubs_to_create={}):
return helpers.create_stubs(stubs_to_create)
class SaharaTestsMixin(object):
def _setup_test_data(self):
super(SaharaTestsMixin, self)._setup_test_data()
utils.load_test_data(self)
self.policy_patcher = mock.patch(
'openstack_auth.policy.check', lambda action, request: True)
self.policy_check = self.policy_patcher.start()
class TestCase(SaharaTestsMixin, helpers.TestCase):
use_mox = True
pass
class BaseAdminViewTests(SaharaTestsMixin, helpers.TestCase):
pass
class SaharaAPITestCase(helpers.APITestCase):
use_mox = True
def setUp(self):
super(SaharaAPITestCase, self).setUp()
self._original_saharaclient = api.sahara.client
api.sahara.client = lambda request: self.stub_saharaclient()
def tearDown(self):
super(SaharaAPITestCase, self).tearDown()
api.sahara.client = self._original_saharaclient
def stub_saharaclient(self):
if not hasattr(self, "saharaclient"):
self.mox.StubOutWithMock(sahara_client, 'Client')
self.saharaclient = self.mox.CreateMock(sahara_client.Client)
return self.saharaclient