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