From 0dd2db1b02ffc1845b99100b9d5636db8a2f98e9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Javier=20Jaramago=20Fern=C3=A1ndez?=
 <jaramago.fernandez.javier@gmail.com>
Date: Tue, 2 Mar 2021 13:01:39 +0000
Subject: [PATCH] Added new iterations to 'test_ps_async-t' fetching just one
 and two rows

---
 test/tap/tests/test_ps_async-t.cpp | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/test/tap/tests/test_ps_async-t.cpp b/test/tap/tests/test_ps_async-t.cpp
index d00a2bd07..649c58bd2 100644
--- a/test/tap/tests/test_ps_async-t.cpp
+++ b/test/tap/tests/test_ps_async-t.cpp
@@ -160,8 +160,8 @@ static int wait_for_mysql(MYSQL *mysql, int status) {
 
 
 const int NUM_ROWS=100;
-const int NUM_ROWS_READ=1000;
-const int NLOOPS=1;
+const std::vector<int> NUM_ROWS_READ { 1000, 1, 2 };
+const int NLOOPS = NUM_ROWS_READ.size();
 
 int select_config_file(MYSQL* mysql, std::string& resultset) {
 	if (mysql_query(mysql, "select config file")) {
@@ -201,7 +201,7 @@ int main(int argc, char** argv) {
 	if(cl.getEnv())
 		return exit_status();
 
-	plan(2*NLOOPS);
+	plan(1 + NLOOPS);
 	diag("Testing PS async store result");
 
 	MYSQL* mysqladmin = mysql_init(NULL);
@@ -287,6 +287,9 @@ int main(int argc, char** argv) {
 	std::string query = "";
 
 	for (int loops=0; loops<NLOOPS; loops++) {
+
+	int IT_NUM_ROWS_READ = NUM_ROWS_READ[loops];
+
 	MYSQL_STMT *stmt2a = mysql_stmt_init(mysql);
 	if (!stmt2a)
 	{
@@ -294,9 +297,9 @@ int main(int argc, char** argv) {
 		return restore_admin(mysqladmin);
 	}
 	// NOTE: the first 2 columns we select are 3 ids, so we can later print and verify
-	query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3, t1.k k1, t1.c c1, t1.pad pad1, t2.k k2, t2.c c2, t2.pad pad2 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 ORDER BY t1.id, t2.id LIMIT " + std::to_string(NUM_ROWS_READ);
-	//query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 LIMIT " + std::to_string(NUM_ROWS_READ);
-	//query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 ORDER BY t1.id, t2.id LIMIT " + std::to_string(NUM_ROWS_READ);
+	query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3, t1.k k1, t1.c c1, t1.pad pad1, t2.k k2, t2.c c2, t2.pad pad2 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 ORDER BY t1.id, t2.id LIMIT " + std::to_string(IT_NUM_ROWS_READ);
+	//query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 LIMIT " + std::to_string(IT_NUM_ROWS_READ);
+	//query = "SELECT t1.id id1, t2.id id2, t1.id+t2.id id3 FROM test.sbtest1 t1 JOIN test.sbtest1 t2 ORDER BY t1.id, t2.id LIMIT " + std::to_string(IT_NUM_ROWS_READ);
 	if (mysql_stmt_prepare(stmt2a,query.c_str(), query.size())) {
 		fprintf(stderr, "Query error %s\n", mysql_error(mysql));
 		mysql_close(mysql);
@@ -405,7 +408,7 @@ int main(int argc, char** argv) {
 				row_count2a++;		
 			}
 		}
-		ok(row_count2a+rows_read==NUM_ROWS_READ, "Fetched %d rows, expected %d. Details: %d rows processed while buffering, %d at the end", row_count2a+rows_read, NUM_ROWS_READ , rows_read, row_count2a);
+		ok(row_count2a+rows_read==IT_NUM_ROWS_READ, "Fetched %d rows, expected %d. Details: %d rows processed while buffering, %d at the end", row_count2a+rows_read, IT_NUM_ROWS_READ , rows_read, row_count2a);
 	}
 
 	if (prepare_meta_result) {