def test_shachain_produce_consume(self): seed = bitcoin.sha256(b"shachaintest") consumer = RevocationStore(StoredDict({}, None, [])) for i in range(10000): secret = get_per_commitment_secret_from_seed(seed, RevocationStore.START_INDEX - i) try: consumer.add_next_entry(secret) except Exception as e: raise Exception("iteration " + str(i) + ": " + str(e)) if i % 1000 == 0: c1 = consumer s1 = json.dumps(c1.storage, cls=MyEncoder) c2 = RevocationStore(StoredDict(json.loads(s1), None, [])) s2 = json.dumps(c2.storage, cls=MyEncoder) self.assertEqual(s1, s2)
def test_shachain_produce_consume(self): seed = bitcoin.sha256(b"shachaintest") consumer = RevocationStore() for i in range(10000): secret = get_per_commitment_secret_from_seed(seed, RevocationStore.START_INDEX - i) try: consumer.add_next_entry(secret) except Exception as e: raise Exception("iteration " + str(i) + ": " + str(e)) if i % 1000 == 0: self.assertEqual(consumer.serialize(), RevocationStore.from_json_obj(json.loads(json.dumps(consumer.serialize()))).serialize())
def test_shachain_store(self): tests = [ { "name": "insert_secret correct sequence", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1" +\ "a8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab2" +\ "1e9b506fd4998a51d54502e99116", "successful": True }, { "index": 281474976710651, "secret": "c65716add7aa98ba7acb236352d665cab173" +\ "45fe45b55fb879ff80e6bd0c41dd", "successful": True }, { "index": 281474976710650, "secret": "969660042a28f32d9be17344e09374b37996" +\ "2d03db1574df5a8a5a47e19ce3f2", "successful": True }, { "index": 281474976710649, "secret": "a5a64476122ca0925fb344bdc1854c1c0a59" +\ "fc614298e50a33e331980a220f32", "successful": True }, { "index": 281474976710648, "secret": "05cde6323d949933f7f7b78776bcc1ea6d9b" +\ "31447732e3802e1f7ac44b650e17", "successful": True } ] }, { "name": "insert_secret #1 incorrect", "inserts": [ { "index": 281474976710655, "secret": "02a40c85b6f28da08dfdbe0926c53fab2d" +\ "e6d28c10301f8f7c4073d5e42e3148", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659" +\ "c1a8b4b5bec0c4b872abeba4cb8964", "successful": False } ] }, { "name": "insert_secret #2 incorrect (#1 derived from incorrect)", "inserts": [ { "index": 281474976710655, "secret": "02a40c85b6f28da08dfdbe0926c53fab2de6" +\ "d28c10301f8f7c4073d5e42e3148", "successful": True }, { "index": 281474976710654, "secret": "dddc3a8d14fddf2b68fa8c7fbad274827493" +\ "7479dd0f8930d5ebb4ab6bd866a3", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22a" +\ "b21e9b506fd4998a51d54502e99116", "successful": False } ] }, { "name": "insert_secret #3 incorrect", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1" +\ "a8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "c51a18b13e8527e579ec56365482c62f180b" +\ "7d5760b46e9477dae59e87ed423a", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab2" +\ "1e9b506fd4998a51d54502e99116", "successful": False } ] }, { "name": "insert_secret #4 incorrect (1,2,3 derived from incorrect)", "inserts": [ { "index": 281474976710655, "secret": "02a40c85b6f28da08dfdbe0926c53fab2de6" +\ "d28c10301f8f7c4073d5e42e3148", "successful": True }, { "index": 281474976710654, "secret": "dddc3a8d14fddf2b68fa8c7fbad274827493" +\ "7479dd0f8930d5ebb4ab6bd866a3", "successful": True }, { "index": 281474976710653, "secret": "c51a18b13e8527e579ec56365482c62f18" +\ "0b7d5760b46e9477dae59e87ed423a", "successful": True }, { "index": 281474976710652, "secret": "ba65d7b0ef55a3ba300d4e87af29868f39" +\ "4f8f138d78a7011669c79b37b936f4", "successful": True }, { "index": 281474976710651, "secret": "c65716add7aa98ba7acb236352d665cab1" +\ "7345fe45b55fb879ff80e6bd0c41dd", "successful": True }, { "index": 281474976710650, "secret": "969660042a28f32d9be17344e09374b379" +\ "962d03db1574df5a8a5a47e19ce3f2", "successful": True }, { "index": 281474976710649, "secret": "a5a64476122ca0925fb344bdc1854c1c0a" +\ "59fc614298e50a33e331980a220f32", "successful": True }, { "index": 281474976710649, "secret": "05cde6323d949933f7f7b78776bcc1ea6d9b" +\ "31447732e3802e1f7ac44b650e17", "successful": False } ] }, { "name": "insert_secret #5 incorrect", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1a" +\ "8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab21" +\ "e9b506fd4998a51d54502e99116", "successful": True }, { "index": 281474976710651, "secret": "631373ad5f9ef654bb3dade742d09504c567" +\ "edd24320d2fcd68e3cc47e2ff6a6", "successful": True }, { "index": 281474976710650, "secret": "969660042a28f32d9be17344e09374b37996" +\ "2d03db1574df5a8a5a47e19ce3f2", "successful": False } ] }, { "name": "insert_secret #6 incorrect (5 derived from incorrect)", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1a" +\ "8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab21" +\ "e9b506fd4998a51d54502e99116", "successful": True }, { "index": 281474976710651, "secret": "631373ad5f9ef654bb3dade742d09504c567" +\ "edd24320d2fcd68e3cc47e2ff6a6", "successful": True }, { "index": 281474976710650, "secret": "b7e76a83668bde38b373970155c868a65330" +\ "4308f9896692f904a23731224bb1", "successful": True }, { "index": 281474976710649, "secret": "a5a64476122ca0925fb344bdc1854c1c0a59f" +\ "c614298e50a33e331980a220f32", "successful": True }, { "index": 281474976710648, "secret": "05cde6323d949933f7f7b78776bcc1ea6d9b" +\ "31447732e3802e1f7ac44b650e17", "successful": False } ] }, { "name": "insert_secret #7 incorrect", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1a" +\ "8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab21" +\ "e9b506fd4998a51d54502e99116", "successful": True }, { "index": 281474976710651, "secret": "c65716add7aa98ba7acb236352d665cab173" +\ "45fe45b55fb879ff80e6bd0c41dd", "successful": True }, { "index": 281474976710650, "secret": "969660042a28f32d9be17344e09374b37996" +\ "2d03db1574df5a8a5a47e19ce3f2", "successful": True }, { "index": 281474976710649, "secret": "e7971de736e01da8ed58b94c2fc216cb1d" +\ "ca9e326f3a96e7194fe8ea8af6c0a3", "successful": True }, { "index": 281474976710648, "secret": "05cde6323d949933f7f7b78776bcc1ea6d" +\ "9b31447732e3802e1f7ac44b650e17", "successful": False } ] }, { "name": "insert_secret #8 incorrect", "inserts": [ { "index": 281474976710655, "secret": "7cc854b54e3e0dcdb010d7a3fee464a9687b" +\ "e6e8db3be6854c475621e007a5dc", "successful": True }, { "index": 281474976710654, "secret": "c7518c8ae4660ed02894df8976fa1a3659c1a" +\ "8b4b5bec0c4b872abeba4cb8964", "successful": True }, { "index": 281474976710653, "secret": "2273e227a5b7449b6e70f1fb4652864038b1" +\ "cbf9cd7c043a7d6456b7fc275ad8", "successful": True }, { "index": 281474976710652, "secret": "27cddaa5624534cb6cb9d7da077cf2b22ab21" +\ "e9b506fd4998a51d54502e99116", "successful": True }, { "index": 281474976710651, "secret": "c65716add7aa98ba7acb236352d665cab173" +\ "45fe45b55fb879ff80e6bd0c41dd", "successful": True }, { "index": 281474976710650, "secret": "969660042a28f32d9be17344e09374b37996" +\ "2d03db1574df5a8a5a47e19ce3f2", "successful": True }, { "index": 281474976710649, "secret": "a5a64476122ca0925fb344bdc1854c1c0a" +\ "59fc614298e50a33e331980a220f32", "successful": True }, { "index": 281474976710648, "secret": "a7efbc61aac46d34f77778bac22c8a20c6" +\ "a46ca460addc49009bda875ec88fa4", "successful": False } ] } ] for test in tests: receiver = RevocationStore() for insert in test["inserts"]: secret = bytes.fromhex(insert["secret"]) try: receiver.add_next_entry(secret) except Exception as e: if insert["successful"]: raise Exception("Failed ({}): error was received but it shouldn't: {}".format(test["name"], e)) else: if not insert["successful"]: raise Exception("Failed ({}): error wasn't received".format(test["name"])) for insert in test["inserts"]: secret = bytes.fromhex(insert["secret"]) index = insert["index"] if insert["successful"]: self.assertEqual(secret, receiver.retrieve_secret(index)) print("Passed ({})".format(test["name"]))