From 0d487d9481fd315abfbffd71bf6123a03e5efae4 Mon Sep 17 00:00:00 2001
From: Rahim Kanji <rahim.kanji@outlook.com>
Date: Mon, 18 Dec 2023 11:32:45 +0500
Subject: [PATCH] Added TAP test

---
 .../tests/reg_test_4402-mysql_fields-t.cpp    | 129 ++++++++++++++++++
 1 file changed, 129 insertions(+)
 create mode 100644 test/tap/tests/reg_test_4402-mysql_fields-t.cpp

diff --git a/test/tap/tests/reg_test_4402-mysql_fields-t.cpp b/test/tap/tests/reg_test_4402-mysql_fields-t.cpp
new file mode 100644
index 0000000000..bc9c098082
--- /dev/null
+++ b/test/tap/tests/reg_test_4402-mysql_fields-t.cpp
@@ -0,0 +1,129 @@
+/**
+ * @file reg_test_4402-mysql-fields-t.cpp
+ * @brief This TAP test checks if the length of the column alias and table alias surpasses 250 characters, 
+ *		should not impact MySQL field name length (MySQL_FIELD::name_length) and the MySQL field database length (MySQL_FIELD::db_length)
+ */
+
+#include <stdio.h>
+#include <unistd.h>
+#include <string>
+#include <thread>
+#include "tap.h"
+#include "command_line.h"
+#include "utils.h" 
+
+#define MYSQL_QUERY__(mysql, query) \
+	do { \
+		if (mysql_query(mysql, query)) { \
+			fprintf(stderr, "File %s, line %d, Error: %s\n", \
+					__FILE__, __LINE__, mysql_error(mysql)); \
+			goto cleanup; \
+		} \
+	} while(0)
+
+CommandLine cl;
+
+std::string generate_random_string(size_t length) {
+	std::srand(static_cast<unsigned int>(std::time(nullptr)));
+	static const char characters[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
+	static const int numCharacters = (sizeof(characters) - 1)/ sizeof(char);
+
+	std::string randomString;
+	randomString.reserve(length);
+
+	for (size_t i = 0; i < length; ++i) {
+		char randomChar = characters[std::rand() % numCharacters];
+		randomString.push_back(randomChar);
+	}
+
+	return randomString;
+}
+
+int main(int argc, char** argv) {
+
+	if (cl.getEnv()) {
+		diag("Failed to get the required environmental variables.");
+		return -1;
+	}
+
+	plan(256*2);
+
+	// Initialize ProxySQL connection
+	MYSQL* proxysql = mysql_init(NULL);
+	if (!proxysql) {
+		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxysql));
+		return -1;
+	}
+
+	// Connect to ProxySQL
+	if (!mysql_real_connect(proxysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
+		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxysql));
+		return exit_status();
+	}
+
+	MYSQL_QUERY__(proxysql, "DROP DATABASE IF EXISTS testdb");
+	MYSQL_QUERY__(proxysql, "CREATE DATABASE testdb");
+
+	diag("Creating echo_int function...");
+	MYSQL_QUERY__(proxysql, "CREATE FUNCTION IF NOT EXISTS testdb.echo_int(N INT) RETURNS INT DETERMINISTIC RETURN N;");
+	
+	diag("Creating dummy_table...");
+	MYSQL_QUERY__(proxysql, "CREATE TABLE IF NOT EXISTS testdb.dummy_table(data VARCHAR(10))");
+
+	// wait for replication
+	std::this_thread::sleep_for(std::chrono::seconds(2));
+	
+	// alias maximum length is 256.  
+	// https://dev.mysql.com/doc/refman/8.2/en/identifier-length.html
+	// https://mariadb.com/kb/en/identifier-names/#maximum-length
+	for (unsigned int length = 1; length <= 256; length++) {
+
+		// to check column alias issue:
+		{
+			const std::string& query = "SELECT testdb.echo_int(1) AS " + generate_random_string(length);
+			MYSQL_QUERY__(proxysql, query.c_str());
+
+			MYSQL_RES* res = mysql_use_result(proxysql);
+			if (!res) {
+				fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxysql));
+				return exit_status();
+			}
+
+			MYSQL_FIELD* field = mysql_fetch_fields(res);
+
+			ok(field->name_length == length, "name_length: '%u'. Expected length: '%u'", field->name_length, length);
+
+			if (res) {
+				mysql_free_result(res);
+				res = NULL;
+			}
+		}
+
+		// to check table alias issue:
+		{
+			const std::string& query = "SELECT data FROM testdb.dummy_table AS " + generate_random_string(length);
+			MYSQL_QUERY__(proxysql, query.c_str());
+
+			MYSQL_RES* res = mysql_use_result(proxysql);
+			if (!res) {
+				fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxysql));
+				return exit_status();
+			}
+
+			MYSQL_FIELD* field = mysql_fetch_fields(res);
+
+			ok(field->db_length == (sizeof("testdb")-1), "db_length: '%u'. Expected length: '%u'", 
+				field->db_length, (unsigned int)(sizeof("testdb")-1));
+
+			if (res) {
+				mysql_free_result(res);
+				res = NULL;
+			}
+		}
+	}
+cleanup:
+	mysql_query(proxysql, "DROP DATABASE IF EXISTS testdb");
+	mysql_close(proxysql);
+
+	return exit_status();
+}