mirror of git://holbrook.no/eth-faucet
135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
# standard imports
|
||
import os
|
||
import unittest
|
||
import logging
|
||
|
||
# external imports
|
||
from chainlib.connection import RPCConnection
|
||
from chainlib.eth.nonce import RPCNonceOracle
|
||
from chainlib.eth.unittest.ethtester import EthTesterCase
|
||
from chainlib.eth.tx import receipt
|
||
from chainlib.eth.address import to_checksum_address
|
||
from chainlib.eth.gas import balance
|
||
from chainlib.eth.gas import Gas
|
||
from chainlib.eth.gas import OverrideGasOracle
|
||
|
||
# local imports
|
||
from eth_faucet.faucet import EthFaucet
|
||
|
||
logging.basicConfig(level=logging.DEBUG)
|
||
logg = logging.getLogger()
|
||
|
||
|
||
class TestFaucet(EthTesterCase):
|
||
|
||
def setUp(self):
|
||
super(TestFaucet, self).setUp()
|
||
self.conn = RPCConnection.connect(self.chain_spec, 'default')
|
||
nonce_oracle = RPCNonceOracle(self.accounts[0], self.conn)
|
||
c = EthFaucet(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||
(tx_hash_hex, o) = c.constructor(self.accounts[0])
|
||
r = self.conn.do(o)
|
||
|
||
o = receipt(r)
|
||
r = self.conn.do(o)
|
||
self.address = to_checksum_address(r['contract_address'])
|
||
logg.debug('faucet contract {}'.format(self.address))
|
||
|
||
|
||
def test_basic(self):
|
||
nonce_oracle = RPCNonceOracle(self.accounts[0], self.conn)
|
||
c = EthFaucet(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||
(tx_hash_hex, o) = c.give_to(self.address, self.accounts[0], self.accounts[2])
|
||
self.conn.do(o)
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
o = balance(self.accounts[9])
|
||
r = self.conn.do(o)
|
||
prebalance = int(r, 16)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
o = balance(self.accounts[2])
|
||
r = self.conn.do(o)
|
||
self.assertEqual(int(r, 16), prebalance)
|
||
|
||
(tx_hash, o) = c.set_amount(self.address, self.accounts[0], 1000)
|
||
r = self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
(tx_hash_hex, o) = c.give_to(self.address, self.accounts[0], self.accounts[2])
|
||
self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 0)
|
||
|
||
contract_gas_oracle = OverrideGasOracle(limit=21055, conn=self.conn)
|
||
cg = Gas(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle, gas_oracle=contract_gas_oracle)
|
||
(tx_hash_hex, o) = cg.create(self.accounts[0], self.address, 1000)
|
||
self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
(tx_hash_hex, o) = c.give_to(self.address, self.accounts[0], self.accounts[2])
|
||
self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
o = balance(self.accounts[2])
|
||
r = self.conn.do(o)
|
||
self.assertEqual(int(r, 16), prebalance + 1000)
|
||
|
||
|
||
def test_basic(self):
|
||
nonce_oracle = RPCNonceOracle(self.accounts[0], self.conn)
|
||
c = EthFaucet(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||
|
||
contract_gas_oracle = OverrideGasOracle(limit=21055, conn=self.conn)
|
||
cg = Gas(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle, gas_oracle=contract_gas_oracle)
|
||
(tx_hash_hex, o) = cg.create(self.accounts[0], self.address, 1000)
|
||
self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
(tx_hash, o) = c.set_amount(self.address, self.accounts[0], 1000)
|
||
r = self.conn.do(o)
|
||
|
||
o = balance(self.accounts[1])
|
||
r = self.conn.do(o)
|
||
prebalance = int(r, 16)
|
||
|
||
nonce_oracle = RPCNonceOracle(self.accounts[1], self.conn)
|
||
gas_price = 1000000000
|
||
gas_oracle = OverrideGasOracle(limit=100000, price=gas_price, conn=self.conn)
|
||
c = EthFaucet(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
||
(tx_hash_hex, o) = c.gimme(self.address, self.accounts[1])
|
||
self.conn.do(o)
|
||
|
||
o = receipt(tx_hash_hex)
|
||
r = self.conn.do(o)
|
||
logg.debug('rrrr {}'.format(r))
|
||
cost = r['gas_used'] * gas_price
|
||
self.assertEqual(r['status'], 1)
|
||
|
||
o = balance(self.accounts[1])
|
||
r = self.conn.do(o)
|
||
self.assertEqual(int(r, 16), prebalance - cost + 1000)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|