fix clearStorage function

processing · May 3, 2024 · c658000 · c658000
1 parent 7cb4f9e
commit c658000
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/src/data/local_storage.js b/src/data/local_storage.js
@@ -194,8 +194,13 @@ p5.prototype.getItem = function(key) {
  * }
  * </code></div>
  */
-p5.prototype.clearStorage = function() {
-  localStorage.clear();
+p5.prototype.clearStorage = function () {
+  const keys = Object.keys(localStorage);
+  keys.forEach(key => {
+    if (key.endsWith('p5TypeID')) {
+      this.removeItem(key.replace('p5TypeID', ''));
+    }
+  });
 };
 
 /**

diff --git a/test/unit/data/local_storage.js b/test/unit/data/local_storage.js
@@ -105,4 +105,18 @@ suite('local storage', function() {
       checkRemoval('myVector');
     });
   });
+
+  suite('should be able to clear all items at once', function () {
+    test('should remove all items set by storeItem()', function () {
+      localStorage.setItem('extra', 'stuff');
+      myp5.clearStorage();
+      assert.deepEqual(myp5.getItem('myBoolean'), null);
+      assert.deepEqual(myp5.getItem('myNumber'), null);
+      assert.deepEqual(myp5.getItem('myObject'), null);
+      assert.deepEqual(myp5.getItem('myString'), null);
+      assert.deepEqual(myp5.getItem('myColor'), null);
+      assert.deepEqual(myp5.getItem('myVector'), null);
+      assert.deepEqual(myp5.getItem('extra'), 'stuff');
+    });
+  });
 });