ConfirmAssetViewControllerTests.swift

//
// Wire
// Copyright (C) 2016 Wire Swiss GmbH
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see http://www.gnu.org/licenses/.
//

import Cartography
@testable import Wire

class ConfirmAssetViewControllerTests: ZMSnapshotTestCase {

    var sut: ConfirmAssetViewController!
    
    override func setUp() {
        super.setUp()
        sut = ConfirmAssetViewController()
        snapshotBackgroundColor = UIColor.white
    }

    func testThatItRendersTheAssetViewControllerWithLandscapeImage() {
        accentColor = .strongLimeGreen
        sut.image = image(inTestBundleNamed: "unsplash_matterhorn.jpg")
        sut.previewTitle = "Matterhorn"
        verifyInAllIPhoneSizes(view: sut.prepareForSnapshot())
    }
        
    func testThatItRendersTheAssetViewControllerWithPortraitImage() {
        accentColor = .vividRed
        sut.image = image(inTestBundleNamed: "unsplash_burger.jpg")
        sut.previewTitle = "Burger & Beer"
        verifyInAllIPhoneSizes(view: sut.prepareForSnapshot())
    }
    
    func testThatItRendersTheAssetViewControllerWithSmallImage() {
        accentColor = .vividRed
        sut.image = image(inTestBundleNamed: "unsplash_small.jpg").imageScaled(withFactor: 0.5);
        sut.previewTitle = "Sea Food"
        verifyInAllIPhoneSizes(view: sut.prepareForSnapshot())
    }

}

private extension UIViewController {
    func prepareForSnapshot() -> UIView {
        beginAppearanceTransition(true, animated: false)
        endAppearanceTransition()
        return view
    }
}