Add Then extension to Set

devxoul · Sep 7, 2019 · 894b510 · 894b510
1 parent 87bcf03
commit 894b510
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Sources/Then/Then.swift b/Sources/Then/Then.swift
@@ -79,6 +79,7 @@ extension CGSize: Then {}
 extension CGVector: Then {}
 extension Array: Then {}
 extension Dictionary: Then {}
+extension Set: Then {}
 
 #if os(iOS) || os(tvOS)
   extension UIEdgeInsets: Then {}

diff --git a/Tests/ThenTests/ThenTests.swift b/Tests/ThenTests/ThenTests.swift
@@ -47,6 +47,13 @@ class ThenTests: XCTestCase {
     XCTAssertEqual(dict, ["Korea": "Seoul", "Japan": "Tokyo", "China": "Beijing"])
   }
 
+  func testWith_Set() {
+    let set = Set(["A", "B", "C"]).with {
+      $0.insert("D")
+    }
+    XCTAssertEqual(set, Set(["A", "B", "C", "D"]))
+  }
+
   func testDo() {
     UserDefaults.standard.do {
       $0.removeObject(forKey: "username")