Increased test coverage

devyhan · Dec 19, 2022 · 5291e0a · 5291e0a
1 parent 410cb9f
commit 5291e0a
Showing 1 changed file with 38 additions and 27 deletions.
diff --git a/Tests/APIRouterTests/URLBuilderTests.swift b/Tests/APIRouterTests/URLBuilderTests.swift
@@ -158,43 +158,54 @@ final class URLBuilderTests: XCTestCase {
     enum SchemeOptions {
       case https
       case http
-    }
-
-    let options: SchemeOptions = .http
-
-    let request = Request {
-      URL {
-        switch options {
-        case .https:
-          Scheme(.https)
-        case .http:
-          Scheme(.http)
-        }
-        Host("www.urltest.com")
+      
+      var request: Request {
+        Request {
+         URL {
+           switch self {
+           case .https:
+             Scheme(.https)
+           case .http:
+             Scheme(.http)
+           }
+           Host("www.urltest.com")
+         }
+       }
       }
     }
 
-    if let url = request.urlRequest?.url?.absoluteString {
-      XCTAssertEqual(url, "http://www.urltest.com")
+    if let httpUrl = SchemeOptions.http.request.urlRequest?.url?.absoluteString {
+      XCTAssertEqual(httpUrl, "http://www.urltest.com")
+    }
+    if let httpsUrl = SchemeOptions.https.request.urlRequest?.url?.absoluteString {
+      XCTAssertEqual(httpsUrl, "https://www.urltest.com")
     }
   }
 
   func testIfConditionalStatementWorkingForBuildEitherInUrlBuilder() {
-    let conditional = true
-
-    let request = Request {
-      URL {
-        if conditional == true {
-          Scheme(.http)
-        } else {
-          Scheme(.https)
-        }
-        Host("www.urltest.com")
+    enum SchemeOptions {
+      case https
+      case http
+
+      var request: Request {
+        Request {
+         URL {
+           if self == .http {
+             Scheme(.http)
+           } else {
+             Scheme(.https)
+           }
+           Host("www.urltest.com")
+         }
+       }
       }
     }
 
-    if let url = request.urlRequest?.url?.absoluteString {
-      XCTAssertEqual(url, "http://www.urltest.com")
+    if let httpUrl = SchemeOptions.http.request.urlRequest?.url?.absoluteString {
+      XCTAssertEqual(httpUrl, "http://www.urltest.com")
+    }
+    if let httpsUrl = SchemeOptions.https.request.urlRequest?.url?.absoluteString {
+      XCTAssertEqual(httpsUrl, "https://www.urltest.com")
     }
   }