SNOW-27749: query_results method doesn't return sfqid and sqlstate in Python Connector.

Author: smtakeda

cursor.py

@@ -589,6 +589,14 @@ def chunk_info(self, data, use_ijson=False):
     def query_result(self, qid, _use_ijson=False):
         url = ('/queries/{qid}/result').format(qid=qid)
         ret = self._connection._con.request(url=url, method='get')
+        self._sfqid = ret[u'data'][
+            u'queryId'] if u'data' in ret and u'queryId' in ret[
+            u'data'] else None
+        self._sqlstate = ret[u'data'][
+            u'sqlState'] if u'data' in ret and u'sqlState' in ret[
+            u'data'] else None
+        self.logger.debug(u'sfqid=%s', self._sfqid)
+
         if ret.get(u'success'):
             data = ret.get(u'data')
             self.chunk_info(data, use_ijson=_use_ijson)

test/test_results.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2012-2017 Snowflake Computing Inc. All right reserved.
+#
+
+import pytest
+
+from snowflake.connector import ProgrammingError
+
+
+def test_results(conn_cnx):
+    """
+    Gets results for the given qid
+    """
+    with conn_cnx() as cnx:
+        cur = cnx.cursor()
+        cur.execute("select * from values(1,2),(3,4)")
+        sfqid = cur.sfqid
+        cur = cur.query_result(sfqid)
+        got_sfqid = cur.sfqid
+        assert cur.fetchall() == [(1, 2), (3, 4)]
+        assert sfqid == got_sfqid
+
+
+def test_results_with_error(conn_cnx):
+    """
+    Gets results with error
+    """
+    with conn_cnx() as cnx:
+        cur = cnx.cursor()
+        sfqid = None
+        try:
+            cur.execute("select blah")
+            pytest.fail("Should fail here!")
+        except ProgrammingError as e:
+            sfqid = e.sfqid
+
+        got_sfqid = None
+        try:
+            cur.query_result(sfqid)
+            pytest.fail("Should fail here again!")
+        except ProgrammingError as e:
+            got_sfqid = e.sfqid
+
+        assert got_sfqid is not None
+        assert got_sfqid == sfqid