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))