34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
"""
|
||
|
SQL related core tests
|
||
|
"""
|
||
|
|
||
|
import unittest
|
||
|
|
||
|
import helper_db
|
||
|
|
||
|
|
||
|
class TestSQL(unittest.TestCase):
|
||
|
"""A test case for common SQL-related functions"""
|
||
|
test_address_1 = 'BM-2cVRo3WuKL5vDCKq2XJi1yR366nkHgHfcL'
|
||
|
test_address_2 = 'BM-2cVJn1BusL1j4UezuYkvpqNPCr7kCebeLw'
|
||
|
|
||
|
def test_subscriptions(self):
|
||
|
"""Put address into subscriptions and check that it was added"""
|
||
|
self.assertTrue(
|
||
|
helper_db.put_subscriptions('pass_1', self.test_address_1))
|
||
|
self.assertFalse(
|
||
|
helper_db.put_subscriptions('pass_2', self.test_address_1))
|
||
|
|
||
|
helper_db.put_subscriptions('pass_3', self.test_address_2, False)
|
||
|
|
||
|
for label, addr in helper_db.get_subscriptions():
|
||
|
if addr == self.test_address_1:
|
||
|
self.assertEqual(label, 'pass_1')
|
||
|
break
|
||
|
else:
|
||
|
self.fail('Could not find added address in subscriptions')
|
||
|
|
||
|
for label, addr in helper_db.get_subscriptions():
|
||
|
if addr == self.test_address_2:
|
||
|
self.fail('Got disabled subscription')
|