Enable vector test to run only in AzureDB (#2749)

* Enable vector test in AzureDB

* jdk8 failure : for testing

* 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;
@@ -609,6 +610,7 @@ public void testBulkCopyWithComputedColumnNotAsLastColumn() throws SQLException
      * BulkCopyCSVTestInputWithVector.csv file
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSV() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTableBulkCopyVectorCsv"));
@@ -686,6 +688,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("dstTableVectorCsvMulti"));
@@ -753,6 +756,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("dstTableBulkCopyVectorCsvMultiPipe"));
@@ -820,6 +824,7 @@ public void testBulkCopyVectorFromCSVWithMultipleColumnsWithPipeDelimiter() thro
      * metadata.
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorFromCSVWithIncorrectDimension() throws SQLException {
         String dstTable = AbstractSQLGenerator.escapeIdentifier(RandomUtil.getIdentifier("dstTableBulkCopyVectorCsvBadDim"));

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
@@ -706,6 +710,7 @@ public void testBulkCopyVectorNull() throws SQLException {
      * incompatible with vector".
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopySourceAsVarBinary() {
         String varbinaryTable = TestUtils
@@ -767,6 +772,7 @@ public void testBulkCopyVectorUsingBulkCopySourceAsVarBinary() {
      * incompatible with varbinary(max)".
      */
     @Test
+    @AzureDB
     @Tag(Constants.vectorTest)
     public void testBulkCopyVectorUsingBulkCopyDestinationAsVarBinary() {
         String vectorTable = TestUtils.escapeSingleQuotes(AbstractSQLGenerator.escapeIdentifier("testVectorTable"));
@@ -834,6 +840,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"));
@@ -901,6 +908,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"));
@@ -971,6 +979,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"));
@@ -1029,6 +1038,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

@@ -39,13 +39,15 @@
 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;
 import microsoft.sql.Vector.VectorDimensionType;
 
 @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 {}