SNOW-460809 Attempt to add INFO level logging message regarding first chunk rowcount (#1249)

* Attempt to add INFO level logging message regarding first chunk rowcount
* Added test for logging results in first chunk
* Moved log tests to test_insert_select, test_insert_and_select_by_separate_connection, and test_fetchmany

Author: sfc-gh-heshah

src/snowflake/connector/cursor.py

@@ -850,6 +850,11 @@ def _init_result_and_meta(self, data):
             self, self._query_result_format, data, self._description
         )
 
+        if not (is_dml or self.is_file_transfer):
+            logger.info(
+                "Number of results in first chunk: %s", result_chunks[0].rowcount
+            )
+
         self._result_set = ResultSet(
             self,
             result_chunks,

test/integ/test_cursor.py

@@ -131,7 +131,8 @@ def _check_results(cursor, results):
     assert results[2] == 123456, "the third result was wrong"
 
 
-def test_insert_select(conn, db_parameters):
+@pytest.mark.skipolddriver
+def test_insert_select(conn, db_parameters, caplog):
     """Inserts and selects integer data."""
     with conn() as cnx:
         c = cnx.cursor()
@@ -157,20 +158,25 @@ def test_insert_select(conn, db_parameters):
             for rec in c:
                 results.append(rec[0])
             _check_results(c, results)
+            assert "Number of results in first chunk: 3" in caplog.text
         finally:
             c.close()
 
         with cnx.cursor(snowflake.connector.DictCursor) as c:
+            caplog.clear()
+            assert "Number of results in first chunk: 3" not in caplog.text
             c.execute(
                 "select aa from {name} order by aa".format(name=db_parameters["name"])
             )
             results = []
             for rec in c:
                 results.append(rec["AA"])
             _check_results(c, results)
+            assert "Number of results in first chunk: 3" in caplog.text
 
 
-def test_insert_and_select_by_separate_connection(conn, db_parameters):
+@pytest.mark.skipolddriver
+def test_insert_and_select_by_separate_connection(conn, db_parameters, caplog):
     """Inserts a record and select it by a separate connection."""
     with conn() as cnx:
         result = cnx.cursor().execute(
@@ -204,6 +210,7 @@ def test_insert_and_select_by_separate_connection(conn, db_parameters):
         c.close()
         assert results[0] == 1234, "the first result was wrong"
         assert result.rowcount == 1, "wrong number of records were selected"
+        assert "Number of results in first chunk: 1" in caplog.text
     finally:
         cnx2.close()
 
@@ -752,7 +759,9 @@ def test_closed_cursor(conn, db_parameters):
         ), "SNOW-647539: rowcount should remain available after cursor is closed"
 
 
-def test_fetchmany(conn, db_parameters):
+@pytest.mark.skipolddriver
+def test_fetchmany(conn, db_parameters, caplog):
+
     table_name = random_string(5, "test_fetchmany_")
     with conn() as cnx:
         with cnx.cursor() as c:
@@ -773,6 +782,7 @@ def test_fetchmany(conn, db_parameters):
 
         with cnx.cursor() as c:
             c.execute(f"select aa from {table_name} order by aa desc")
+            assert "Number of results in first chunk: 6" in caplog.text
 
             rows = c.fetchmany(2)
             assert len(rows) == 2, "The number of records"