refactor(DemoTests): lazy var

muyexi · Nov 8, 2018 · c8dc833 · c8dc833
1 parent 39f2e6d
commit c8dc833
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/DemoTests/DemoTests.swift b/DemoTests/DemoTests.swift
@@ -2,15 +2,17 @@ import XCTest
 @testable import Demo
 
 class DemoTests: XCTestCase {
-    var viewController: ViewController!
+    lazy var viewController: ViewController = {
+        let storyboard = UIStoryboard(name: "Main", bundle: Bundle.main)
+        let id = String(describing: ViewController.self)
+
+        let viewController = storyboard.instantiateViewController(withIdentifier: id) as! ViewController
+        return viewController
+    }()
 
     override func setUp() {
         super.setUp()
 
-        let storyboard = UIStoryboard(name: "Main", bundle: Bundle.main)
-        let id = String(describing: ViewController.self)
-
-        viewController = storyboard.instantiateViewController(withIdentifier: id) as! ViewController
         viewController.viewDidLoad()
     }