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