Test TTLMap.delete

Author: TheKodeToad

backend/src/common/ttlMap.test.ts

@@ -105,6 +105,19 @@ suite("TTLMap", () => {
 		assert.ok(!map.has("a"), "!map.has('a')");
 	}));
 
+	test("delete", () => mockTime(() => {
+		const map: TTLMap<string, null> = new TTLMap(SECOND);
+
+		map.set("a", null);
+		assert.ok(map.delete("a"), "map.delete('a')");
+		assert.ok(!map.delete("a"), "!map.delete('a')");
+
+		map.set("a", null);
+		time += SECOND;
+
+		assert.ok(!map.delete("a"), "!map.delete('a') (after expiry)");
+	}));
+
 	test("precision", () => mockTime(() => {
 		const map: TTLMap<string, number> = new TTLMap(SECOND);
 

backend/src/common/ttlMap.ts

@@ -42,7 +42,9 @@ export class TTLMap<K, V> {
 	}
 
 	delete(key: K): boolean {
-		return this._map.delete(key);
+		const had = this.has(key);
+		this._map.delete(key);
+		return had;
 	}
 
 	forEach(callbackfn: (value: V, key: K, map: TTLMap<K, V>) => void, thisArg?: any): void {