Merge pull request #5 from Woollim/master

Add removeAll() method to WeakMapTable

Author: tokijh

Sources/WeakMapTable/WeakMapTable.swift

@@ -68,6 +68,13 @@ final public class WeakMapTable<Key, Value> where Key: AnyObject {
     }
   }
 
+  public func removeAll() {
+    self.lock.lock()
+    defer {
+        self.lock.unlock()
+    }
+    self.dictionary.removeAll()
+  }
 
   // MARK: Getting and Setting Values without Locking
 

Tests/WeakMapTableTests/WeakMapTableTests.swift

@@ -81,6 +81,23 @@ final class WeakMapTableTests: XCTestCase {
     XCTAssertNil(weakKey)
     XCTAssertNil(weakValue)
   }
+
+  func testRemoveAll() {
+    let map = WeakMapTable<KeyObject, ValueObject>()
+    let key = KeyObject()
+    weak var weakValue: ValueObject?
+
+    _ = {
+      let value = ValueObject()
+      map.setValue(value, forKey: key)
+      weakValue = value
+
+      map.removeAll()
+    }()
+
+    XCTAssertNil(map.value(forKey: key))
+    XCTAssertNil(weakValue)
+  }
 }
 
 private final class KeyObject {