class TestPubMed(unittest.TestCase): def setUp(self): Registry.load_default_settings() s = Registry.getSchema("org.pybliographer/pubmed/0.1") fmt = Store.get("memory") self.db = fmt.dbcreate(None, s) self.cnx = PubMed(self.db) self.port = reactor.listenTCP(8000, Server()) def tearDown(self): self.port.stopListening() Registry.reset() return def testCount(self): """ Test that one can get a result count """ self.cnx.baseURL = "http://localhost:8000/count" d = self.cnx.count("gobry") def check(count): self.failUnlessEqual(count, 7) return d.addCallback(check) def testSearch(self): """ Test a simple search. """ self.cnx.baseURL = "http://localhost:8000/search" self.cnx.BATCH_SIZE = 5 d, rs = self.cnx.search(u"gobry véronique") def check(count): self.failUnlessEqual(len(rs), 7) return d.addCallback(check) def testFailure(self): """ Check that error codes are properly detected.""" self.cnx.baseURL = "http://localhost:8000/broken" d = self.cnx.count(query="TOTO") def check(e): if e.check(unittest.FailTest): return e return d.addCallback(self.fail).addErrback(check)
def setUp(self): Registry.load_default_settings() s = Registry.getSchema("org.pybliographer/pubmed/0.1") fmt = Store.get("memory") self.db = fmt.dbcreate(None, s) self.cnx = PubMed(self.db) self.port = reactor.listenTCP(8000, Server())