Add more tests and test with MySql.Data.

MySql.Data 8.4.0 already supports the VECTOR data type.

Signed-off-by: Bradley Grainger <[email protected]>

Author: bgrainger

tests/IntegrationTests/DataTypes.cs

@@ -1,4 +1,6 @@
 using System.Globalization;
+using System.Runtime.InteropServices;
+
 #if MYSQL_DATA
 using MySql.Data.Types;
 #endif
@@ -1143,7 +1145,9 @@ private static object CreateGeometry(byte[] data)
 	[InlineData("Int64", "datatypes_integers", MySqlDbType.Int64, 20, typeof(long), "N", 0, 0)]
 	[InlineData("UInt64", "datatypes_integers", MySqlDbType.UInt64, 20, typeof(ulong), "N", 0, 0)]
 	[InlineData("value", "datatypes_json_core", MySqlDbType.JSON, int.MaxValue, typeof(string), "LN", 0, 0)]
-#if !MYSQL_DATA
+#if MYSQL_DATA
+	[InlineData("value", "datatypes_vector", MySqlDbType.Vector, 12, typeof(byte[]), "N", 0, 31)]
+#else
 	[InlineData("value", "datatypes_vector", MySqlDbType.Vector, 3, typeof(float[]), "N", 0, 31)]
 #endif
 	[InlineData("Single", "datatypes_reals", MySqlDbType.Float, 12, typeof(float), "N", 0, 31)]
@@ -1604,17 +1608,28 @@ public void QueryJson(string column, string[] expected)
 		DoQuery("json_core", column, dataTypeName, expected, reader => reader.GetString(0), omitWhereTest: true);
 	}
 
-#if !MYSQL_DATA
 	[SkippableTheory(ServerFeatures.Vector)]
 	[InlineData("value", new[] { null, "0,0,0", "1,1,1", "1,2,3", "3.40282347E+38,3.40282347E+38,3.40282347E+38" })]
 	public void QueryVector(string column, string[] expected)
 	{
 		string dataTypeName = "VECTOR";
 		DoQuery("vector", column, dataTypeName,
-			expected.Select(x => x?.Split(',').Select(x => float.Parse(x, CultureInfo.InvariantCulture)).ToArray()).ToArray(),
-			static x => (float[]) x.GetValue(0), omitWhereTest: true);
-	}
+			expected.Select(x =>
+#if MYSQL_DATA
+				// Connector/NET returns the float array as a byte[]
+				x is null ? null : MemoryMarshal.AsBytes<float>(x.Split(',').Select(x => float.Parse(x, CultureInfo.InvariantCulture)).ToArray()).ToArray())
+#else
+				x?.Split(',').Select(x => float.Parse(x, CultureInfo.InvariantCulture)).ToArray())
+#endif
+			.ToArray(),
+#if !MYSQL_DATA
+			static x => (float[]) x.GetValue(0),
+#else
+			// NOTE: Connector/NET returns 'null' for NULL so simulate an exception for the tests
+			x => x.IsDBNull(0) ? throw new GetValueWhenNullException() : x.GetValue(0),
 #endif
+			omitWhereTest: true);
+	}
 
 	[SkippableTheory(MySqlData = "https://bugs.mysql.com/bug.php?id=97067")]
 	[InlineData(false, "MIN", 0)]

tests/IntegrationTests/QueryTests.cs

@@ -1,3 +1,5 @@
+using System.Runtime.InteropServices;
+
 namespace IntegrationTests;
 
 public class QueryTests : IClassFixture<DatabaseFixture>, IDisposable
@@ -1688,6 +1690,56 @@ public void GetBytesByName()
 	}
 #endif
 
+	[SkippableTheory(ServerFeatures.Vector)]
+	[InlineData(false)]
+	[InlineData(true)]
+	public void QueryVector(bool prepare)
+	{
+		using var connection = new MySqlConnection(AppConfig.ConnectionString);
+		connection.Open();
+
+		connection.Execute("""
+			drop table if exists test_vector;
+			create table test_vector(id int auto_increment not null primary key, vec vector not null);
+			""");
+
+		using var cmd = m_database.Connection.CreateCommand();
+		cmd.CommandText = "INSERT INTO test_vector(vec) VALUES(@vec)";
+		cmd.Parameters.Add(new MySqlParameter
+		{
+			ParameterName = "@vec",
+			MySqlDbType = MySqlDbType.Vector,
+		});
+
+		var floatArray = new[] { 1.2f, 3.4f, 5.6f };
+#if MYSQL_DATA
+		// Connector/NET requires the float vector to be passed as a byte array
+		cmd.Parameters[0].Value = MemoryMarshal.AsBytes<float>(floatArray).ToArray();
+#else
+		cmd.Parameters[0].Value = floatArray;
+#endif
+
+		if (prepare)
+			cmd.Prepare();
+		cmd.ExecuteNonQuery();
+
+		// Select and verify the value
+		cmd.CommandText = "SELECT vec FROM test_vector";
+		if (prepare)
+			cmd.Prepare();
+
+		using var reader = cmd.ExecuteReader();
+		Assert.True(reader.Read());
+		var value = reader.GetValue(0);
+
+#if MYSQL_DATA
+		var result = MemoryMarshal.Cast<byte, float>((byte[]) value).ToArray();
+#else
+		var result = (float[]) value;
+#endif
+		Assert.Equal(floatArray, result);
+	}
+
 	private class BoolTest
 	{
 		public int Id { get; set; }

tests/IntegrationTests/StoredProcedureTests.cs

@@ -761,6 +761,44 @@ public void PassJsonParameter()
 		Assert.False(reader.Read());
 	}
 
+#if false
+	[SkippableTheory(ServerFeatures.Vector)]
+	[InlineData(false)]
+	[InlineData(true)]
+	public void VectorOutputParameter(bool prepare)
+	{
+		using var cmd = m_database.Connection.CreateCommand();
+		cmd.CommandText = """
+			DROP PROCEDURE IF EXISTS sp_vector_out;
+			CREATE PROCEDURE sp_vector_out (OUT vec VECTOR)
+			BEGIN
+				SELECT STRING_TO_VECTOR('[1.2, 3.4, 5.6]') INTO vec;
+			END;
+			""";
+		cmd.ExecuteNonQuery();
+
+		cmd.CommandText = "sp_vector_out";
+		cmd.CommandType = CommandType.StoredProcedure;
+		cmd.Parameters.Add(new MySqlParameter
+		{
+			ParameterName = "@vec",
+			MySqlDbType = MySqlDbType.Vector,
+			Direction = ParameterDirection.Output,
+		});
+
+		if (prepare)
+			cmd.Prepare();
+		cmd.ExecuteNonQuery();
+
+		var value = cmd.Parameters[0].Value;
+		Assert.IsType<float[]>(value);
+
+		var result = (float[]) value;
+
+		Assert.Equal(new float[] { 1.2f, 3.4f, 5.6f }, result);
+	}
+#endif
+
 	private static Action<MySqlParameter> AssertParameter(string name, ParameterDirection direction, MySqlDbType mySqlDbType)
 	{
 		return x =>