diff --git a/sdk/resourcemanager/azure-resourcemanager-dns/src/test/java/com/azure/resourcemanager/dns/DnsZoneRecordSetETagTests.java b/sdk/resourcemanager/azure-resourcemanager-dns/src/test/java/com/azure/resourcemanager/dns/DnsZoneRecordSetETagTests.java index 4e0b3b3ffa55..7d3416d359b6 100644 --- a/sdk/resourcemanager/azure-resourcemanager-dns/src/test/java/com/azure/resourcemanager/dns/DnsZoneRecordSetETagTests.java +++ b/sdk/resourcemanager/azure-resourcemanager-dns/src/test/java/com/azure/resourcemanager/dns/DnsZoneRecordSetETagTests.java @@ -187,15 +187,18 @@ public void canCreateRecordSetsWithDefaultETag() throws Exception { } Assertions.assertNotNull(compositeException); - Assertions.assertEquals(5, compositeException.getSuppressed().length); - for (int i = 0; i < 4; ++i) { + int managementExceptionCount = 0; + for (int i = 0; i < compositeException.getSuppressed().length; ++i) { Throwable exception = compositeException.getSuppressed()[i]; - Assertions.assertTrue(exception instanceof ManagementException); - ManagementError cloudError = ((ManagementException) exception).getValue(); - Assertions.assertNotNull(cloudError); - Assertions.assertNotNull(cloudError.getCode()); - Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + if (exception instanceof ManagementException) { + ++managementExceptionCount; + ManagementError cloudError = ((ManagementException) exception).getValue(); + Assertions.assertNotNull(cloudError); + Assertions.assertNotNull(cloudError.getCode()); + Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + } } + Assertions.assertTrue(managementExceptionCount > 0); } @Test @@ -252,15 +255,18 @@ public void canUpdateRecordSetWithExplicitETag() throws Exception { compositeException = exception; } Assertions.assertNotNull(compositeException); - Assertions.assertEquals(3, compositeException.getSuppressed().length); - for (int i = 0; i < 2; ++i) { + int managementExceptionCount = 0; + for (int i = 0; i < compositeException.getSuppressed().length; ++i) { Throwable exception = compositeException.getSuppressed()[i]; - Assertions.assertTrue(exception instanceof ManagementException); - ManagementError cloudError = ((ManagementException) exception).getValue(); - Assertions.assertNotNull(cloudError); - Assertions.assertNotNull(cloudError.getCode()); - Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + if (exception instanceof ManagementException) { + ++managementExceptionCount; + ManagementError cloudError = ((ManagementException) exception).getValue(); + Assertions.assertNotNull(cloudError); + Assertions.assertNotNull(cloudError.getCode()); + Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + } } + Assertions.assertTrue(managementExceptionCount > 0); // Try update with correct etags dnsZone .update() @@ -335,15 +341,18 @@ public void canDeleteRecordSetWithExplicitETag() throws Exception { compositeException = exception; } Assertions.assertNotNull(compositeException); - Assertions.assertEquals(3, compositeException.getSuppressed().length); - for (int i = 0; i < 2; ++i) { + int managementExceptionCount = 0; + for (int i = 0; i < compositeException.getSuppressed().length; ++i) { Throwable exception = compositeException.getSuppressed()[i]; - Assertions.assertTrue(exception instanceof ManagementException); - ManagementError cloudError = ((ManagementException) exception).getValue(); - Assertions.assertNotNull(cloudError); - Assertions.assertNotNull(cloudError.getCode()); - Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + if (exception instanceof ManagementException) { + ++managementExceptionCount; + ManagementError cloudError = ((ManagementException) exception).getValue(); + Assertions.assertNotNull(cloudError); + Assertions.assertNotNull(cloudError.getCode()); + Assertions.assertTrue(cloudError.getCode().contains("PreconditionFailed")); + } } + Assertions.assertTrue(managementExceptionCount > 0); // Try delete with correct etags dnsZone .update()