¡@

Home 

OpenStack Study: mockpatch.py

OpenStack Index

**** CubicPower OpenStack Study ****

# Copyright 2010 United States Government as represented by the

# Administrator of the National Aeronautics and Space Administration.

# Copyright 2013 Hewlett-Packard Development Company, L.P.

# 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 fixtures

import mock

**** CubicPower OpenStack Study ****

class PatchObject(fixtures.Fixture):

"""Deal with code around mock."""

**** CubicPower OpenStack Study ****

    def __init__(self, obj, attr, **kwargs):

        self.obj = obj

        self.attr = attr

        self.kwargs = kwargs

**** CubicPower OpenStack Study ****

    def setUp(self):

        super(PatchObject, self).setUp()

        _p = mock.patch.object(self.obj, self.attr, **self.kwargs)

        self.mock = _p.start()

        self.addCleanup(_p.stop)

**** CubicPower OpenStack Study ****

class Patch(fixtures.Fixture):

"""Deal with code around mock.patch."""

**** CubicPower OpenStack Study ****

    def __init__(self, obj, **kwargs):

        self.obj = obj

        self.kwargs = kwargs

**** CubicPower OpenStack Study ****

    def setUp(self):

        super(Patch, self).setUp()

        _p = mock.patch(self.obj, **self.kwargs)

        self.mock = _p.start()

        self.addCleanup(_p.stop)