def test_1_depth_dict_db(self, context, key_value_db, keys, old_values, new_values, value_type):
context_db = ContextDatabase(key_value_db, is_shared=False)
score_address = Address(AddressPrefix.CONTRACT, os.urandom(20))
score_db = IconScoreDatabase(score_address, context_db)
self._init_context(context, score_address)
self._set_revision(context, Revision.USE_RLP.value - 1)
name = "dict_db_depth_1"
dict_db = DictDB(name, score_db, depth=1, value_type=value_type)
# Put two items to dict_db
for i in range(2):
k, v = keys[i], old_values[i]
dict_db[k] = v
assert dict_db[k] == v
key = _get_final_key(
score_address, ContainerTag.DICT, name.encode(), ContainerUtil.encode_key(k),
use_rlp=False
)
assert key_value_db.get(key) == ContainerUtil.encode_value(v)
self._set_revision(context, Revision.USE_RLP.value)
# Read old-formatted data on Revision.USE_RLP
for i in range(2):
k, v = keys[i], old_values[i]
assert dict_db[k] == v
key = _get_final_key(
score_address, ContainerTag.DICT, name.encode(), ContainerUtil.encode_key(k),
use_rlp=False
)
assert key_value_db.get(key) == ContainerUtil.encode_value(v)
# Put 4 items to dict_db
for i, k in enumerate(keys):
old_v = old_values[i]
new_v = new_values[i]
dict_db[k] = new_v
assert dict_db[k] == new_v
assert dict_db[k] != old_v
key = _get_final_key(
score_address, ContainerTag.DICT, name.encode(), ContainerUtil.encode_key(k),
use_rlp=True
)
assert key_value_db.get(key) == ContainerUtil.encode_value(new_v)
# If there is no value for a given key, default value is returned
for k in keys:
del dict_db[k]
assert dict_db[k] == get_default_value(value_type)
assert len(key_value_db) == 0
def test_dict_depth2(self, score_db):
name = 'test_dict'
test_dict = DictDB(name, score_db, depth=3, value_type=int)
test_dict['a']['b']['c'] = 1
test_dict['a']['b']['d'] = 2
test_dict['a']['b']['e'] = 3
test_dict['a']['b']['f'] = 4
assert test_dict['a']['b']['c'] == 1
def test_success_dict_depth1(self):
name = 'test_dict'
test_dict = DictDB(name, self.db, value_type=int)
prefix: bytes = ContainerUtil.create_db_prefix(DictDB, name)
self.assertEqual(b'\x01|' + name.encode(), prefix)
test_dict['a'] = 1
test_dict['b'] = 2
self.assertEqual(test_dict['a'], 1)
def test_dict_depth2(self, score_db):
name = 'test_dict'
test_dict = DictDB(name, score_db, depth=3, value_type=int)
self._check_the_db_prefix_format(name)
test_dict['a']['b']['c'] = 1
test_dict['a']['b']['d'] = 2
test_dict['a']['b']['e'] = 3
test_dict['a']['b']['f'] = 4
assert test_dict['a']['b']['c'] == 1
def test_dict_other_Key(self, score_db):
name = 'test_dict'
test_dict = DictDB(name, score_db, depth=2, value_type=int)
addr1 = create_address(1)
addr2 = create_address(0)
test_dict['a'][addr1] = 1
test_dict['a'][addr2] = 2
assert test_dict['a'][addr1] == 1
assert test_dict['a'][addr2] == 2
def test_dict_depth1(self, score_db):
name = 'test_dict'
test_dict = DictDB(name, score_db, value_type=int)
test_dict['a'] = 1
test_dict['b'] = 2
test_dict['b'] += 1
assert test_dict['a'] == 1
assert test_dict['b'] == 3
def test_dict_depth1(self, score_db):
name = 'test_dict'
test_dict = DictDB(name, score_db, value_type=int)
self._check_the_db_prefix_format(name)
test_dict['a'] = 1
test_dict['b'] = 2
test_dict['b'] += 1
assert test_dict['a'] == 1
assert test_dict['b'] == 3
def test_success_dict_other_Key(self):
name = 'test_dict'
test_dict = DictDB(name, self.db, depth=2, value_type=int)
prefix: bytes = ContainerUtil.create_db_prefix(DictDB, name)
self.assertEqual(b'\x01|' + name.encode(), prefix)
addr1 = create_address(1)
addr2 = create_address(0)
test_dict['a'][addr1] = 1
test_dict['a'][addr2] = 2
self.assertEqual(test_dict['a'][addr1], 1)
self.assertEqual(test_dict['a'][addr2], 2)
def test_2_depth_dict_db(self, context, key_value_db, keys1, keys2, old_values, new_values, value_type):
context_db = ContextDatabase(key_value_db, is_shared=False)
score_address = Address(AddressPrefix.CONTRACT, os.urandom(20))
score_db = IconScoreDatabase(score_address, context_db)
self._init_context(context, score_address)
self._set_revision(context, Revision.USE_RLP.value - 1)
name = "dict_db_depth_2"
dict_db = DictDB(name, score_db, depth=2, value_type=value_type)
# To assign a value to middle-layer dict_db is forbidden
for k1, v in zip(keys1, old_values):
with pytest.raises(InvalidContainerAccessException):
dict_db[k1] = v
# Assign values to dict_db on Revision.USE_RLP - 1
for k1 in keys1:
for k2, v in zip(keys2, old_values):
dict_db[k1][k2] = v
assert len(key_value_db) == len(keys1) * len(keys2)
for k1 in keys1:
for k2, v in zip(keys2, old_values):
assert dict_db[k1][k2] == v
key: bytes = _get_final_key(
score_address,
ContainerTag.DICT, name.encode(),
ContainerTag.DICT, ContainerUtil.encode_key(k1),
ContainerUtil.encode_key(k2),
use_rlp=False
)
assert key_value_db.get(key) == ContainerUtil.encode_value(v)
self._set_revision(context, Revision.USE_RLP.value)
# Check if reading old-formatted key:value data works on Revision.USE_RLP
for k1 in keys1:
for k2, v in zip(keys2, old_values):
assert dict_db[k1][k2] == v
# Replace all old_values with new_values on Revision.USE_RLP
for k1 in keys1:
for k2, v in zip(keys2, new_values):
dict_db[k1][k2] = v
# old_values + new_values
assert len(key_value_db) == len(keys1) * len(keys2) * 2
for k1 in keys1:
for k2, v in zip(keys2, new_values):
assert dict_db[k1][k2] == v
key: bytes = _get_final_key(
score_address,
ContainerTag.DICT, name.encode(),
ContainerUtil.encode_key(k1),
ContainerUtil.encode_key(k2),
use_rlp=True
)
assert key_value_db.get(key) == ContainerUtil.encode_value(v)
for k1 in keys1:
for k2 in keys2:
del dict_db[k1][k2]
assert dict_db[k1][k2] == get_default_value(value_type)
assert len(key_value_db) == 0