SqlBulkTruncation (#331)

* This commit adds more tests for AE. Such as TestExecuteXmlReader, TestBeginAndEndExecuteReaderWithAsyncCallback, TestSqlCommandCancel, TestSqlCommandCancellationToken

* code clean up

* Resolving PR comments.

* Fixing issues on AE tests.

* fixing open SqlDataReader

* Fixing some Failings on the pipeline tests.

* fix TargetFrameWork issue

* SqlBulkTruncation

* Taking XMLReader test out.

* fixing hard coded table name issue.

* BulkCopy Truncation

* Fixing Primary key issue

* Update SqlBulkCopyTruncation.cs

* more changes for SqlBulkCopyTruncation and AEV2

* Adding

* bulkCopy

* SwlBulkCopyTest

* TestBulkCopy

* Fix SqlBulkcopyTruncation

* SqlBulkTruncation

* Final SqlBulkCopyTruncations

* SqlBulkCopy

* sqlBulk

* sqlBulkCopy

* sqlBulkCopy

Author: JRahnama

src/Microsoft.Data.SqlClient/tests/ManualTests/AlwaysEncrypted/ApiShould.cs

@@ -1809,10 +1809,8 @@ public void TestSqlCommandSequentialAccessCodePaths(string connection, CommandBe
                     }
                 }
             }
-
         }
 
-        [SkipOnTargetFramework(TargetFrameworkMonikers.Netcoreapp)]
         [ConditionalTheory(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringSetupForAE))]
         [ClassData(typeof(AEConnectionStringProvider))]
         public void TestExecuteXmlReader(string connection)
@@ -1822,20 +1820,20 @@ public void TestExecuteXmlReader(string connection)
             IList<object> values = GetValues(dataHint: 60);
             int numberOfRows = 10;
 
-            // Insert a bunch of rows in to the table.
+            // Insert a bunch of rows in to the table.	
             int rowsAffected = InsertRows(tableName: tableName, numberofRows: numberOfRows, values: values, connection: connection);
             Assert.True(rowsAffected == numberOfRows, "number of rows affected is unexpected.");
 
             using (SqlConnection sqlConnection = new SqlConnection(connection))
             {
                 sqlConnection.Open();
 
-                // select the set of rows that were inserted just now.
+                // select the set of rows that were inserted just now.	
                 using (SqlCommand sqlCommand = new SqlCommand($"SELECT LastName FROM [{tableName}] WHERE FirstName = @FirstName AND CustomerId = @CustomerId FOR XML AUTO;", sqlConnection, transaction: null, columnEncryptionSetting: SqlCommandColumnEncryptionSetting.Enabled))
                 {
                     if (DataTestUtility.EnclaveEnabled)
                     {
-                        //Increase Time out for enclave-enabled server.
+                        //Increase Time out for enclave-enabled server.	
                         sqlCommand.CommandTimeout = 90;
                     }
                     sqlCommand.Parameters.Add(@"CustomerId", SqlDbType.Int);
@@ -1849,11 +1847,6 @@ public void TestExecuteXmlReader(string connection)
 
                     var ex = Assert.Throws<SqlException>(() => sqlCommand.ExecuteXmlReader());
                     Assert.Equal($"'FOR XML' clause is unsupported for encrypted columns.{Environment.NewLine}Statement(s) could not be prepared.", ex.Message);
-
-                    //string xmlResult;
-                    IAsyncResult asyncResult = sqlCommand.BeginExecuteXmlReader();
-
-                    Assert.Throws<SqlException>(() => sqlCommand.EndExecuteXmlReader(asyncResult));
                 }
             }
         }

src/Microsoft.Data.SqlClient/tests/ManualTests/AlwaysEncrypted/ExceptionsGenericError.cs

@@ -4,13 +4,14 @@
 
 namespace Microsoft.Data.SqlClient.ManualTesting.Tests.AlwaysEncrypted
 {
-    public class ExceptionsGenericErrors : IClassFixture<ExceptionGenericErrorFixture> {
-
+    public class ExceptionsGenericErrors : IClassFixture<ExceptionGenericErrorFixture>
+    {
         [ConditionalTheory(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringSetupForAE), nameof(DataTestUtility.IsNotAzureServer), Skip = "ActiveIssue 10036")]
         [ClassData(typeof(AEConnectionStringProvider))]
-        public void TestCommandOptionWithNoTceFeature (string connectionString) {
+        public void TestCommandOptionWithNoTceFeature(string connectionString)
+        {
             SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder(connectionString);
-            CertificateUtility.ChangeServerTceSetting (false, sb); // disable TCE on engine.
+            CertificateUtility.ChangeServerTceSetting(false, sb); // disable TCE on engine.
             using (SqlConnection conn = CertificateUtility.GetOpenConnection(false, sb, fSuppressAttestation: true))
             {
                 using (SqlCommand cmd = new SqlCommand(ExceptionGenericErrorFixture.encryptedProcedureName, conn, null, SqlCommandColumnEncryptionSetting.Enabled))
@@ -23,12 +24,13 @@ public void TestCommandOptionWithNoTceFeature (string connectionString) {
                 }
             }
             // Turn on TCE now
-            CertificateUtility.ChangeServerTceSetting (true, sb); // enable tce
+            CertificateUtility.ChangeServerTceSetting(true, sb); // enable tce
         }
 
         [ConditionalTheory(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringSetupForAE), nameof(DataTestUtility.IsNotAzureServer))]
         [ClassData(typeof(AEConnectionStringProvider))]
-        public void TestDataAdapterAndEncrytionSetting (string connectionString) {
+        public void TestDataAdapterAndEncrytionSetting(string connectionString)
+        {
             SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder(connectionString);
             // Create a new SqlCommand for select and delete
             using (SqlConnection conn = CertificateUtility.GetOpenConnection(false, sb))
@@ -56,7 +58,7 @@ public void TestDataAdapterAndEncrytionSetting (string connectionString) {
                     rowInserted["c1"] = 5;
                     table.Rows.Add(rowInserted);
                     adapter.UpdateBatchSize = 0; // remove batch size limit
-                                                    // run batch update
+                                                 // run batch update
 
                     string expectedErrorMessage = "SqlCommandColumnEncryptionSetting should be identical on all commands (SelectCommand, InsertCommand, UpdateCommand, DeleteCommand) when doing batch updates.";
                     InvalidOperationException e = Assert.Throws<InvalidOperationException>(() => adapter.Update(dataset));
@@ -67,7 +69,8 @@ public void TestDataAdapterAndEncrytionSetting (string connectionString) {
 
         [ConditionalTheory(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringSetupForAE), nameof(DataTestUtility.IsNotAzureServer))]
         [ClassData(typeof(AEConnectionStringProvider))]
-        public void TestInvalidForceColumnEncryptionSetting(string connectionString) {
+        public void TestInvalidForceColumnEncryptionSetting(string connectionString)
+        {
             SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder(connectionString);
             using (SqlConnection conn = CertificateUtility.GetOpenConnection(false, sb))
             {
@@ -85,7 +88,8 @@ public void TestInvalidForceColumnEncryptionSetting(string connectionString) {
 
         [ConditionalTheory(typeof(DataTestUtility), nameof(DataTestUtility.AreConnStringSetupForAE), nameof(DataTestUtility.IsNotAzureServer))]
         [ClassData(typeof(AEConnectionStringProvider))]
-        public void TestParamUnexpectedEncryptionMD(string connectionString) {
+        public void TestParamUnexpectedEncryptionMD(string connectionString)
+        {
             SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder(connectionString);
             using (SqlConnection conn = CertificateUtility.GetOpenConnection(true, sb))
             {