def test_reuse_after_removal(self):
pin_string = PinNumber.generate(TEST_KEY)
pin = PinNumber.retrive(pin_string)
self.assertTrue(pin.used)
pin.free()
self.assertFalse(pin.used)
PinNumber.generate(TEST_KEY)
self.assertEqual(len(PinNumber.query().fetch()), 1)
PinNumber.generate(TEST_KEY)
self.assertEqual(len(PinNumber.query().fetch()), 2)
def test_reuse_expired(self):
m = mox.Mox()
m.StubOutWithMock(time, 'time')
time.time().MultipleTimes().AndReturn(0)
m.ReplayAll()
PinNumber.generate(TEST_KEY, lifetime=10)
PinNumber.generate(TEST_KEY)
m.VerifyAll()
self.assertEqual(len(PinNumber.query().fetch()), 2)
m.StubOutWithMock(PinNumber, 'time')
PinNumber.time().MultipleTimes().AndReturn(11)
m.ReplayAll()
PinNumber.generate(TEST_KEY)
m.VerifyAll()
m.UnsetStubs()
self.assertEqual(len(PinNumber.query().fetch()), 2)
