Add transaction executor helper

This commit is contained in:
nolash
2021-01-09 22:05:24 +01:00
parent 6ba0ac6820
commit f7c1f05a1f
8 changed files with 185 additions and 20 deletions

73
test/test_helper.py Normal file
View File

@@ -0,0 +1,73 @@
# standard imports
import unittest
import logging
import os
# local imports
from crypto_dev_signer.keystore import DictKeystore
from crypto_dev_signer.eth.signer import ReferenceSigner
from crypto_dev_signer.helper import TxExecutor
logging.basicConfig(level=logging.DEBUG)
logg = logging.getLogger()
script_dir = os.path.realpath(os.path.dirname(__file__))
class MockEthTxBackend:
def dispatcher(self, tx):
logg.debug('sender {}'.format(tx))
return os.urandom(32)
def reporter(self, tx):
logg.debug('reporter {}'.format(tx))
def fee_price_helper(self):
return 21
def fee_helper(self, sender, code, inputs):
logg.debug('fee helper code {} inputs {}'.format(code, inputs))
return 2
def builder(self, a, b):
b = {
'from': a['from'],
'to': '0x' + os.urandom(20).hex(),
'data': '',
'gasPrice': a['feePrice'],
'gas': a['fee'],
}
return b
def builder_two(self, a, b):
b['value'] = 1024
return b
class TestHelper(unittest.TestCase):
def setUp(self):
logg.debug('setup')
self.db = DictKeystore()
keystore_filepath = os.path.join(script_dir, 'testdata', 'UTC--2021-01-08T18-37-01.187235289Z--00a329c0648769a73afac7f9381e08fb43dbea72')
self.address_hex = self.db.import_keystore_file(keystore_filepath, '')
self.signer = ReferenceSigner(self.db)
def tearDown(self):
pass
def test_helper(self):
backend = MockEthTxBackend()
executor = TxExecutor(self.address_hex, self.signer, backend.dispatcher, backend.reporter, 666, 13, backend.fee_helper, backend.fee_price_helper)
tx_ish = {'from': self.address_hex}
executor.sign_and_send([backend.builder, backend.builder_two])
if __name__ == '__main__':
unittest.main()

View File

@@ -6,11 +6,6 @@ import logging
import base64
import os
# third-party imports
import psycopg2
from psycopg2 import sql
from cryptography.fernet import Fernet, InvalidToken
# local imports
from crypto_dev_signer.keystore import DictKeystore
from crypto_dev_signer.error import UnknownAccountError
@@ -22,29 +17,16 @@ logg = logging.getLogger()
script_dir = os.path.realpath(os.path.dirname(__file__))
class TestDatabase(unittest.TestCase):
class TestDict(unittest.TestCase):
conn = None
cur = None
symkey = None
address_hex = None
db = None
def setUp(self):
logg.debug('setup')
# arbitrary value
#symkey_hex = 'E92431CAEE69313A7BE9E443C4ABEED9BF8157E9A13553B4D5D6E7D51B5021D9'
#self.symkey = bytes.fromhex(symkey_hex)
#kw = {
# 'symmetric_key': self.symkey,
# }
self.db = DictKeystore()
keystore_filepath = os.path.join(script_dir, 'testdata', 'UTC--2021-01-08T18-37-01.187235289Z--00a329c0648769a73afac7f9381e08fb43dbea72')
#f = open(
#s = f.read()
#f.close()
self.address_hex = self.db.import_keystore_file(keystore_filepath, '')