Skip to content

[FormRecognizer] Cleaning up tests (1/?) #37705

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
kinelski merged 10 commits into from
Jul 21, 2023
Merged

[FormRecognizer] Cleaning up tests (1/?) #37705

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
c95e41d
Removed fixed/dynamic rows tests
kinelski Jul 17, 2023
a31ac9e
Added [AsyncOnly] attribute to samples
kinelski Jul 18, 2023
69f0c5e
Replaced options.Pages live tests with mock tests in v2
kinelski Jul 18, 2023
b3f59ab
Removed DamagedFile tests in v2
kinelski Jul 18, 2023
8bbc590
Replace reading order live tests with mock tests in v2
kinelski Jul 18, 2023
99ae6cb
Replaced language/locale live tests with mock tests in v2
kinelski Jul 18, 2023
a8a2ee7
Replace LRO live tests with mock tests in v2
kinelski Jul 18, 2023
e170f4a
Replace LRO live tests with mock tests in v3
kinelski Jul 18, 2023
423cf97
Updated recordings
kinelski Jul 18, 2023
542e8ad
Minor fixes
kinelski Jul 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 0 additions & 48 deletions ...r/Azure.AI.FormRecognizer/tests/DocumentAnalysisClient/DocumentAnalysisClientLiveTests.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -443,54 +443,6 @@ public async Task AnalyzeDocumentWithCustomModelCanParseDifferentTypeOfForm()
field.FieldType == DocumentFieldType.Unknown && field.ExpectedFieldType == DocumentFieldType.String));
}

[RecordedTest]
public async Task AnalyzeDocumentWithCustomModelWithTableDynamicRows()
{
var client = CreateDocumentAnalysisClient();
var modelId = Recording.GenerateId();
AnalyzeDocumentOperation operation;

await using var customModel = await BuildDisposableDocumentModelAsync(modelId, ContainerType.TableVariableRows);

using var stream = DocumentAnalysisTestEnvironment.CreateStream(TestFile.FormTableDynamicRows);
using (Recording.DisableRequestBodyRecording())
{
operation = await client.AnalyzeDocumentAsync(WaitUntil.Completed, customModel.ModelId, stream);
}

AnalyzeResult result = operation.Value;

ValidateAnalyzeResult(
result,
customModel.ModelId,
expectedFirstPageNumber: 1,
expectedLastPageNumber: 1);
}

[RecordedTest]
public async Task AnalyzeDocumentWithCustomModelWithTableFixedRows()
{
var client = CreateDocumentAnalysisClient();
var modelId = Recording.GenerateId();
AnalyzeDocumentOperation operation;

await using var customModel = await BuildDisposableDocumentModelAsync(modelId, ContainerType.TableFixedRows);

using var stream = DocumentAnalysisTestEnvironment.CreateStream(TestFile.FormTableFixedRows);
using (Recording.DisableRequestBodyRecording())
{
operation = await client.AnalyzeDocumentAsync(WaitUntil.Completed, customModel.ModelId, stream);
}

AnalyzeResult result = operation.Value;

ValidateAnalyzeResult(
result,
customModel.ModelId,
expectedFirstPageNumber: 1,
expectedLastPageNumber: 1);
}

[RecordedTest]
[Ignore("Service error. Issue https://github.com/Azure/azure-sdk-for-net/issues/24995")]
public async Task AnalyzeDocumentWithCustomModelCanParseBlankPage()
Expand Down
112 changes: 0 additions & 112 deletions ...ormrecognizer/Azure.AI.FormRecognizer/tests/DocumentAnalysisModels/OperationsLiveTests.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,6 @@
using Azure.Core.TestFramework;
using NUnit.Framework;

using TestFile = Azure.AI.FormRecognizer.Tests.TestFile;

