def test_insert_balance(self):
"""Test adding balance for new address """
storage = SQLBalanceStorage(self.db_session)
with self.assertRaises(KeyError):
storage.get("address_one")
storage.update(insert={"address_one": 32})
self.assertEqual(storage.get("address_one"), 32)
#
with make_session_scope(self.db_session) as session:
addr_bal = session.query(AddressBalance)\
.filter_by(address="address_one")\
.first()
self.assertEqual(addr_bal.balance, 32)
# Error when insertion an existing address
with self.assertRaises(sqlalchemy.exc.IntegrityError):
storage.update(insert={'address_one': 33})
# Insert several at a time
storage.update(insert={"a1": 55, "a2": 66})
self.assertEqual(storage.get('a1'), 55)
self.assertEqual(storage.get('a2'), 66)
def test_delete_balance(self):
"""Test deleting existing address"""
storage = SQLBalanceStorage(self.db_session)
storage.update(insert={'addr1': 33, 'addr2': 44, 'addr3': 55})
self.assertEqual(storage.get('addr1'), 33)
self.assertEqual(storage.get('addr2'), 44)
storage.update(delete=['addr1', 'addr2'])
self.assertEqual(storage.get('addr1', 99), 99)
self.assertEqual(storage.get('addr2', 88), 88)
with make_session_scope(self.db_session) as session:
addr1 = session.query(AddressBalance)\
.filter_by(address="addr1")\
.first()
addr2 = session.query(AddressBalance)\
.filter_by(address="addr2")\
.first()
self.assertEqual(addr1, None)
self.assertEqual(addr2, None)
# Check deleting unknown address
storage.update(delete=['addr1'])
def test_get_balance(self):
"""Test get method with and without default"""
storage = SQLBalanceStorage(self.db_session)
with make_session_scope(self.db_session) as session:
session.add(AddressBalance(address='address_one', balance=44))
session.add(AddressBalance(address='address_two', balance=55))
self.assertEqual(storage.get('address_one'), 44)
self.assertEqual(storage.get('address_two'), 55)
self.assertEqual(storage.get('address_one', 99), 44)
with self.assertRaises(KeyError):
storage.get('address_three')
self.assertEqual(storage.get('address_three', 3), 3)
def test_mixed_ops(self):
"""Test mixing inserts/updates/deletions in a single operation"""
storage = SQLBalanceStorage(self.db_session)
insert = {str(x): x for x in range(100)}
storage.update(insert=insert)
for a in range(100):
self.assertEqual(storage.get(str(a)), a)
# Insert and update
insert = {str(x): x for x in range(100, 110)}
update = {str(x): 5 for x in range(100)}
storage.update(insert=insert, update=update)
for a in range(100):
self.assertEqual(storage.get(str(a)), 5)
for a in range(100, 110):
self.assertEqual(storage.get(str(a)), a)
# Insert udpate and delete
insert = {str(x): x for x in range(200, 300)}
update = {str(x): 7 for x in range(50)}
delete = [str(x) for x in range(50, 100)]
storage.update(insert=insert, update=update, delete=delete, height=10)
for a in range(50):
self.assertEqual(storage.get(str(a)), 7)
for a in range(200, 300):
self.assertEqual(storage.get(str(a)), a)
for a in range(50, 100):
self.assertEqual(storage.get(str(a), 99999), 99999)
def test_udpate_balance(self):
storage = SQLBalanceStorage(self.db_session)
storage.update(insert={"address_four": 44})
# Error when insertion an existing address
with self.assertRaises(sqlalchemy.exc.IntegrityError):
storage.update(insert={'address_four': 33})
# Fine when updating an existing balance
storage.update(update={'address_four': 777})
self.assertEqual(storage.get('address_four'), 777)
with make_session_scope(self.db_session) as session:
addr_bal = session.query(AddressBalance)\
.filter_by(address="address_one")\
.first()
# Update several address at a time
storage.update(insert={'b1': 11, 'b2': 22})
self.assertEqual(storage.get('b1'), 11)
self.assertEqual(storage.get('b2'), 22)
storage.update(update={'b1': 44, 'b2': 55})
self.assertEqual(storage.get('b1'), 44)
self.assertEqual(storage.get('b2'), 55)
with make_session_scope(self.db_session) as session:
addr1 = session.query(AddressBalance)\
.filter_by(address="b1")\
.first()
addr2 = session.query(AddressBalance)\
.filter_by(address="b2")\
.first()
self.assertEqual(addr1.balance, 44)
self.assertEqual(addr2.balance, 55)