From c4d854ad93da4e6296a7703f19887d095880c10f Mon Sep 17 00:00:00 2001
From: Blockiosaurus <blockiosaurus@gmail.com>
Date: Mon, 4 Nov 2024 22:19:36 -0500
Subject: [PATCH] Fixing feedback.

---
 clients/js/test/close/nonFungible.test.ts | 26 ++++++++++++++++-------
 1 file changed, 18 insertions(+), 8 deletions(-)

diff --git a/clients/js/test/close/nonFungible.test.ts b/clients/js/test/close/nonFungible.test.ts
index 7027c93d..01f64492 100644
--- a/clients/js/test/close/nonFungible.test.ts
+++ b/clients/js/test/close/nonFungible.test.ts
@@ -92,8 +92,7 @@ test.skip('it can close t22 ownerless metadata for a non-fungible with zero supp
   }
 
   const masterEditionLamports = await umi.rpc.getBalance(
-    // @ts-ignore
-    asset.edition.publicKey
+    asset.edition!.publicKey
   );
   const lamportsBefore = await umi.rpc.getBalance(closeDestination);
   await closeAccounts(umi, {
@@ -108,6 +107,12 @@ test.skip('it can close t22 ownerless metadata for a non-fungible with zero supp
     publicKey: asset.metadata.publicKey,
     exists: false,
   });
+  t.deepEqual(await umi.rpc.getAccount(asset.edition!.publicKey), <
+    MaybeRpcAccount
+  >{
+    publicKey: asset.edition!.publicKey,
+    exists: false,
+  });
   t.deepEqual(await umi.rpc.getBalance(asset.metadata.publicKey), lamports(0));
 
   const lamportsAfter = await umi.rpc.getBalance(closeDestination);
@@ -159,8 +164,7 @@ test.skip('it can close ownerless metadata for a non-fungible with zero supply',
   }
 
   const masterEditionLamports = await umi.rpc.getBalance(
-    // @ts-ignore
-    asset.edition.publicKey
+    asset.edition!.publicKey
   );
   const lamportsBefore = await umi.rpc.getBalance(closeDestination);
   await closeAccounts(umi, {
@@ -351,8 +355,8 @@ test.skip('it can close ownerless metadata for a non-fungible edition with zero
   if (asset.edition === undefined) {
     t.fail('Expected edition to exist');
   }
-  // @ts-ignore
-  const editionLamports = await umi.rpc.getBalance(asset.edition.publicKey);
+
+  const editionLamports = await umi.rpc.getBalance(asset.edition!.publicKey);
   const lamportsBefore = await umi.rpc.getBalance(closeDestination);
   await closeAccounts(umi, {
     mint: editionMint.publicKey,
@@ -450,8 +454,8 @@ test.skip('it can close ownerless metadata for a t22 non-fungible edition with z
   if (asset.edition === undefined) {
     t.fail('Expected edition to exist');
   }
-  // @ts-ignore
-  const editionLamports = await umi.rpc.getBalance(asset.edition.publicKey);
+
+  const editionLamports = await umi.rpc.getBalance(asset.edition!.publicKey);
   const lamportsBefore = await umi.rpc.getBalance(closeDestination);
   await closeAccounts(umi, {
     mint: editionMint.publicKey,
@@ -465,6 +469,12 @@ test.skip('it can close ownerless metadata for a t22 non-fungible edition with z
     publicKey: asset.metadata.publicKey,
     exists: false,
   });
+  t.deepEqual(await umi.rpc.getAccount(asset.edition!.publicKey), <
+    MaybeRpcAccount
+  >{
+    publicKey: asset.edition!.publicKey,
+    exists: false,
+  });
   t.deepEqual(await umi.rpc.getBalance(asset.metadata.publicKey), lamports(0));
 
   const lamportsAfter = await umi.rpc.getBalance(closeDestination);