[Text Analytics] Fix flaky tests (Azure#16784)

* fix flaky tests

* update pagination test

* cleanup

Co-authored-by: Suhas Mehta <suhas.mehta@microsoft.com>

Author: suhas92

sdk/textanalytics/Azure.AI.TextAnalytics/src/TextAnalyticsClient.cs

@@ -2084,7 +2084,6 @@ private Response<RecognizeLinkedEntitiesResultCollection> RecognizeLinkedEntitie
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="document">The document to analyze.</param>
         /// <param name="language">The language that the document is written in.</param>
@@ -2130,7 +2129,6 @@ public virtual async Task<HealthcareOperation> StartHealthcareAsync(string docum
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="document">The document to analyze.</param>
         /// <param name="language">The language that the document is written in.</param>
@@ -2175,7 +2173,6 @@ public virtual HealthcareOperation StartHealthcare(string document, string langu
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="documents">The documents to analyze.</param>
         /// <param name="language">The language that the document is written in.</param>
@@ -2201,7 +2198,6 @@ public virtual async Task<HealthcareOperation> StartHealthcareBatchAsync(IEnumer
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="documents">The documents to analyze.</param>
         /// <param name="language">The language that the document is written in.
@@ -2234,7 +2230,6 @@ public virtual HealthcareOperation StartHealthcareBatch(IEnumerable<string> docu
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="documents">The documents to analyze.</param>
         /// <param name="options">The additional configurable options<see cref="HealthcareOptions"/></param>
@@ -2260,7 +2255,6 @@ public virtual HealthcareOperation StartHealthcareBatch(IEnumerable<TextDocument
         /// <a href="https://docs.microsoft.com/en-us/azure/cognitive-services/text-analytics/language-support"/>.
         /// For document length limits, maximum batch size, and supported text encoding, see
         /// <a href="https://docs.microsoft.com/azure/cognitive-services/text-analytics/overview#data-limits"/>.
-        /// Use <see cref="GetHealthcareEntities(HealthcareOperation, CancellationToken)"/> for documents greater than 20.
         /// </summary>
         /// <param name="documents">The documents to analyze.</param>
         /// <param name="options">The additional configurable options<see cref="HealthcareOptions"/></param>
@@ -2375,8 +2369,6 @@ public virtual string StartCancelHealthJob(HealthcareOperation operation, Cancel
 
         /// <summary>
         /// Gets collection of healthcare entities from the HealthOperation using async pageable.
-        /// This method is used if the number of documents are greater than 20
-        /// Otherwise, use the StartHealthcareAsync and StartHealthcareBatchAsync for less than 20 documents.
         /// </summary>
         /// <param name="operation"> Healthcare operation class object which is returned when operation is started. <see cref="HealthcareOperation"/></param>
         /// <param name="cancellationToken">A <see cref="CancellationToken"/> controlling the request lifetime.</param>

sdk/textanalytics/Azure.AI.TextAnalytics/tests/RecognizeHealthcareEntitiesTests.cs

@@ -270,12 +270,17 @@ public async Task RecognizeHealthcareEntitiesBatchWithPagination()
 
             var list = new List<string>();
 
-            for (int i = 0; i < 23; i++)
+            for (int i = 0; i < 10; i++)
             {
                 list.Add(document);
             };
 
-            HealthcareOperation healthOperation = await client.StartHealthcareBatchAsync(list);
+            HealthcareOptions options = new HealthcareOptions()
+            {
+                Top = 2
+            };
+
+            HealthcareOperation healthOperation = await client.StartHealthcareBatchAsync(list, "en", options);
 
             AsyncPageable<DocumentHealthcareResult> results = client.GetHealthcareEntities(healthOperation);
 
@@ -285,7 +290,7 @@ public async Task RecognizeHealthcareEntitiesBatchWithPagination()
                 resultCount += 1;
             }
 
-            Assert.AreEqual(23, resultCount);
+            Assert.AreEqual(10, resultCount);
         }
     }
 }