~justin-fathomdb/nova/justinsb-openstack-api-volumes

« back to all changes in this revision

Viewing changes to vendor/pymox/stubout_test.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/usr/bin/python2.4
 
2
#
 
3
# Unit tests for stubout.
 
4
#
 
5
# Licensed under the Apache License, Version 2.0 (the "License");
 
6
# you may not use this file except in compliance with the License.
 
7
# You may obtain a copy of the License at
 
8
#
 
9
#      http://www.apache.org/licenses/LICENSE-2.0
 
10
#
 
11
# Unless required by applicable law or agreed to in writing, software
 
12
# distributed under the License is distributed on an "AS IS" BASIS,
 
13
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
14
# See the License for the specific language governing permissions and
 
15
# limitations under the License.
 
16
 
 
17
import unittest
 
18
 
 
19
import mox
 
20
import stubout
 
21
import stubout_testee
 
22
 
 
23
 
 
24
class StubOutForTestingTest(unittest.TestCase):
 
25
  def setUp(self):
 
26
    self.mox = mox.Mox()
 
27
    self.sample_function_backup = stubout_testee.SampleFunction
 
28
 
 
29
  def tearDown(self):
 
30
    stubout_testee.SampleFunction = self.sample_function_backup
 
31
 
 
32
  def testSmartSetOnModule(self):
 
33
    mock_function = self.mox.CreateMockAnything()
 
34
    mock_function()
 
35
 
 
36
    stubber = stubout.StubOutForTesting()
 
37
    stubber.SmartSet(stubout_testee, 'SampleFunction', mock_function)
 
38
 
 
39
    self.mox.ReplayAll()
 
40
 
 
41
    stubout_testee.SampleFunction()
 
42
 
 
43
    self.mox.VerifyAll()
 
44
 
 
45
 
 
46
if __name__ == '__main__':
 
47
  unittest.main()