def testSetCount(self): """Verify that settting the count works""" c = Controller() a = BaseAction('x') c.record(a) self.failUnless(a.playbackPolicy.remaining == 1) c.setCount(2) self.failUnless(a.playbackPolicy.remaining == 2)
def testPlaybackWithMultipleCounts(self): """Perform playback with multiple values""" c = Controller() action = BaseAction('x') c.record(action) c.setCount(2) self.failIf(action.playbackPolicy.hasBeenPlayedBack) self.failIf(action.playbackPolicy.isReadyForRemoval) c.playback(action) self.failIf(action.playbackPolicy.hasBeenPlayedBack) self.failIf(action.playbackPolicy.isReadyForRemoval) c.playback(action) self.failUnless(action.playbackPolicy.hasBeenPlayedBack) self.failUnless(action.playbackPolicy.isReadyForRemoval)
def testSpecifyingMultiCounts(self): """Specify multi counts for a single argument""" c = Controller() x = c.mock(KlassBeingMocked) x.w = 5 c.setCount(2) c.replay() x.w = 5 x.w = 5 try: x.w = 5 self.fail() except PlaybackFailure, e: pass