From c8dc83303f96a34670e6d1a6b28119ee50c1d2db Mon Sep 17 00:00:00 2001
From: muyexi <muyexi@gmail.com>
Date: Thu, 8 Nov 2018 22:18:41 +0800
Subject: [PATCH] refactor(DemoTests): `lazy var`

---
 DemoTests/DemoTests.swift | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/DemoTests/DemoTests.swift b/DemoTests/DemoTests.swift
index 17ac00c..2a7aac8 100644
--- 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()
     }