def test_items_multi_hash_returns_all(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h[0, '1'] = 123
h[0, '2'] = 456
h[0, '3'] = 789
h[1, '1'] = 999
h[1, '2'] = 888
h[1, '3'] = 777
driver.commit()
kvs = {
'blah.scoob:0:3': 789,
'blah.scoob:0:1': 123,
'blah.scoob:0:2': 456,
'blah.scoob:1:3': 777,
'blah.scoob:1:1': 999,
'blah.scoob:1:2': 888
}
got = h._items()
self.assertDictEqual(kvs, got)
def test_multihash_multiple_dims_clear_behaves_similar_to_single_dim(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h[1, 0, '1'] = 123
h[1, 0, '2'] = 456
h[1, 0, '3'] = 789
h[1, 1, '1'] = 999
h[1, 1, '2'] = 888
h[1, 1, '3'] = 777
driver.commit()
kvs = {
'blah.scoob:1:0:3': 789,
'blah.scoob:1:0:1': 123,
'blah.scoob:1:0:2': 456
}
h.clear(1, 1)
driver.commit()
got = h._items()
self.assertDictEqual(kvs, got)
def test_items_clear_deletes_only_multi_hash(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h[0, '1'] = 123
h[0, '2'] = 456
h[0, '3'] = 789
h[1, '1'] = 999
h[1, '2'] = 888
h[1, '3'] = 777
driver.commit()
kvs = {
'blah.scoob:0:3': 789,
'blah.scoob:0:1': 123,
'blah.scoob:0:2': 456
}
h.clear(1)
driver.commit()
got = h._items()
self.assertDictEqual(kvs, got)
def test_get_all_when_none_exist(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
self.assertEqual(h.all(), [])
def test_get(self):
contract = 'stustu'
name = 'balance'
delimiter = driver.delimiter
raw_key_1 = '{}{}{}'.format(contract, delimiter, name)
raw_key_1 += ':stu'
driver.set(raw_key_1, 1234)
h = Hash(contract, name, driver=driver)
self.assertEqual(h._get('stu'), 1234)
def test_get(self):
# set up the foreign variable
contract = 'stustu'
name = 'balance'
f_contract = 'colinbucks'
f_name = 'balances'
f = ForeignHash(contract, name, f_contract, f_name, driver=driver)
h = Hash(f_contract, f_name, driver=driver)
h._set('howdy', 555)
self.assertEqual(f._get('howdy'), 555)
def test_set(self):
contract = 'stustu'
name = 'balance'
delimiter = driver.delimiter
raw_key_1 = '{}{}{}'.format(contract, delimiter, name)
raw_key_1 += ':stu'
h = Hash(contract, name, driver=driver)
h._set('stu', 1234)
driver.commit()
self.assertEqual(driver.get(raw_key_1), 1234)
def test_setitem_delimiter_illegal(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver)
with self.assertRaises(AssertionError):
h['stu:123'] = 123
def test_get_all_after_setting(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h['1'] = 123
h['2'] = 456
h['3'] = 789
l = [123, 456, 789]
driver.commit()
# we care about whats included, not order
self.assertSetEqual(set(h.all()), set(l))
def test_getitem_returns_default_value_if_none(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
self.assertEqual(h['hello'], 0)
def test_items_returns_kv_pairs(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h['1'] = 123
h['2'] = 456
h['3'] = 789
driver.commit()
kvs = {'blah.scoob:3': 789, 'blah.scoob:1': 123, 'blah.scoob:2': 456}
got = h._items()
self.assertDictEqual(kvs, got)
def test_getitems_too_many_dimensions_fails(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver)
with self.assertRaises(Exception):
a = h['a', 'b', 'c', 'a', 'b', 'c', 'a', 'b', 'c', 'a', 'b', 'c',
'a', 'b', 'c', 'a', 'b', 'c']
def test_getitems_key_too_large(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver)
key = 'a' * 1025
with self.assertRaises(Exception):
a = h[key]
def test_setitems(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver)
h['stu'] = 123
h['stu', 'raghu'] = 1000
driver.commit()
val = driver.get('blah.scoob:stu:raghu')
self.assertEqual(val, 1000)
def test_all_multihash_returns_values(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h[0, '1'] = 123
h[0, '2'] = 456
h[0, '3'] = 789
h[1, '1'] = 999
h[1, '2'] = 888
h[1, '3'] = 777
l = [123, 456, 789]
driver.commit()
# we care about whats included, not order
self.assertSetEqual(set(h.all(0)), set(l))
def test_multihash_multiple_dims_all_gets_items_similar_to_single_dim(
self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h[1, 0, '1'] = 123
h[1, 0, '2'] = 456
h[1, 0, '3'] = 789
h[1, 1, '1'] = 999
h[1, 1, '2'] = 888
h[1, 1, '3'] = 777
l = [123, 456, 789]
driver.commit()
# we care about whats included, not order
self.assertSetEqual(set(h.all(1, 0)), set(l))
def test_getsetitems(self):
contract = 'blah'
name = 'scoob'
delimiter = driver.delimiter
h = Hash(contract, name, driver=driver)
h['stu', 'raghu'] = 999
driver.commit()
self.assertEqual(h['stu', 'raghu'], 999)
def test_getitems_keys_too_large(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver)
key1 = 'a' * 800
key2 = 'b' * 100
key3 = 'c' * 200
with self.assertRaises(Exception):
x = h[key1, key2, key3]
def test_clear_items_deletes_all_key_value_pairs(self):
contract = 'blah'
name = 'scoob'
h = Hash(contract, name, driver=driver, default_value=0)
h['1'] = 123
h['2'] = 456
h['3'] = 789
driver.commit()
kvs = {'blah.scoob:3': 789, 'blah.scoob:1': 123, 'blah.scoob:2': 456}
got = h._items()
self.assertDictEqual(kvs, got)
h.clear()
driver.commit()
got = h._items()
self.assertDictEqual({}, got)
def test_set_get(self):
contract = 'stustu'
name = 'balance'
h = Hash(contract, name, driver=driver)
h._set('stu', 1234)
_h = h._get('stu')
self.assertEqual(_h, 1234)
h._set('colin', 5678)
_h2 = h._get('colin')
self.assertEqual(_h2, 5678)
def test_setitem(self):
contract = 'blah'
name = 'scoob'
delimiter = driver.delimiter
h = Hash(contract, name, driver=driver)
prefix = '{}{}{}{}'.format(contract, delimiter, name, h._delimiter)
h['stu'] = 9999999
raw_key = '{}stu'.format(prefix)
self.assertEqual(driver.get(raw_key), 9999999)
def test_getitem(self):
# set up the foreign variable
contract = 'stustu'
name = 'balance'
f_contract = 'colinbucks'
f_name = 'balances'
f = ForeignHash(contract, name, f_contract, f_name, driver=driver)
h = Hash(f_contract, f_name, driver=driver)
h['howdy'] = 555
self.assertEqual(f['howdy'], 555)
def test_getitem(self):
contract = 'blah'
name = 'scoob'
delimiter = driver.delimiter
h = Hash(contract, name, driver=driver)
prefix = '{}{}{}{}'.format(contract, delimiter, name, h._delimiter)
raw_key = '{}stu'.format(prefix)
driver.set(raw_key, 54321)
self.assertEqual(h['stu'], 54321)
# Monkey patch for testing, as this is purely for 'interface enforcement' testing
from contracting.db.orm import Variable, Hash
class ctx:
caller = 1
__supply = Variable(contract='__main__', name='supply')
__balances = Hash(default_value=0, contract='__main__', name='balances')
def ____():
__balances['stu'] = 1000000
__balances['colin'] = 100
__supply.set(__balances['stu'] + __balances['colin'])
def transfer(amount, to):
sender = ctx.caller
assert __balances[sender] >= amount, 'Not enough coins to send!'
__balances[sender] -= amount
__balances[to] += amount
def balance_of(account):
return __balances[account]
def total_supply():
return __supply.get()