diff --git a/tests/test_resolve_command.py b/tests/test_resolve_command.py index 8d99fb9..bb424f7 100644 --- a/tests/test_resolve_command.py +++ b/tests/test_resolve_command.py @@ -1463,27 +1463,27 @@ class ResolveClaimTakeovers(BaseResolveTestCase): COIN = int(1E8) self.assertEqual(self.conductor.spv_node.writer.height, 207) - self.conductor.spv_node.writer.db.prefix_db.trending_notification.stage_put( + self.conductor.spv_node.writer.db.prefix_db.trending_notification.stash_put( (208, bytes.fromhex(claim_id1)), (0, 10 * COIN) ) await self.generate(1) self.assertEqual(self.conductor.spv_node.writer.height, 208) self.assertEqual(1.7090807854206793, await get_trending_score(claim_id1)) - self.conductor.spv_node.writer.db.prefix_db.trending_notification.stage_put( + self.conductor.spv_node.writer.db.prefix_db.trending_notification.stash_put( (209, bytes.fromhex(claim_id1)), (10 * COIN, 100 * COIN) ) await self.generate(1) self.assertEqual(self.conductor.spv_node.writer.height, 209) self.assertEqual(2.2437974397778886, await get_trending_score(claim_id1)) - self.conductor.spv_node.writer.db.prefix_db.trending_notification.stage_put( + self.conductor.spv_node.writer.db.prefix_db.trending_notification.stash_put( (309, bytes.fromhex(claim_id1)), (100 * COIN, 1000000 * COIN) ) await self.generate(100) self.assertEqual(self.conductor.spv_node.writer.height, 309) self.assertEqual(5.157053472135866, await get_trending_score(claim_id1)) - self.conductor.spv_node.writer.db.prefix_db.trending_notification.stage_put( + self.conductor.spv_node.writer.db.prefix_db.trending_notification.stash_put( (409, bytes.fromhex(claim_id1)), (1000000 * COIN, 1 * COIN) ) diff --git a/tests/test_revertable.py b/tests/test_revertable.py index fa62f4d..0f750e9 100644 --- a/tests/test_revertable.py +++ b/tests/test_revertable.py @@ -125,24 +125,24 @@ class TestRevertablePrefixDB(unittest.TestCase): takeover_height = 10000000 self.assertIsNone(self.db.claim_takeover.get(name)) - self.db.claim_takeover.stage_put((name,), (claim_hash1, takeover_height)) + self.db.claim_takeover.stash_put((name,), (claim_hash1, takeover_height)) self.assertIsNone(self.db.claim_takeover.get(name)) self.assertEqual(10000000, self.db.claim_takeover.get_pending(name).height) self.db.commit(10000000, b'\x00' * 32) self.assertEqual(10000000, self.db.claim_takeover.get(name).height) - self.db.claim_takeover.stage_delete((name,), (claim_hash1, takeover_height)) - self.db.claim_takeover.stage_put((name,), (claim_hash2, takeover_height + 1)) - self.db.claim_takeover.stage_delete((name,), (claim_hash2, takeover_height + 1)) + self.db.claim_takeover.stash_delete((name,), (claim_hash1, takeover_height)) + self.db.claim_takeover.stash_put((name,), (claim_hash2, takeover_height + 1)) + self.db.claim_takeover.stash_delete((name,), (claim_hash2, takeover_height + 1)) self.db.commit(10000001, b'\x01' * 32) self.assertIsNone(self.db.claim_takeover.get(name)) - self.db.claim_takeover.stage_put((name,), (claim_hash3, takeover_height + 2)) + self.db.claim_takeover.stash_put((name,), (claim_hash3, takeover_height + 2)) self.db.commit(10000002, b'\x02' * 32) self.assertEqual(10000002, self.db.claim_takeover.get(name).height) - self.db.claim_takeover.stage_delete((name,), (claim_hash3, takeover_height + 2)) - self.db.claim_takeover.stage_put((name,), (claim_hash2, takeover_height + 3)) + self.db.claim_takeover.stash_delete((name,), (claim_hash3, takeover_height + 2)) + self.db.claim_takeover.stash_put((name,), (claim_hash2, takeover_height + 3)) self.db.commit(10000003, b'\x03' * 32) self.assertEqual(10000003, self.db.claim_takeover.get(name).height) @@ -162,15 +162,15 @@ class TestRevertablePrefixDB(unittest.TestCase): claim_hash2 = 20 * b'\x02' claim_hash3 = 20 * b'\x03' overflow_value = 0xffffffff - self.db.claim_expiration.stage_put((99, 999, 0), (claim_hash0, name)) - self.db.claim_expiration.stage_put((100, 1000, 0), (claim_hash1, name)) - self.db.claim_expiration.stage_put((100, 1001, 0), (claim_hash2, name)) - self.db.claim_expiration.stage_put((101, 1002, 0), (claim_hash3, name)) - self.db.claim_expiration.stage_put((overflow_value - 1, 1003, 0), (claim_hash3, name)) - self.db.claim_expiration.stage_put((overflow_value, 1004, 0), (claim_hash3, name)) - self.db.tx_num.stage_put((b'\x00' * 32,), (101,)) - self.db.claim_takeover.stage_put((name,), (claim_hash3, 101)) - self.db.db_state.stage_put((), (b'n?\xcf\x12\x99\xd4\xec]y\xc3\xa4\xc9\x1dbJJ\xcf\x9e.\x17=\x95\xa1\xa0POgvihuV', 0, 1, b'VuhivgOP\xa0\xa1\x95=\x17.\x9e\xcfJJb\x1d\xc9\xa4\xc3y]\xec\xd4\x99\x12\xcf?n', 1, 0, 1, 7, 1, -1, -1, 0, 0, 0)) + self.db.claim_expiration.stash_put((99, 999, 0), (claim_hash0, name)) + self.db.claim_expiration.stash_put((100, 1000, 0), (claim_hash1, name)) + self.db.claim_expiration.stash_put((100, 1001, 0), (claim_hash2, name)) + self.db.claim_expiration.stash_put((101, 1002, 0), (claim_hash3, name)) + self.db.claim_expiration.stash_put((overflow_value - 1, 1003, 0), (claim_hash3, name)) + self.db.claim_expiration.stash_put((overflow_value, 1004, 0), (claim_hash3, name)) + self.db.tx_num.stash_put((b'\x00' * 32,), (101,)) + self.db.claim_takeover.stash_put((name,), (claim_hash3, 101)) + self.db.db_state.stash_put((), (b'n?\xcf\x12\x99\xd4\xec]y\xc3\xa4\xc9\x1dbJJ\xcf\x9e.\x17=\x95\xa1\xa0POgvihuV', 0, 1, b'VuhivgOP\xa0\xa1\x95=\x17.\x9e\xcfJJb\x1d\xc9\xa4\xc3y]\xec\xd4\x99\x12\xcf?n', 1, 0, 1, 7, 1, -1, -1, 0, 0, 0)) self.db.unsafe_commit() state = self.db.db_state.get() @@ -217,9 +217,9 @@ class TestRevertablePrefixDB(unittest.TestCase): tx_num = 101 for x in range(255): claim_hash = 20 * chr(x).encode() - self.db.active_amount.stage_put((claim_hash, 1, 200, tx_num, 1), (100000,)) - self.db.active_amount.stage_put((claim_hash, 1, 201, tx_num + 1, 1), (200000,)) - self.db.active_amount.stage_put((claim_hash, 1, 202, tx_num + 2, 1), (300000,)) + self.db.active_amount.stash_put((claim_hash, 1, 200, tx_num, 1), (100000,)) + self.db.active_amount.stash_put((claim_hash, 1, 201, tx_num + 1, 1), (200000,)) + self.db.active_amount.stash_put((claim_hash, 1, 202, tx_num + 2, 1), (300000,)) tx_num += 3 self.db.unsafe_commit()