chainsyncer/tests/test_database.py

# standard imports
import unittest
import logging

# external imports
from chainlib.chain import ChainSpec

# local imports
from chainsyncer.db.models.base import SessionBase
from chainsyncer.db.models.filter import BlockchainSyncFilter
from chainsyncer.backend.sql import SQLBackend
from chainsyncer.error import LockError

# testutil imports
from tests.chainsyncer_base import TestBase

logg = logging.getLogger()


class TestDatabase(TestBase):


    def test_backend_live(self):
        s = SQLBackend.live(self.chain_spec, 42)
        self.assertEqual(s.object_id, 1)
        backend = SQLBackend.first(self.chain_spec)
        #SQLBackend(self.chain_spec, sync_id)
        self.assertEqual(backend.object_id, 1)

        bogus_chain_spec = ChainSpec('bogus', 'foo', 13, 'baz')
        sync_id = SQLBackend.first(bogus_chain_spec)
        self.assertIsNone(sync_id)


    def test_backend_filter_lock(self):
        s = SQLBackend.live(self.chain_spec, 42)

        s.connect()
        filter_id = s.db_object_filter.id
        s.disconnect()

        session = SessionBase.create_session()
        o = session.query(BlockchainSyncFilter).get(filter_id)
        self.assertEqual(len(o.flags), 0)
        session.close()

        s.register_filter(str(0))
        s.register_filter(str(1))

        s.connect()
        filter_id = s.db_object_filter.id
        s.disconnect()

        session = SessionBase.create_session()
        o = session.query(BlockchainSyncFilter).get(filter_id)

        o.set(1)
        with self.assertRaises(LockError):
            o.set(2)
        o.release()
        o.set(2)


    def test_backend_filter(self):
        s = SQLBackend.live(self.chain_spec, 42)

        s.connect()
        filter_id = s.db_object_filter.id
        s.disconnect()

        session = SessionBase.create_session()
        o = session.query(BlockchainSyncFilter).get(filter_id)
        self.assertEqual(len(o.flags), 0)
        session.close()

        for i in range(9):
            s.register_filter(str(i))

        s.connect()
        filter_id = s.db_object_filter.id
        s.disconnect()

        session = SessionBase.create_session()
        o = session.query(BlockchainSyncFilter).get(filter_id)
        self.assertEqual(len(o.flags), 2)

        (t, c, d) = o.target()
        self.assertEqual(t, (1 << 9) - 1)

        for i in range(9):
            o.set(i)
            o.release()

        (f, c, d) = o.cursor()
        self.assertEqual(f, t)
        self.assertEqual(c, 9)
        self.assertEqual(d, o.digest)

        session.close()


    def test_backend_retrieve(self):
        s = SQLBackend.live(self.chain_spec, 42)
        s.register_filter('foo')
        s.register_filter('bar')
        s.register_filter('baz')

        s.set(42, 13)

        s = SQLBackend.first(self.chain_spec)
        self.assertEqual(s.get(), ((42,13), 0))


    def test_backend_initial(self):
        with self.assertRaises(ValueError):
            s = SQLBackend.initial(self.chain_spec, 42, 42)
        
        with self.assertRaises(ValueError):
            s = SQLBackend.initial(self.chain_spec, 42, 43)
        
        s = SQLBackend.initial(self.chain_spec, 42, 13)

        s.set(43, 13)

        s = SQLBackend.first(self.chain_spec)
        self.assertEqual(s.get(), ((43,13), 0))
        self.assertEqual(s.start(), ((13,0), 0))


    def test_backend_resume(self):
        s = SQLBackend.resume(self.chain_spec, 666)
        self.assertEqual(len(s), 0)

        s = SQLBackend.live(self.chain_spec, 42)
        original_id = s.object_id
        s = SQLBackend.resume(self.chain_spec, 666)
        self.assertEqual(len(s), 1)
        resumed_id = s[0].object_id
        self.assertEqual(resumed_id, original_id + 1)
        self.assertEqual(s[0].get(), ((42, 0), 0))


    def test_backend_resume_when_completed(self):
        s = SQLBackend.live(self.chain_spec, 42)

        s = SQLBackend.resume(self.chain_spec, 666)
        s[0].set(666, 0)
    
        s = SQLBackend.resume(self.chain_spec, 666)
        self.assertEqual(len(s), 0)
        
    
    def test_backend_resume_several(self):
        s = SQLBackend.live(self.chain_spec, 42)
        s.set(43, 13)
        
        s = SQLBackend.resume(self.chain_spec, 666)
        SQLBackend.live(self.chain_spec, 666)
        s[0].set(123, 2)

        s = SQLBackend.resume(self.chain_spec, 1024)
        SQLBackend.live(self.chain_spec, 1024) 

        self.assertEqual(len(s), 2)
        self.assertEqual(s[0].target(), (666, 0))
        self.assertEqual(s[0].get(), ((123, 2), 0))
        self.assertEqual(s[1].target(), (1024, 0))
        self.assertEqual(s[1].get(), ((666, 0), 0))


    def test_backend_resume_filter(self):
        s = SQLBackend.live(self.chain_spec, 42)
        s.register_filter('foo')
        s.register_filter('bar')
        s.register_filter('baz')

        s.set(43, 13)
        s.begin_filter(0)
        s.begin_filter(2)

        s = SQLBackend.resume(self.chain_spec, 666)
        (pair, flags) = s[0].get()

        self.assertEqual(flags, 5)


    def test_backend_sql_custom(self):
        chain_spec = ChainSpec('evm', 'bloxberg', 8996, 'foo')
        flags = 5
        flags_target = 1023
        flag_count = 10
        backend = SQLBackend.custom(chain_spec, 666, 42, 2, flags, flag_count)
        self.assertEqual(((42, 2), flags), backend.start())
        self.assertEqual(((42, 2), flags), backend.get())
        self.assertEqual((666, flags_target), backend.target())


if __name__ == '__main__':
    unittest.main()