def test_token_registry(self):
# given
ERC20Token.register_token(
Address('0x0100000000000000000000000000000000000000'), 'ABC')
ERC20Token.register_token(
Address('0x0200000000000000000000000000000000000000'), 'DEF')
ERC20Token.register_token(
Address('0x0300000000000000000000000000000000000000'), 'GHI')
# expect
assert ERC20Token.token_name_by_address(
Address('0x0100000000000000000000000000000000000000')) == 'ABC'
assert ERC20Token.token_name_by_address(
Address('0x0200000000000000000000000000000000000000')) == 'DEF'
assert ERC20Token.token_name_by_address(
Address('0x0300000000000000000000000000000000000000')) == 'GHI'
with pytest.raises(Exception):
assert ERC20Token.token_name_by_address(
Address('0x0400000000000000000000000000000000000000'))
# and
assert ERC20Token.token_address_by_name('ABC') == Address(
'0x0100000000000000000000000000000000000000')
assert ERC20Token.token_address_by_name('DEF') == Address(
'0x0200000000000000000000000000000000000000')
assert ERC20Token.token_address_by_name('GHI') == Address(
'0x0300000000000000000000000000000000000000')
with pytest.raises(Exception):
ERC20Token.token_address_by_name('XXX')
def __init__(self):
super().__init__()
self.tub_address = Address(self.config.get_contract_address("saiTub"))
self.tub = Tub(web3=self.web3, address=self.tub_address)
self.tap_address = Address(self.config.get_contract_address("saiTap"))
self.tap = Tap(web3=self.web3, address=self.tap_address)
self.top_address = Address(self.config.get_contract_address("saiTop"))
self.top = Top(web3=self.web3, address=self.top_address)
self.otc_address = Address(self.config.get_contract_address("otc"))
self.otc = SimpleMarket(web3=self.web3, address=self.otc_address)
self.skr = ERC20Token(web3=self.web3, address=self.tub.skr())
self.sai = ERC20Token(web3=self.web3, address=self.tub.sai())
self.gem = DSEthToken(web3=self.web3, address=self.tub.gem())
ERC20Token.register_token(self.tub.skr(), 'SKR')
ERC20Token.register_token(self.tub.sai(), 'SAI')
ERC20Token.register_token(self.tub.gem(), 'WETH')
def register_tokens(token1, token2):
ERC20Token.register_token(token1, 'TK1')
ERC20Token.register_token(token2, 'TK2')
def __init__(self, chain: str):
with open('keeper/config.json') as data_file:
self.chain = chain
self.config = json.load(data_file)
for key, value in self.config[self.chain]["tokens"].items():
ERC20Token.register_token(Address(value), key)
