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