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()