1# vim: tabstop=4 shiftwidth=4 softtabstop=4 2 3# Copyright 2013 Hewlett-Packard Development Company, L.P. 4# All Rights Reserved. 5# 6# Licensed under the Apache License, Version 2.0 (the "License"); you may 7# not use this file except in compliance with the License. You may obtain 8# a copy of the License at 9# 10# http://www.apache.org/licenses/LICENSE-2.0 11# 12# Unless required by applicable law or agreed to in writing, software 13# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 14# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 15# License for the specific language governing permissions and limitations 16# under the License. 17 18import fixtures 19from mox3 import mox 20from mox3 import stubout 21 22 23class MoxStubout(fixtures.Fixture): 24 """Deal with code around mox and stubout as a fixture.""" 25 26 def setUp(self): 27 super(MoxStubout, self).setUp() 28 self.mox = mox.Mox() 29 self.stubs = stubout.StubOutForTesting() 30 self.addCleanup(self.mox.UnsetStubs) 31 self.addCleanup(self.stubs.UnsetAll) 32 self.addCleanup(self.stubs.SmartUnsetAll) 33 self.addCleanup(self.mox.VerifyAll) 34