Merge pull request #164 from hartbit/set-unboxable-collection

Make set into an unboxable collection

Sources/Unbox.swift

@@ -465,6 +465,19 @@ extension Array: UnboxableCollection {
     }
 }
 
+/// Extension making Set an unboxable collection
+extension Set: UnboxableCollection {
+    public typealias UnboxValue = Element
+
+    public static func unbox<T: UnboxCollectionElementTransformer>(value: Any, allowInvalidElements: Bool, transformer: T) throws -> Set? where T.UnboxedElement == UnboxValue {
+        guard let array = try [UnboxValue].unbox(value: value, allowInvalidElements: allowInvalidElements, transformer: transformer) else {
+            return nil
+        }
+
+        return Set(array)
+    }
+}
+
 /// Extension making Dictionary an unboxable collection
 extension Dictionary: UnboxableCollection {
     public typealias UnboxValue = Value

Tests/UnboxTests.swift

@@ -1711,6 +1711,33 @@ class UnboxTests: XCTestCase {
             XCTFail("\(error)")
         }
     }
+
+    func testSets() {
+        struct Model: Unboxable {
+            let optional: Set<String>?
+            let required: Set<String>
+
+            init(unboxer: Unboxer) throws {
+                self.optional = unboxer.unbox(key: "optional")
+                self.required = try unboxer.unbox(key: "required")
+            }
+        }
+
+        let dictionary: UnboxableDictionary = [
+            "optional" : ["A", "A", "B"],
+            "required" : ["A"]
+        ]
+
+        do {
+            let unboxed: Model = try unbox(dictionary: dictionary)
+            XCTAssertEqual(unboxed.optional?.count, 2)
+            XCTAssertTrue(unboxed.optional?.contains("A") ?? false)
+            XCTAssertTrue(unboxed.optional?.contains("B") ?? false)
+            XCTAssertEqual(unboxed.required, ["A"])
+        } catch {
+            XCTFail("\(error)")
+        }
+    }
 }
 
 private func UnboxTestDictionaryWithAllRequiredKeysWithValidValues(nested: Bool) -> UnboxableDictionary {