class TestIcxEngine(unittest.TestCase, ContextContainer):
def setUp(self):
self.db_name = 'engine.db'
db = ContextDatabase.from_path(self.db_name)
self.engine = IcxEngine()
self.from_ = Address.from_string('hx' + 'a' * 40)
self.to = Address.from_string('hx' + 'b' * 40)
self.genesis_address = Address.from_string('hx' + '0' * 40)
self.fee_treasury_address = Address.from_string('hx' + '1' * 40)
self.total_supply = 10**20 # 100 icx
self.context = IconScoreContext(IconScoreContextType.DIRECT)
icx_storage = IcxStorage(db)
self.engine.open(icx_storage)
self.engine.init_account(self.context, AccountType.GENESIS, 'genesis',
self.genesis_address, self.total_supply)
self.engine.init_account(self.context, AccountType.TREASURY,
'treasury', self.fee_treasury_address, 0)
def tearDown(self):
self._clear_context()
self.engine.close()
# Remove a state db for test
shutil.rmtree(self.db_name)
def test_init_account(self):
def intercept_storage_put_account(context: 'IconScoreContext',
address: 'Address',
account: 'Account'):
if not (isinstance(address, Address)
and isinstance(account, Account)):
raise Exception(
"invalid type of params was set when calling put_account")
self.engine._init_special_account = Mock()
self.engine._storage.put_account = Mock(
side_effect=intercept_storage_put_account)
genesis_address = Address.from_string('hx' + 'f' * 40)
self.engine.init_account(self.context, AccountType.GENESIS, 'genesis',
genesis_address, 1000)
self.total_supply += 1000
self.engine._storage.put_account.assert_called()
self.engine._init_special_account.assert_called()
self.assertEqual(self.total_supply, self.engine._total_supply_amount)
self.engine._init_special_account = Mock()
general_addr = Address.from_string('hx' + 'f' * 40)
self.engine.init_account(self.context, AccountType.GENERAL, 'general',
general_addr, 1000)
self.total_supply += 1000
self.engine._storage.put_account.assert_called()
self.engine._init_special_account.assert_not_called()
self.assertEqual(self.total_supply, self.engine._total_supply_amount)
def test_init_special_account(self):
# failure case: input general account
general_account = Mock(spec=Account)
general_account.attach_mock(AccountType.GENERAL, 'type')
self.assertRaises(AssertionError, self.engine._init_special_account,
self.context, general_account)
# success case: input genesis and treasury account
genesis_account = Mock(spec=Account)
genesis_account.attach_mock(AccountType.GENESIS, 'type')
self.engine._init_special_account(self.context, genesis_account)
self.assertEqual(genesis_account.address, self.engine._genesis_address)
treasury_account = Mock(spec=Account)
treasury_account.attach_mock(AccountType.TREASURY, 'type')
self.engine._init_special_account(self.context, treasury_account)
self.assertEqual(treasury_account.address,
self.engine._fee_treasury_address)
def intercept_storage_put_text(*args, **kwargs):
text = args[-1]
text_dict = json.loads(text)
expected_version = 0
expected_address = str(treasury_account.address)
self.assertEqual(expected_version, text_dict['version'])
self.assertEqual(expected_address, text_dict['address'])
self.engine.storage.put_text = Mock(
side_effect=intercept_storage_put_text)
self.engine._init_special_account(self.context, treasury_account)
self.engine.storage.put_text.assert_called()
def test_load_address_from_storage(self):
# success case: when stored genesis address is exist
genesis_address = Address.from_string('hx' + 'f' * 40)
expected_text = {'version': 0, 'address': str(genesis_address)}
# make mock which returns stringified json data
icx_storage_mock = Mock(spec=IcxStorage)
icx_storage_mock.get_text = Mock(
return_value=json.dumps(expected_text))
self.engine._load_address_from_storage(self.context, icx_storage_mock,
self.engine._GENESIS_DB_KEY)
# check genesis_address
self.assertEqual(genesis_address, self.engine._genesis_address)
# success case: when stored genesis address is not exist
# reset icx engine's genesis address
self.engine._genesis_address = None
icx_storage_mock.get_text = Mock(return_value=None)
self.engine._load_address_from_storage(self.context, icx_storage_mock,
self.engine._GENESIS_DB_KEY)
self.assertEqual(None, self.engine._genesis_address)
def test_load_total_supply_amount_from_storage(self):
pass
def test_get_balance(self):
address = Address.from_string(
'hx0123456789012345678901234567890123456789')
balance = self.engine.get_balance(self.context, address)
self.assertEqual(0, balance)
def test_get_total_supply(self):
total_supply = self.engine.get_total_supply(self.context)
self.assertEqual(self.total_supply, total_supply)
def test_get_charge_fee(self):
pass
def test_get_account(self):
pass
def test_transfer(self):
context = self.context
amount = 10**18 # 1 icx
_from = self.genesis_address
self.engine.transfer(context=context,
from_=_from,
to=self.to,
amount=amount)
from_balance = self.engine.get_balance(context, self.genesis_address)
fee_treasury_balance = self.engine.get_balance(
context, self.fee_treasury_address)
to_balance = self.engine.get_balance(context, self.to)
self.assertEqual(amount, to_balance)
self.assertEqual(0, fee_treasury_balance)
self.assertEqual(self.total_supply,
from_balance + to_balance + fee_treasury_balance)
class TestIcxEngine(unittest.TestCase, ContextContainer):
def setUp(self):
self.db_name = 'engine.db'
db = ContextDatabase.from_path(self.db_name)
self.engine = IcxEngine()
self.from_ = Address.from_string('hx' + 'a' * 40)
self.to = Address.from_string('hx' + 'b' * 40)
self.genesis_address = Address.from_string('hx' + '0' * 40)
self.fee_treasury_address = Address.from_string('hx' + '1' * 40)
self.total_supply = 10**20 # 100 icx
self.factory = IconScoreContextFactory(max_size=1)
self.context = self.factory.create(IconScoreContextType.DIRECT)
icx_storage = IcxStorage(db)
self.engine.open(icx_storage)
self.engine.init_account(self.context, AccountType.GENESIS, 'genesis',
self.genesis_address, self.total_supply)
self.engine.init_account(self.context, AccountType.TREASURY,
'treasury', self.fee_treasury_address, 0)
def tearDown(self):
self._clear_context()
self.engine.close()
self.factory.destroy(self.context)
# Remove a state db for test
shutil.rmtree(self.db_name)
def test_get_balance(self):
address = Address.from_string(
'hx0123456789012345678901234567890123456789')
balance = self.engine.get_balance(self.context, address)
self.assertEqual(0, balance)
def test_get_total_supply(self):
total_supply = self.engine.get_total_supply(self.context)
self.assertEqual(self.total_supply, total_supply)
def test_transfer(self):
context = self.context
amount = 10**18 # 1 icx
_from = self.genesis_address
self.engine.transfer(context=context,
from_=_from,
to=self.to,
amount=amount)
from_balance = self.engine.get_balance(context, self.genesis_address)
fee_treasury_balance = self.engine.get_balance(
context, self.fee_treasury_address)
to_balance = self.engine.get_balance(context, self.to)
self.assertEqual(amount, to_balance)
self.assertEqual(0, fee_treasury_balance)
self.assertEqual(self.total_supply,
from_balance + to_balance + fee_treasury_balance)