class TestIcxStorageForMalformedAddress(unittest.TestCase):
def setUp(self):
empty_address = MalformedAddress.from_string('')
short_address_without_hx = MalformedAddress.from_string('12341234')
short_address = MalformedAddress.from_string('hx1234512345')
long_address_without_hx = MalformedAddress.from_string(
'cf85fac2d0b507a2db9ce9526e6d01476f16a2d269f51636f9c4b2d512017faf')
long_address = MalformedAddress.from_string(
'hxcf85fac2d0b507a2db9ce9526e6d01476f16a2d269f51636f9c4b2d512017faf'
)
self.addresses = [
empty_address, short_address_without_hx, short_address,
long_address_without_hx, long_address
]
self.db_name = 'icx.db'
db = ContextDatabase.from_path(self.db_name)
self.assertIsNotNone(db)
self.storage = IcxStorage(db)
self.factory = IconScoreContextFactory(max_size=1)
context = self.factory.create(IconScoreContextType.DIRECT)
self.context = context
def tearDown(self):
context = self.context
self.storage.close(context)
shutil.rmtree(self.db_name)
def test_get_put_account(self):
context = self.context
account = Account()
account.deposit(10**19)
for address in self.addresses:
account.address = address
self.storage.put_account(context, account.address, account)
account2 = self.storage.get_account(context, account.address)
self.assertEqual(account, account2)
def test_delete_account(self):
context = self.context
account = Account()
for address in self.addresses:
account.address = address
self.storage.put_account(context, account.address, account)
ret = self.storage.is_address_present(context, account.address)
self.assertTrue(ret)
self.storage.delete_account(context, account.address)
ret = self.storage.is_address_present(context, account.address)
self.assertFalse(ret)
class TestIcxStorage(unittest.TestCase):
def setUp(self):
self.db_name = 'icx.db'
self.address = create_address(AddressPrefix.EOA)
db = ContextDatabase.from_path(self.db_name)
self.assertIsNotNone(db)
self.storage = IcxStorage(db)
self.factory = IconScoreContextFactory(max_size=1)
context = self.factory.create(IconScoreContextType.DIRECT)
context.tx_batch = TransactionBatch()
context.block_batch = BlockBatch()
self.context = context
def tearDown(self):
context = self.context
self.storage.delete_account(context, self.address)
self.storage.close(context)
shutil.rmtree(self.db_name)
def test_get_put_account(self):
context = self.context
account = Account()
account.address = create_address(AddressPrefix.EOA)
account.deposit(10**19)
self.storage.put_account(context, account.address, account)
account2 = self.storage.get_account(context, account.address)
self.assertEqual(account, account2)
def test_delete_account(self):
context = self.context
account = Account()
account.address = create_address(AddressPrefix.EOA)
self.storage.put_account(context, account.address, account)
ret = self.storage.is_address_present(context, account.address)
self.assertTrue(ret)
self.storage.delete_account(context, account.address)
ret = self.storage.is_address_present(context, self.address)
self.assertFalse(ret)
class TestIcxStorage(unittest.TestCase):
def setUp(self):
self.db_name = 'icx.db'
self.address = create_address(AddressPrefix.EOA)
db = ContextDatabase.from_path(self.db_name)
self.assertIsNotNone(db)
self.storage = IcxStorage(db)
context = IconScoreContext(IconScoreContextType.DIRECT)
context.tx_batch = TransactionBatch()
context.block_batch = BlockBatch()
self.context = context
def tearDown(self):
context = self.context
self.storage.delete_account(context, self.address)
self.storage.close(context)
shutil.rmtree(self.db_name)
def test_get_put_account(self):
context = self.context
account = Account()
account.address = create_address(AddressPrefix.EOA)
account.deposit(10**19)
self.storage.put_account(context, account.address, account)
account2 = self.storage.get_account(context, account.address)
self.assertEqual(account, account2)
def test_delete_account(self):
context = self.context
account = Account()
account.address = create_address(AddressPrefix.EOA)
self.storage.put_account(context, account.address, account)
ret = self.storage.is_address_present(context, account.address)
self.assertTrue(ret)
self.storage.delete_account(context, account.address)
ret = self.storage.is_address_present(context, self.address)
self.assertFalse(ret)
def test_get_put_text(self):
context = self.context
key_name = 'test_genesis'
expected_text = json.dumps({
'version':
0,
'address':
str(create_address(AddressPrefix.EOA))
})
self.storage.put_text(context, key_name, expected_text)
actual_stored_text = self.storage.get_text(context, key_name)
self.assertEqual(expected_text, actual_stored_text)
def test_get_put_total_supply(self):
context = self.context
current_total_supply = self.storage.get_total_supply(context)
self.assertEqual(0, current_total_supply)
putting_total_supply_amount = 1000
self.storage.put_total_supply(context, putting_total_supply_amount)
actual_stored_total_supply = self.storage.get_total_supply(context)
self.assertEqual(putting_total_supply_amount,
actual_stored_total_supply)