Merge pull request #38 from danramteke/master

Add `hasNotch` property

Sources/DeviceModel.swift

@@ -21,7 +21,7 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 // SOFTWARE.
 
-public enum DeviceModel {
+public enum DeviceModel: CaseIterable {
 
     case iPhone4, iPhone4S
     case iPhone5, iPhone5C, iPhone5S
@@ -201,3 +201,17 @@ extension DeviceModel {
         }
     }
 }
+
+// MARK: Detecting the Notch
+
+extension DeviceModel {
+    public var hasNotch: Bool {
+        switch self {
+
+        case .iPhoneX, .iPhoneXS, .iPhoneXSMax, .iPhoneXR:
+            return true
+        default:
+          return false
+        }
+    }
+}

Tests/DeviceModelTests.swift

@@ -311,5 +311,14 @@ class DeviceModelTests: XCTestCase {
 
         XCTAssert(deviceModel == .unknown , "DeviceModel - .unknown is failing")
     }
-
+
+    // MARK: Notch test
+    func testHasNotch() {
+      let notchModels: [DeviceModel] = [.iPhoneX, .iPhoneXS, .iPhoneXSMax, .iPhoneXR]
+
+      let noNotchModels: [DeviceModel] = DeviceModel.allCases.filter( { !notchModels.contains($0) })
+
+      notchModels.forEach { XCTAssertTrue($0.hasNotch) }
+      noNotchModels.forEach { XCTAssertFalse($0.hasNotch) }
+    }
 }

Tests/UIDeviceExtensionsTests.swift

@@ -55,5 +55,4 @@ class UIDeviceExtensionsTests: XCTestCase {
         let deviceFamily = DeviceFamily(rawValue: "x86_64")
         XCTAssert(deviceFamily == .simulator, "DeviceExtensions - .isSimulator is failing")
     }
-
 }