def test_connection(self):
with HTTMock(local_mock):
query = MyriaQuery(QUERY_ID, connection=self.connection)
self.assertEqual(query.connection, self.connection)
query = MyriaQuery(QUERY_ID)
self.assertEqual(query.connection, MyriaRelation.DefaultConnection)
def test_status(self):
with HTTMock(local_mock):
query = MyriaQuery(COMPLETED_QUERY_ID, connection=self.connection)
self.assertEqual(query.status, STATE_SUCCESS)
query = MyriaQuery(RUNNING_QUERY_ID, connection=self.connection)
self.assertEqual(query.status, STATE_RUNNING)
def test_dict(self):
with HTTMock(local_mock):
query = MyriaQuery(COMPLETED_QUERY_ID, connection=self.connection)
self.assertEqual(query.to_dict(), TUPLES)
query = MyriaQuery(RUNNING_QUERY_ID,
connection=self.connection,
timeout=1)
self.assertRaises(requests.Timeout, query.to_dict)
def test_dict(self):
with HTTMock(local_mock):
query = MyriaQuery(COMPLETED_QUERY_ID, connection=self.connection)
self.assertEqual(query.to_dict(), TUPLES)
for i in xrange(1, len(TUPLES) + 1):
self.assertEqual(query.to_dict(limit=i), TUPLES[:i])
query = MyriaQuery(RUNNING_QUERY_ID,
connection=self.connection,
timeout=1)
self.assertRaises(requests.Timeout, query.to_dict)
def test_timeout(self):
timeout = 999
with HTTMock(local_mock):
query = MyriaQuery(QUERY_ID,
connection=self.connection,
timeout=timeout)
self.assertEqual(query.timeout, timeout)
def test_wait_for_completion(self):
with HTTMock(local_mock):
query = MyriaQuery(COMPLETED_QUERY_ID,
connection=self.connection,
wait_for_completion=True,
timeout=1)
self.assertEqual(query.status, STATE_SUCCESS)
self.assertRaises(requests.Timeout,
MyriaQuery,
RUNNING_QUERY_ID,
connection=self.connection,
wait_for_completion=True,
timeout=1)
query = MyriaQuery(RUNNING_QUERY_ID,
connection=self.connection,
wait_for_completion=False,
timeout=1)
self.assertRaises(requests.Timeout, query.wait_for_completion)
def test_name(self):
with HTTMock(local_mock):
query = MyriaQuery(COMPLETED_QUERY_ID, connection=self.connection)
self.assertEqual(query.name, FULL_NAME)
self.assertEqual(query.qualified_name, QUALIFIED_NAME)
self.assertEqual(query.components, NAME_COMPONENTS)
def test_id(self):
with HTTMock(local_mock):
query = MyriaQuery(QUERY_ID, connection=self.connection)
self.assertEqual(query.query_id, QUERY_ID)
