chore: clean up after failed flatten test run

NomicFoundation · Nov 28, 2024 · f7c461a · f7c461a
1 parent e2ab08e
commit f7c461a
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/packages/hardhat-core/test/builtin-tasks/flatten.ts b/packages/hardhat-core/test/builtin-tasks/flatten.ts
@@ -416,14 +416,17 @@ describe("Flatten task", () => {
 
     it("should write to an output file when the parameter output is specified", async function () {
       const outputFile = `${tmpdir()}/flatten.sol`;
-      await this.env.run(TASK_FLATTEN, {
-        files: ["contracts/A.sol", "contracts/D.sol"],
-        output: outputFile,
-      });
-      const expected = await getExpectedSol();
-      const actual = readFileSync(outputFile, "utf8");
-      assert.equal(actual, expected);
-      removeSync(outputFile);
+      try {
+        await this.env.run(TASK_FLATTEN, {
+          files: ["contracts/A.sol", "contracts/D.sol"],
+          output: outputFile,
+        });
+        const expected = await getExpectedSol();
+        const actual = readFileSync(outputFile, "utf8");
+        assert.equal(actual, expected);
+      } finally {
+        removeSync(outputFile);
+      }
     });
 
     describe("No contracts to flatten", () => {