diff options
Diffstat (limited to 'designate/tests/unit/backend/test_designate.py')
-rw-r--r-- | designate/tests/unit/backend/test_designate.py | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/designate/tests/unit/backend/test_designate.py b/designate/tests/unit/backend/test_designate.py index 41795a52..8b4468ab 100644 --- a/designate/tests/unit/backend/test_designate.py +++ b/designate/tests/unit/backend/test_designate.py @@ -13,32 +13,38 @@ # 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 unittest.mock import NonCallableMagicMock + +from unittest import mock from designateclient import exceptions from designateclient.v2 import client from oslo_log import log as logging +import oslotest.base from designate.backend import impl_designate +from designate import context from designate import objects -import designate.tests from designate.tests import fixtures LOG = logging.getLogger(__name__) -class DesignateBackendTestCase(designate.tests.TestCase): +class DesignateBackendTestCase(oslotest.base.BaseTestCase): def setUp(self): super(DesignateBackendTestCase, self).setUp() self.stdlog = fixtures.StandardLogging() self.useFixture(self.stdlog) + self.admin_context = mock.Mock() + mock.patch.object( + context.DesignateContext, 'get_admin_context', + return_value=self.admin_context).start() + self.zone = objects.Zone( id='e2bed4dc-9d01-11e4-89d3-123b93f75cba', name='example.com.', email='example@example.com', ) - self.target = { 'id': '4588652b-50e7-46b9-b688-a9bad40a873e', 'type': 'designate', @@ -61,8 +67,8 @@ class DesignateBackendTestCase(designate.tests.TestCase): ) # Mock client - self.client = NonCallableMagicMock() - zones = NonCallableMagicMock(spec_set=['create', 'delete']) + self.client = mock.NonCallableMagicMock() + zones = mock.NonCallableMagicMock(spec_set=['create', 'delete']) self.client.configure_mock(zones=zones) |