sdk/platform-tools/systrace/catapult/telemetry/third_party/mox3/mox3/fixture.py

   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
  18 import fixtures
  19 from mox3 import mox
  20 from mox3 import stubout
  21
  22
  23 class 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)