Update the removeSupportAllNFTs method (#9017)

LiskArchive · Sep 21, 2023 · 22213a7 · 22213a7
1 parent cf5c582
commit 22213a7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/framework/src/modules/nft/method.ts b/framework/src/modules/nft/method.ts
@@ -711,6 +711,8 @@ export class NFTMethod extends BaseMethod {
 			await supportedNFTsStore.del(methodContext, key);
 		}
 
+		await supportedNFTsStore.del(methodContext, ALL_SUPPORTED_NFTS_KEY);
+
 		this.events.get(AllNFTsSupportRemovedEvent).log(methodContext);
 	}
 

diff --git a/framework/test/unit/modules/nft/method.spec.ts b/framework/test/unit/modules/nft/method.spec.ts
@@ -1189,6 +1189,20 @@ describe('NFTMethod', () => {
 
 			checkEventResult(methodContext.eventQueue, 1, AllNFTsSupportRemovedEvent, 0, {}, null);
 		});
+
+		it('should remove all existing entries even if the ALL_SUPPORTED_NFTS_KEY entry exists', async () => {
+			await supportedNFTsStore.save(methodContext, ALL_SUPPORTED_NFTS_KEY, {
+				supportedCollectionIDArray: [],
+			});
+
+			await expect(method.removeSupportAllNFTs(methodContext)).resolves.toBeUndefined();
+			await expect(
+				supportedNFTsStore.has(methodContext, ALL_SUPPORTED_NFTS_KEY),
+			).resolves.toBeFalse();
+
+			checkEventResult(methodContext.eventQueue, 1, AllNFTsSupportRemovedEvent, 0, {}, null);
+			expect(methodContext.eventQueue.getEvents()).toHaveLength(1);
+		});
 	});
 
 	describe('supportAllNFTsFromChain', () => {