From 5c45b2e9f5d96e5b047f8b17686de916b389c33f Mon Sep 17 00:00:00 2001
From: Felix Mau <me@felix.hamburg>
Date: Tue, 10 May 2022 18:35:56 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20::=20Adapt=20tests?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ...hGradientLoadingBarViewModelTestCase.swift | 47 +++++++++++++------
 1 file changed, 33 insertions(+), 14 deletions(-)

diff --git a/Example/ExampleTests/ViewModel/NotchGradientLoadingBarViewModelTestCase.swift b/Example/ExampleTests/ViewModel/NotchGradientLoadingBarViewModelTestCase.swift
index 9c7ace7..171bf28 100644
--- a/Example/ExampleTests/ViewModel/NotchGradientLoadingBarViewModelTestCase.swift
+++ b/Example/ExampleTests/ViewModel/NotchGradientLoadingBarViewModelTestCase.swift
@@ -11,6 +11,7 @@ import XCTest
 @testable import GradientLoadingBar
 
 final class NotchGradientLoadingBarViewModelTestCase: XCTestCase {
+
     func test_initializer_shouldSetSafeAreaDevice_toIPhoneX() {
         // Given
         let deviceIdentifiers = ["iPhone10,3", "iPhone10,6", "iPhone11,2", "iPhone11,4", "iPhone11,6"]
@@ -26,28 +27,46 @@ final class NotchGradientLoadingBarViewModelTestCase: XCTestCase {
 
     func test_initializer_shouldSetSafeAreaDevice_toIPhoneXR() {
         // Given
-        let deviceIdentifiers = ["iPhone11,8"]
-        deviceIdentifiers.forEach { deviceIdentifier in
+        let deviceIdentifier = "iPhone11,8"
 
-            // When
-            let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
+        // When
+        let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
 
-            // Then
-            XCTAssertEqual(viewModel.safeAreaDevice, .iPhoneXR)
-        }
+        // Then
+        XCTAssertEqual(viewModel.safeAreaDevice, .iPhoneXR)
     }
 
     func test_initializer_shouldSetSafeAreaDevice_toIPhone11() {
         // Given
-        let deviceIdentifiers = ["iPhone12,1", "iPhone12,3", "iPhone12,5"]
-        deviceIdentifiers.forEach { deviceIdentifier in
+        let deviceIdentifier = "iPhone12,1"
 
-            // When
-            let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
+        // When
+        let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
 
-            // Then
-            XCTAssertEqual(viewModel.safeAreaDevice, .iPhone11)
-        }
+        // Then
+        XCTAssertEqual(viewModel.safeAreaDevice, .iPhone11)
+    }
+
+    func test_initializer_shouldSetSafeAreaDevice_toIPhone11Pro() {
+        // Given
+        let deviceIdentifier = "iPhone12,3"
+
+        // When
+        let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
+
+        // Then
+        XCTAssertEqual(viewModel.safeAreaDevice, .iPhone11Pro)
+    }
+
+    func test_initializer_shouldSetSafeAreaDevice_toIPhone11ProMax() {
+        // Given
+        let deviceIdentifier = "iPhone12,5"
+
+        // When
+        let viewModel = NotchGradientLoadingBarViewModel(deviceIdentifier: deviceIdentifier)
+
+        // Then
+        XCTAssertEqual(viewModel.safeAreaDevice, .iPhone11ProMax)
     }
 
     func test_initializer_shouldSetSafeAreaDevice_toIPhone12() {