class TestSqlitePeerDBHandler(AbstractDB):
@blocking_call_on_reactor_thread
@inlineCallbacks
def setUp(self):
yield super(TestSqlitePeerDBHandler, self).setUp()
self.p1 = str2bin(
'MFIwEAYHKoZIzj0CAQYFK4EEABoDPgAEAAA6SYI4NHxwQ8P7P8QXgWAP+v8SaMVzF5+fSUHdAMrs6NvL5Epe1nCNSdlBHIjNjEiC5iiwSFZhRLsr'
)
self.p2 = str2bin(
'MFIwEAYHKoZIzj0CAQYFK4EEABoDPgAEAABo69alKy95H7RHzvDCsolAurKyrVvtDdT9/DzNAGvky6YejcK4GWQXBkIoQGQgxVEgIn8dwaR9B+3U'
)
self.pdb = PeerDBHandler(self.session)
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
@blocking_call_on_reactor_thread
@inlineCallbacks
def tearDown(self):
self.pdb.close()
self.pdb = None
yield super(TestSqlitePeerDBHandler, self).tearDown()
@blocking_call_on_reactor_thread
def test_getList(self):
peer1 = self.pdb.getPeer(self.p1)
peer2 = self.pdb.getPeer(self.p2)
self.assertIsInstance(peer1, dict)
self.assertIsInstance(peer2, dict)
self.assertEqual(peer1[u'peer_id'], 1)
self.assertEqual(peer2[u'peer_id'], 2)
@blocking_call_on_reactor_thread
def test_addPeer(self):
peer_x = {'permid': FAKE_PERMID_X, 'name': 'fake peer x'}
oldsize = self.pdb.size()
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.assertEqual(self.pdb.size(), oldsize + 1)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertEqual(p['name'], 'fake peer x')
self.assertEqual(self.pdb.getPeer(FAKE_PERMID_X, 'name'),
'fake peer x')
self.pdb.deletePeer(FAKE_PERMID_X)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.assertEqual(self.pdb.size(), oldsize)
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.pdb.addPeer(FAKE_PERMID_X, {
'permid': FAKE_PERMID_X,
'name': 'faka peer x'
})
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertEqual(p['name'], 'faka peer x')
@blocking_call_on_reactor_thread
def test_aa_hasPeer(self):
self.assertTrue(self.pdb.hasPeer(self.p1))
self.assertTrue(self.pdb.hasPeer(self.p1, check_db=True))
self.assertTrue(self.pdb.hasPeer(self.p2))
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
@blocking_call_on_reactor_thread
def test_deletePeer(self):
peer_x = {'permid': FAKE_PERMID_X, 'name': 'fake peer x'}
oldsize = self.pdb.size()
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.assertEqual(self.pdb.size(), oldsize + 1)
self.assertTrue(self.pdb.hasPeer(FAKE_PERMID_X))
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNotNone(p)
self.pdb.deletePeer(FAKE_PERMID_X)
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
self.assertEqual(self.pdb.size(), oldsize)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.assertFalse(self.pdb.deletePeer(FAKE_PERMID_X))
@blocking_call_on_reactor_thread
def test_add_or_get_peer(self):
self.assertIsInstance(self.pdb.addOrGetPeerID(FAKE_PERMID_X), int)
self.assertIsInstance(self.pdb.addOrGetPeerID(FAKE_PERMID_X), int)
@blocking_call_on_reactor_thread
def test_get_peer_by_id(self):
self.assertEqual(self.pdb.getPeerById(1, ['name']), 'Peer 1')
p = self.pdb.getPeerById(1)
self.assertEqual(p['name'], 'Peer 1')
self.assertFalse(self.pdb.getPeerById(1234567))
class TestSqlitePeerDBHandler(AbstractDB):
@inlineCallbacks
def setUp(self):
yield super(TestSqlitePeerDBHandler, self).setUp()
self.p1 = str2bin(
'MFIwEAYHKoZIzj0CAQYFK4EEABoDPgAEAAA6SYI4NHxwQ8P7P8QXgWAP+v8SaMVzF5+fSUHdAMrs6NvL5Epe1nCNSdlBHIjNjEiC5iiwSFZhRLsr')
self.p2 = str2bin(
'MFIwEAYHKoZIzj0CAQYFK4EEABoDPgAEAABo69alKy95H7RHzvDCsolAurKyrVvtDdT9/DzNAGvky6YejcK4GWQXBkIoQGQgxVEgIn8dwaR9B+3U')
self.pdb = PeerDBHandler(self.session)
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
@inlineCallbacks
def tearDown(self):
self.pdb.close()
self.pdb = None
yield super(TestSqlitePeerDBHandler, self).tearDown()
def test_getList(self):
peer1 = self.pdb.getPeer(self.p1)
peer2 = self.pdb.getPeer(self.p2)
self.assertIsInstance(peer1, dict)
self.assertIsInstance(peer2, dict)
self.assertEqual(peer1[u'peer_id'], 1)
self.assertEqual(peer2[u'peer_id'], 2)
def test_addPeer(self):
peer_x = {'permid': FAKE_PERMID_X, 'name': 'fake peer x'}
oldsize = self.pdb.size()
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.assertEqual(self.pdb.size(), oldsize + 1)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertEqual(p['name'], 'fake peer x')
self.assertEqual(self.pdb.getPeer(FAKE_PERMID_X, 'name'), 'fake peer x')
self.pdb.deletePeer(FAKE_PERMID_X)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.assertEqual(self.pdb.size(), oldsize)
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.pdb.addPeer(FAKE_PERMID_X, {'permid': FAKE_PERMID_X, 'name': 'faka peer x'})
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertEqual(p['name'], 'faka peer x')
def test_aa_hasPeer(self):
self.assertTrue(self.pdb.hasPeer(self.p1))
self.assertTrue(self.pdb.hasPeer(self.p1, check_db=True))
self.assertTrue(self.pdb.hasPeer(self.p2))
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
def test_deletePeer(self):
peer_x = {'permid': FAKE_PERMID_X, 'name': 'fake peer x'}
oldsize = self.pdb.size()
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.pdb.addPeer(FAKE_PERMID_X, peer_x)
self.assertEqual(self.pdb.size(), oldsize + 1)
self.assertTrue(self.pdb.hasPeer(FAKE_PERMID_X))
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNotNone(p)
self.pdb.deletePeer(FAKE_PERMID_X)
self.assertFalse(self.pdb.hasPeer(FAKE_PERMID_X))
self.assertEqual(self.pdb.size(), oldsize)
p = self.pdb.getPeer(FAKE_PERMID_X)
self.assertIsNone(p)
self.assertFalse(self.pdb.deletePeer(FAKE_PERMID_X))
def test_add_or_get_peer(self):
self.assertIsInstance(self.pdb.addOrGetPeerID(FAKE_PERMID_X), int)
self.assertIsInstance(self.pdb.addOrGetPeerID(FAKE_PERMID_X), int)
def test_get_peer_by_id(self):
self.assertEqual(self.pdb.getPeerById(1, ['name']), 'Peer 1')
p = self.pdb.getPeerById(1)
self.assertEqual(p['name'], 'Peer 1')
self.assertFalse(self.pdb.getPeerById(1234567))
