def testCheckpointWrite(self): x = Checkpoint(TestCheckpoint.TEST_DIR) x.createCheckpointLog(TestCheckpoint.TEST_KEY) x.writeCheckpoint(TestCheckpoint.TEST_KEY, "a", 1) x.writeCheckpoint(TestCheckpoint.TEST_KEY, "a", 2) x.writeCheckpoint(TestCheckpoint.TEST_KEY, "b", 3) x.releaseCheckpointLog(TestCheckpoint.TEST_KEY) self.assertEqual(sorted(x.getCheckpointKeys()), ["a", "b"]) self.assertEqual(x.getCheckpoints("b"), [3]) self.assertEqual(x.getCheckpoints("a"), [1, 2])
def testCheckpointRestart(self): x = Checkpoint(TestCheckpoint.TEST_DIR) x.createCheckpointLog(TestCheckpoint.TEST_KEY) x.createCheckpointLog(TestCheckpoint.TEST_KEY2) x.writeCheckpoint(TestCheckpoint.TEST_KEY, "a", 1) x.writeCheckpoint(TestCheckpoint.TEST_KEY, "a", 2) x.writeCheckpoint(TestCheckpoint.TEST_KEY2, "b", 3) x.releaseCheckpointLog(TestCheckpoint.TEST_KEY) x.releaseCheckpointLog(TestCheckpoint.TEST_KEY2) y = Checkpoint(TestCheckpoint.TEST_DIR) self.assertEqual(y.getCheckpoints("b"), [3]) self.assertEqual(y.getCheckpoints("a"), [1, 2])