Adjust wrong unit test

AElfProject · Jul 20, 2024 · dc5ea6e · dc5ea6e
1 parent 69835d9
commit dc5ea6e
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/test/AElf.CSharp.CodeOps.Tests/ContractAuditorTests.cs b/test/AElf.CSharp.CodeOps.Tests/ContractAuditorTests.cs
@@ -110,13 +110,8 @@ public void AuditTimeout()
         Should.NotThrow(() => _auditor.Audit(ReadPatchedContractCode(typeof(TokenContract)), true));
         var codeOpsOptions = GetRequiredService<IOptionsMonitor<CSharpCodeOpsOptions>>();
         codeOpsOptions.CurrentValue.AuditTimeoutDuration = 0;
-        var exception =
-            Assert.Throws<AggregateException>(
-                () => _auditor.Audit(ReadPatchedContractCode(typeof(TokenContract)), true));
-        foreach (var ex in exception.InnerExceptions)
-        {
-            ex.GetType().ShouldBe(typeof(ContractAuditTimeoutException));
-        }
+        Should.Throw<ContractAuditTimeoutException>(() =>
+                    _auditor.Audit(ReadPatchedContractCode(typeof(TokenContract)), true));
     }
 
     [Fact]