Incorporate kaste's suggested tests
Niels van Eck authored 4 years ago
herr kaste committed 4 years ago
| 188 | 188 | self.assertEqual("woof!", Dog.bark()) |
| 189 | 189 | self.assertEqual("stick", Retriever.retrieve("stick")) |
| 190 | 190 | |
| 191 | def testDoubleStubStubWorksAfterUnstub(self): | |
| 192 | when(TrickDog).retrieve("stick").thenReturn("ball") | |
| 193 | when(TrickDog).retrieve("stick").thenReturn("cat") | |
| 194 | unstub() | |
| 195 | self.assertEqual("stick", TrickDog.retrieve("stick")) | |
| 196 | ||
| 191 | 197 | def testUnStubWorksOnClassAndSuperClass(self): |
| 198 | self.assertEqual("stick", Retriever.retrieve("stick")) | |
| 199 | self.assertEqual("stick", TrickDog.retrieve("stick")) | |
| 200 | ||
| 201 | when(Retriever).retrieve("stick").thenReturn("ball") | |
| 202 | self.assertEqual("ball", Retriever.retrieve("stick")) | |
| 203 | self.assertEqual("ball", TrickDog.retrieve("stick")) | |
| 204 | ||
| 205 | when(TrickDog).retrieve("stick").thenReturn("cat") | |
| 206 | self.assertEqual("ball", Retriever.retrieve("stick")) | |
| 207 | self.assertEqual("cat", TrickDog.retrieve("stick")) | |
| 208 | ||
| 209 | unstub(TrickDog) | |
| 210 | self.assertEqual("ball", Retriever.retrieve("stick")) | |
| 211 | self.assertEqual("ball", TrickDog.retrieve("stick")) | |
| 212 | ||
| 213 | unstub(Retriever) | |
| 214 | self.assertEqual("stick", Retriever.retrieve("stick")) | |
| 215 | self.assertEqual("stick", TrickDog.retrieve("stick")) | |
| 216 | ||
| 217 | def testReverseOrderWhenUnstubbing(self): | |
| 192 | 218 | when(Retriever).retrieve("stick").thenReturn("ball") |
| 193 | 219 | when(TrickDog).retrieve("stick").thenReturn("cat") |
| 194 | unstub() | |
| 195 | self.assertEqual("stick", TrickDog.retrieve("stick")) | |
| 196 | ||
| 197 | def testDoubleStubStubWorksAfterUnstub(self): | |
| 198 | when(TrickDog).retrieve("stick").thenReturn("ball") | |
| 199 | when(TrickDog).retrieve("stick").thenReturn("cat") | |
| 200 | unstub() | |
| 201 | self.assertEqual("stick", TrickDog.retrieve("stick")) | |
| 220 | ||
| 221 | unstub(Retriever) | |
| 222 | self.assertEqual("stick", Retriever.retrieve("stick")) | |
| 223 | self.assertEqual("cat", TrickDog.retrieve("stick")) | |
| 224 | ||
| 225 | unstub(TrickDog) | |
| 226 | self.assertEqual("stick", Retriever.retrieve("stick")) | |
| 227 | self.assertEqual("stick", TrickDog.retrieve("stick")) |