namespace Azure.AI.FormRecognizer.DocumentAnalysis.Tests
{
/// <summary>
Expand All @@ -30,39 +28,6 @@ public OperationsLiveTests(bool isAsync, DocumentAnalysisClientOptions.ServiceVe
{
}

[RecordedTest]
public async Task AnalyzeDocumentOperationCanPollFromNewObject()
{
var client = CreateDocumentAnalysisClient(out var nonInstrumentedClient);

var uri = DocumentAnalysisTestEnvironment.CreateUri(TestFile.ReceiptJpg);
var operation = await client.AnalyzeDocumentFromUriAsync(WaitUntil.Started, "prebuilt-receipt", uri);
Assert.IsNotNull(operation.GetRawResponse());

var sameOperation = InstrumentOperation(new AnalyzeDocumentOperation(operation.Id, nonInstrumentedClient));
await sameOperation.WaitForCompletionAsync();

Assert.IsTrue(sameOperation.HasValue);
Assert.AreEqual(1, sameOperation.Value.Pages.Count);
}

[RecordedTest]
public async Task BuildModelOperationCanPollFromNewObject()
{
var client = CreateDocumentModelAdministrationClient(out var nonInstrumentedClient);
var trainingFilesUri = new Uri(TestEnvironment.BlobContainerSasUrl);
var modelId = Recording.GenerateId();

var operation = await client.BuildDocumentModelAsync(WaitUntil.Started, trainingFilesUri, DocumentBuildMode.Template, modelId);
Assert.IsNotNull(operation.GetRawResponse());

var sameOperation = InstrumentOperation(new BuildDocumentModelOperation(operation.Id, nonInstrumentedClient));
await sameOperation.WaitForCompletionAsync();

Assert.IsTrue(sameOperation.HasValue);
Assert.AreEqual(modelId, sameOperation.Value.ModelId);
}

[RecordedTest]
public async Task BuildModelOperationPercentageCompletedValue()
{
Expand All @@ -79,27 +44,6 @@ public async Task BuildModelOperationPercentageCompletedValue()
Assert.AreEqual(100, operation.PercentCompleted);
}

[RecordedTest]
public async Task CopyModelToOperationCanPollFromNewObject()
{
var client = CreateDocumentModelAdministrationClient(out var nonInstrumentedClient);
var modelId = Recording.GenerateId();

await using var trainedModel = await BuildDisposableDocumentModelAsync(modelId);

var targetModelId = Recording.GenerateId();
DocumentModelCopyAuthorization targetAuth = await client.GetCopyAuthorizationAsync(targetModelId);

var operation = await client.CopyDocumentModelToAsync(WaitUntil.Started, trainedModel.ModelId, targetAuth);
Assert.IsNotNull(operation.GetRawResponse());

var sameOperation = InstrumentOperation(new CopyDocumentModelToOperation(operation.Id, nonInstrumentedClient));
await sameOperation.WaitForCompletionAsync();

Assert.IsTrue(sameOperation.HasValue);
Assert.AreEqual(targetModelId, sameOperation.Value.ModelId);
}

[RecordedTest]
public async Task CopyModelToOperationPercentageCompletedValue()
{
Expand All @@ -120,62 +64,6 @@ public async Task CopyModelToOperationPercentageCompletedValue()
Assert.AreEqual(100, operation.PercentCompleted);
}

[RecordedTest]
[ServiceVersion(Min = DocumentAnalysisClientOptions.ServiceVersion.V2023_02_28_Preview)]
public async Task ClassifyDocumentOperationCanPollFromNewObject()
{
var client = CreateDocumentAnalysisClient(out var nonInstrumentedClient);
var classifierId = Recording.GenerateId();
await using var disposableClassifier = await BuildDisposableDocumentClassifierAsync(classifierId);
var uri = DocumentAnalysisTestEnvironment.CreateUri(TestFile.Irs1040);

var operation = await client.ClassifyDocumentFromUriAsync(WaitUntil.Started, classifierId, uri);
var sameOperation = InstrumentOperation(new ClassifyDocumentOperation(operation.Id, nonInstrumentedClient));

await sameOperation.WaitForCompletionAsync();

Assert.IsTrue(sameOperation.HasValue);
Assert.AreEqual(4, sameOperation.Value.Pages.Count);
}

[RecordedTest]
[ServiceVersion(Min = DocumentAnalysisClientOptions.ServiceVersion.V2023_02_28_Preview)]
public async Task BuildClassifierOperationCanPollFromNewObject()
{
var client = CreateDocumentModelAdministrationClient(out var nonInstrumentedClient);
var classifierId = Recording.GenerateId();

var trainingFilesUri = new Uri(TestEnvironment.ClassifierTrainingSasUrl);
var sourceA = new AzureBlobContentSource(trainingFilesUri) { Prefix = "IRS-1040-A/train" };
var sourceB = new AzureBlobContentSource(trainingFilesUri) { Prefix = "IRS-1040-B/train" };

var documentTypes = new Dictionary<string, ClassifierDocumentTypeDetails>()
{
{ "IRS-1040-A", new ClassifierDocumentTypeDetails(sourceA) },
{ "IRS-1040-B", new ClassifierDocumentTypeDetails(sourceB) }
};

BuildDocumentClassifierOperation sameOperation = null;

try
{
var operation = await client.BuildDocumentClassifierAsync(WaitUntil.Started, documentTypes, classifierId);

sameOperation = InstrumentOperation(new BuildDocumentClassifierOperation(operation.Id, nonInstrumentedClient));
await sameOperation.WaitForCompletionAsync();
}
finally
{
if (sameOperation != null && sameOperation.HasValue)
{
await client.DeleteDocumentClassifierAsync(classifierId);
}
}

Assert.IsTrue(sameOperation.HasValue);
Assert.AreEqual(classifierId, sameOperation.Value.ClassifierId);
}

[RecordedTest]
[ServiceVersion(Min = DocumentAnalysisClientOptions.ServiceVersion.V2023_02_28_Preview)]
public async Task BuildClassifierOperationPercentageCompletedValue()
Expand Down
Loading