split import in 2

microsoft · Jan 25, 2025 · 81bf463 · 81bf463
1 parent d419578
commit 81bf463
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/test/Microsoft.Health.Fhir.Shared.Tests.E2E/Rest/Import/ImportTests.cs b/test/Microsoft.Health.Fhir.Shared.Tests.E2E/Rest/Import/ImportTests.cs
@@ -217,7 +217,7 @@ EXECUTE dbo.MergeResourcesCommitTransaction @TransactionId
         public async Task GivenIncrementalLoad_80KSurrogateIds_BadRequestIsReturned()
         {
             var ndJson = new StringBuilder();
-            for (int i = 0;  i < 80001; i++)
+            for (int i = 0;  i < 79001; i++)
             {
                 var id = Guid.NewGuid().ToString("N");
                 var str = CreateTestPatient(id, DateTimeOffset.Parse("1900-01-01Z00:00")); // make sure this date is not used by other tests.
@@ -227,7 +227,21 @@ public async Task GivenIncrementalLoad_80KSurrogateIds_BadRequestIsReturned()
             var location = (await ImportTestHelper.UploadFileAsync(ndJson.ToString(), _fixture.StorageAccount)).location;
             var request = CreateImportRequest(location, ImportMode.IncrementalLoad);
             var checkLocation = await ImportTestHelper.CreateImportTaskAsync(_client, request);
-            var message = await ImportWaitAsync(checkLocation, false);
+            await ImportWaitAsync(checkLocation, true);
+
+            ndJson = new StringBuilder();
+            for (int i = 0; i < 1000; i++) // load violators in nlarge batch to avoid large number of retries
+            {
+                var id = Guid.NewGuid().ToString("N");
+                var str = CreateTestPatient(id, DateTimeOffset.Parse("1900-01-01Z00:00")); // make sure this date is not used by other tests.
+                ndJson.Append(str);
+            }
+
+            location = (await ImportTestHelper.UploadFileAsync(ndJson.ToString(), _fixture.StorageAccount)).location;
+            request = CreateImportRequest(location, ImportMode.IncrementalLoad);
+            checkLocation = await ImportTestHelper.CreateImportTaskAsync(_client, request);
+            var message = await ImportWaitAsync(checkLocation, true);
+
             Assert.Equal(HttpStatusCode.BadRequest, message.StatusCode);
             Assert.Contains(ImportProcessingJob.SurrogateIdsErrorMessage, await message.Content.ReadAsStringAsync());
         }