Fix ReviewViewControllerTests

nartex · Apr 25, 2019 · 181a2e9 · 181a2e9
1 parent 0d81fc2
commit 181a2e9
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/WeScanTests/ReviewViewControllerTests.swift b/WeScanTests/ReviewViewControllerTests.swift
@@ -13,8 +13,8 @@ import CoreGraphics
 
 final class ReviewViewControllerTests: FBSnapshotTestCase {
 
-    var demoImage: UIImage!
-    var enhancedDemoImage: UIImage!
+    var demoScan: ImageScannerScan!
+    var enhancedDemoScan: ImageScannerScan!
     var demoQuad = Quadrilateral(topLeft: .zero, topRight: .zero, bottomRight: .zero, bottomLeft: .zero)
 
     override func setUp() {
@@ -35,11 +35,13 @@ final class ReviewViewControllerTests: FBSnapshotTestCase {
 
         UIGraphicsBeginImageContextWithOptions(backgroundSize, true, 1.0)
         backgroundLayer.render(in: UIGraphicsGetCurrentContext()!)
-        demoImage = UIGraphicsGetImageFromCurrentImageContext()!
+        let image = UIGraphicsGetImageFromCurrentImageContext()!
+        demoScan = ImageScannerScan(image: image)
 
         backgroundLayer.backgroundColor = UIColor.black.cgColor
         backgroundLayer.render(in: UIGraphicsGetCurrentContext()!)
-        enhancedDemoImage = UIGraphicsGetImageFromCurrentImageContext()!
+        let enhancedImage = UIGraphicsGetImageFromCurrentImageContext()!
+        enhancedDemoScan = ImageScannerScan(image: enhancedImage)
 
         UIGraphicsEndImageContext()
     }
@@ -96,7 +98,7 @@ final class ReviewViewControllerTests: FBSnapshotTestCase {
     }
 
     func testEnhancedImage() {
-        let results = ImageScannerResults(originalImage: demoImage, scannedImage: demoImage, enhancedImage: enhancedDemoImage, doesUserPreferEnhancedImage: false, detectedRectangle: demoQuad)
+        let results = ImageScannerResults(detectedRectangle: demoQuad, originalScan: demoScan, croppedScan: demoScan, enhancedScan: enhancedDemoScan, doesUserPreferEnhancedScan: false)
         let viewController = ReviewViewController(results: results)
         viewController.viewDidLoad()