def testExecute(self):
q = dalq.DALQuery("http://localhost:{0}/sia".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
results = q.execute()
self.assert_(isinstance(results, dalq.DALResults))
self.assertEquals(results.nrecs, 2)
def testExecuteStream(self):
q = dalq.DALQuery("http://localhost:{0}/sia".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
strm = q.execute_stream()
self.assert_(strm is not None)
self.assert_(hasattr(strm, "read"))
results = strm.read()
strm.close()
self.assert_(results.startswith(b"<?xml version="))
def testExecuteRaw(self):
q = dalq.DALQuery("http://localhost:{0}/sia".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
data = q.execute_raw()
self.assert_(data is not None)
if sys.version_info[0] >= 3:
self.assert_(isinstance(data, str) or isinstance(data, bytes))
else:
self.assert_(isinstance(data, unicode) or isinstance(data, str))
self.assert_(data.startswith(b"<?xml version="))
def testExecuteRawQueryErr(self):
q = dalq.DALQuery("http://localhost:{0}/err".format(self.srvr.port))
q["foo"] = "bar"
# pdb.set_trace()
data = q.execute_raw()
self.assert_(data is not None)
if sys.version_info[0] >= 3:
self.assert_(isinstance(data, str) or isinstance(data, bytes))
else:
self.assert_(isinstance(data, unicode) or isinstance(data, str))
self.assert_(data.startswith(b"<?xml version="))
self.assert_(b'<INFO name="QUERY_STATUS" value="ERR' in data)
def testExecuteStreamServiceErr(self):
q = dalq.DALQuery("http://localhost:{0}/goob".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
try:
q.execute_stream()
self.fail("failed to raise exception on bad url")
except dalq.DALServiceError as e:
self.assertEquals(e.code, 404)
self.assertEquals(e.reason, "Not Found")
self.assert_(isinstance(e.cause, HTTPError))
except Exception as e:
self.fail("wrong exception raised: " + str(type(e)))
def testExecuteStreamServiceErr(self):
q = dalq.DALQuery("http://localhost:{0}/goob".format(self.srvr.port))
q["foo"] = "bar"
# pdb.set_trace()
try:
q.execute_stream()
self.fail("failed to raise exception on bad url")
except dalq.DALServiceError as e:
self.assertEquals(e.code, 404)
self.assertIn("not found", e.reason.lower())
self.assert_(isinstance(e.cause, requests.RequestException))
except Exception as e:
self.fail("wrong exception raised: " + str(type(e)))
def testExecuteQueryErr(self):
q = dalq.DALQuery("http://localhost:{0}/err".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
try:
q.execute()
self.fail("failed to raise exception for syntax error")
except dalq.DALQueryError as e:
self.assertEquals(e.label, "ERROR")
self.assertEquals(str(e), "Forced Fail")
except dalq.DALServiceError as e:
self.fail("wrong exception raised: DALServiceError: " + str(e))
except Exception as e:
self.fail("wrong exception raised: " + str(type(e)))
def testExecuteVotableServiceErr(self):
q = dalq.DALQuery("http://localhost:{0}/goob".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
self.assertRaises(dalq.DALServiceError, q.execute_votable)
def testExecuteVotable(self):
q = dalq.DALQuery("http://localhost:{0}/sia".format(self.srvr.port))
q.setparam("foo", "bar")
# pdb.set_trace()
results = q.execute_votable()
self.assert_(isinstance(results, VOTableFile))
def testCtor(self):
self.query = dalq.DALQuery(self.baseurl, "sga", "2.0")
self.assert_(self.query.getparam("format") is None)
def testExecuteRawServiceErr(self):
q = dalq.DALQuery("http://localhost:{0}/goob".format(self.srvr.port))
q["foo"] = "bar"
# pdb.set_trace()
self.assertRaises(dalq.DALServiceError, q.execute_raw)
