Moved and renamed hash256 from util.py to zmq_interface.py

This commit is contained in:
Elichai Turkel 2019-08-13 14:55:24 -04:00
parent b799ebcc17
commit afc0966d72
No known key found for this signature in database
GPG key ID: 9383CDE9E8E66A7F
2 changed files with 7 additions and 14 deletions

View file

@ -7,15 +7,15 @@ import struct
from test_framework.address import ADDRESS_BCRT1_UNSPENDABLE
from test_framework.test_framework import BitcoinTestFramework
from test_framework.messages import CTransaction
from test_framework.util import (
assert_equal,
hash256,
)
from test_framework.messages import CTransaction, hash256
from test_framework.util import assert_equal
from io import BytesIO
ADDRESS = "tcp://127.0.0.1:28332"
def hash256_reversed(byte_str):
return hash256(byte_str)[::-1]
class ZMQSubscriber:
def __init__(self, socket, topic):
self.sequence = 0
@ -103,7 +103,7 @@ class ZMQTest (BitcoinTestFramework):
# Should receive the generated raw block.
block = self.rawblock.receive()
assert_equal(genhashes[x], hash256(block[:80]).hex())
assert_equal(genhashes[x], hash256_reversed(block[:80]).hex())
if self.is_wallet_compiled():
self.log.info("Wait for tx from second node")
@ -116,7 +116,7 @@ class ZMQTest (BitcoinTestFramework):
# Should receive the broadcasted raw transaction.
hex = self.rawtx.receive()
assert_equal(payment_txid, hash256(hex).hex())
assert_equal(payment_txid, hash256_reversed(hex).hex())
self.log.info("Test the getzmqnotifications RPC")

View file

@ -7,7 +7,6 @@
from base64 import b64encode
from binascii import unhexlify
from decimal import Decimal, ROUND_DOWN
import hashlib
import inspect
import json
import logging
@ -183,12 +182,6 @@ def check_json_precision():
def count_bytes(hex_string):
return len(bytearray.fromhex(hex_string))
def hash256(byte_str):
sha256 = hashlib.sha256()
sha256.update(byte_str)
sha256d = hashlib.sha256()
sha256d.update(sha256.digest())
return sha256d.digest()[::-1]
def hex_str_to_bytes(hex_str):
return unhexlify(hex_str.encode('ascii'))