Backport : Enable vector test to run only in AzureDB (#2762)

* Enable vector test in AzureDB

* jdk8 failure : for testing

* Added VectorTag interface

* Added custom annotation to run tests on AzureDB

* Added vectorTest tag along with AzureDB tag

Author: muskan124947

src/test/java/com/microsoft/sqlserver/jdbc/bulkCopy/BulkCopyCSVTest.java

@@ -44,6 +44,7 @@
 import com.microsoft.sqlserver.jdbc.TestUtils;
 import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
 import com.microsoft.sqlserver.testframework.AbstractTest;
+import com.microsoft.sqlserver.testframework.AzureDB;
 import com.microsoft.sqlserver.testframework.Constants;
 import com.microsoft.sqlserver.testframework.DBConnection;
 import com.microsoft.sqlserver.testframework.DBResultSet;
@@ -620,6 +621,7 @@ public void testBulkCopyWithComputedColumnNotAsLastColumn() {
      * BulkCopyCSVTestInputWithVector.csv file
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSV() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTable"));
@@ -697,6 +699,7 @@ public void testBulkCopyVectorFromCSV() throws SQLException {
      * BulkCopyCSVTestInputWithMultipleVectorColumn.csv file
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSVWithMultipleColumns() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTable"));
@@ -764,6 +767,7 @@ public void testBulkCopyVectorFromCSVWithMultipleColumns() throws SQLException {
      * BulkCopyCSVTestWithMultipleVectorColumnWithPipeDelimiter.csv file
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSVWithMultipleColumnsWithPipeDelimiter() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTable"));
@@ -831,6 +835,7 @@ public void testBulkCopyVectorFromCSVWithMultipleColumnsWithPipeDelimiter() thro
      * metadata.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSVWithIncorrectDimension() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTable"));

src/test/java/com/microsoft/sqlserver/jdbc/bulkCopy/BulkCopyISQLServerBulkRecordTest.java

@@ -42,6 +42,7 @@
 import com.microsoft.sqlserver.jdbc.TestUtils;
 import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
 import com.microsoft.sqlserver.testframework.AbstractTest;
+import com.microsoft.sqlserver.testframework.AzureDB;
 import com.microsoft.sqlserver.testframework.Constants;
 import com.microsoft.sqlserver.testframework.DBConnection;
 import com.microsoft.sqlserver.testframework.DBStatement;
@@ -161,6 +162,7 @@ public void testBulkCopyDateTimePrecision() throws SQLException {
      * Test bulk copy with a single Vector row.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVector() throws SQLException {
         String dstTable = TestUtils
@@ -586,6 +588,7 @@ public void testBulkCopyWithCountVerification() throws SQLException {
      * to destination table with VECTOR column.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyTableToTableJsonToVector() throws Exception {
         String srcTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testSrcJsonTable"));
@@ -661,6 +664,7 @@ private static Object[] parseJsonArrayToFloatArray(String json) {
      * Test bulk copy with null Vector data.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorNull() throws SQLException {
         String dstTable = TestUtils
@@ -708,6 +712,7 @@ public void testBulkCopyVectorNull() throws SQLException {
      * incompatible with vector".
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopySourceAsVarBinary() {
         String varbinaryTable = TestUtils
@@ -769,6 +774,7 @@ public void testBulkCopyVectorUsingBulkCopySourceAsVarBinary() {
      * incompatible with varbinary(max)".
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopyDestinationAsVarBinary() {
         String vectorTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testVectorTable"));
@@ -836,6 +842,7 @@ public void testBulkCopyVectorUsingBulkCopyDestinationAsVarBinary() {
      * The operation should succeed, and the data should be validated.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopySourceAsVarchar() {
         String varcharTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testVarcharTable"));
@@ -903,6 +910,7 @@ public void testBulkCopyVectorUsingBulkCopySourceAsVarchar() {
      * The operation should succeed, and the data should be validated.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopyDestinationAsVarchar() {
         String vectorTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testVectorTable"));
@@ -973,6 +981,7 @@ public void testBulkCopyVectorUsingBulkCopyDestinationAsVarchar() {
      * The operation should fail with an error: "The vector dimensions 3 and 4 do not match."
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorWithMismatchedDimensions() {
         String srcTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testSrcTable"));
@@ -1031,6 +1040,7 @@ public void testBulkCopyVectorWithMismatchedDimensions() {
      * Test bulk copy with a large number of records to check performance.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyPerformance() throws SQLException {
         String tableName = AbstractSQLGenerator.escapeIdentifier("srcTable");

src/test/java/com/microsoft/sqlserver/jdbc/databasemetadata/DatabaseMetaDataTest.java

@@ -56,6 +56,7 @@
 import com.microsoft.sqlserver.jdbc.TestUtils;
 import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
 import com.microsoft.sqlserver.testframework.AbstractTest;
+import com.microsoft.sqlserver.testframework.AzureDB;
 import com.microsoft.sqlserver.testframework.Constants;
 
 
@@ -1091,6 +1092,7 @@ public void testGetSchemasWithAndWithoutCatalog() throws SQLException {
      * @throws SQLException
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testVectorMetaData() throws SQLException {
         String vectorTableName = RandomUtil.getIdentifier("vectorTable");

src/test/java/com/microsoft/sqlserver/jdbc/datatypes/VectorTest.java

@@ -30,6 +30,7 @@
 import com.microsoft.sqlserver.jdbc.TestUtils;
 import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
 import com.microsoft.sqlserver.testframework.AbstractTest;
+import com.microsoft.sqlserver.testframework.AzureDB;
 import com.microsoft.sqlserver.testframework.Constants;
 
 import microsoft.sql.Vector;
@@ -45,6 +46,7 @@
 
 @RunWith(JUnitPlatform.class)
 @DisplayName("Test Vector Data Type")
+@AzureDB
 @Tag(Constants.vectorTest)
 public class VectorTest extends AbstractTest {
 

src/test/java/com/microsoft/sqlserver/jdbc/preparedStatement/BatchExecutionWithBulkCopyTest.java

@@ -54,6 +54,7 @@
 import com.microsoft.sqlserver.jdbc.TestUtils;
 import com.microsoft.sqlserver.testframework.AbstractSQLGenerator;
 import com.microsoft.sqlserver.testframework.AbstractTest;
+import com.microsoft.sqlserver.testframework.AzureDB;
 import com.microsoft.sqlserver.testframework.Constants;
 import com.microsoft.sqlserver.testframework.PrepUtil;
 
@@ -1266,6 +1267,7 @@ public void testBulkInsertWithNullDataForAllTemporalTypesAndMoney() throws Excep
      * Test inserting vector data using prepared statement with bulk copy enabled.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testInsertVectorWithBulkCopy() throws Exception {
         String tableName = RandomUtil.getIdentifier("BulkCopyVectorTest");
@@ -1304,6 +1306,7 @@ public void testInsertVectorWithBulkCopy() throws Exception {
      * Test inserting null vector data using prepared statement with bulk copy enabled.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testInsertNullVectorWithBulkCopy() throws Exception {
         String tableName = RandomUtil.getIdentifier("BulkCopyVectorTest");
@@ -1343,6 +1346,7 @@ public void testInsertNullVectorWithBulkCopy() throws Exception {
      * Test inserting vector data using prepared statement with bulk copy enabled for performance.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testInsertWithBulkCopyPerformance() throws SQLException {
         String tableName = AbstractSQLGenerator.escapeIdentifier("BulkCopyVectorPerformanceTest");

src/test/java/com/microsoft/sqlserver/testframework/AzureDB.java

@@ -0,0 +1,27 @@
+/*
+ * Microsoft JDBC Driver for SQL Server Copyright(c) Microsoft Corporation All rights reserved. This program is made
+ * available under the terms of the MIT License. See the LICENSE file in the project root for more information.
+ */
+
+package com.microsoft.sqlserver.testframework;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+import org.junit.jupiter.api.Tag;
+
+/*
+ * Custom annotation to run tests only on Azure DBs
+ */
+@Target({ElementType.METHOD, ElementType.TYPE})
+@Retention(RetentionPolicy.RUNTIME)
+@Tag(Constants.xSQLv11)
+@Tag(Constants.xSQLv12)
+@Tag(Constants.xSQLv14)
+@Tag(Constants.xSQLv15)
+@Tag(Constants.xSQLv16)
+@Tag(Constants.xAzureSQLDW)
+@Tag(Constants.xAzureSQLMI)
+public @interface AzureDB {}