SwiftFormat/Snapshots/Layout/LayoutTests/Loading/ReloadManagerTests.swift

//  Copyright © 2017 Schibsted. All rights reserved.

import XCTest
@testable import Layout

final class ReloadManagerTests: XCTestCase {
    private final class TestController: UIViewController, LayoutLoading {
        var loadCount = 0

        func layoutDidLoad(_: LayoutNode) {
            loadCount += 1
        }
    }

    private final class TestView: UIView, LayoutLoading {}

    func testReload() {
        let vc = TestController()
        vc.loadLayout(named: "LayoutDidLoad_Valid.xml", bundle: Bundle(for: type(of: self)))
        XCTAssertNotNil(vc.layoutNode)
        XCTAssertEqual(vc.loadCount, 1)

        ReloadManager.reload(hard: true)
        XCTAssertEqual(vc.loadCount, 2)
    }

    func testIgnoreDuplicateRegistration() {
        let vc = TestController()
        vc.loadLayout(named: "LayoutDidLoad_Valid.xml", bundle: Bundle(for: type(of: self)))
        XCTAssertEqual(vc.loadCount, 1)
        ReloadManager.addObserver(vc)
        ReloadManager.addObserver(vc)
        ReloadManager.addObserver(vc)

        ReloadManager.reload(hard: true)
        XCTAssertEqual(vc.loadCount, 2)
    }

    func testObserverDoesNotRetainView() {
        weak var weakRef: TestController?
        let initialObserverCount = ReloadManager.observers.count
        autoreleasepool {
            var vc: TestController? = TestController()
            weakRef = vc
            vc?.loadLayout(named: "LayoutDidLoad_Valid.xml", bundle: Bundle(for: type(of: self)))
            ReloadManager.reload(hard: true)
            XCTAssert(ReloadManager.observers.contains { $0 === vc })
            XCTAssertNotNil(weakRef)
            vc = nil
        }
        XCTAssertNil(weakRef)
        XCTAssertEqual(ReloadManager.observers.count, initialObserverCount)
    }

    func testViewReleasedAfterReload() {
        weak var view: UIView?
        let container = TestView()
        autoreleasepool {
            container.loadLayout(named: "LayoutDidLoad_Valid.xml", bundle: Bundle(for: type(of: self)))
            view = container.layoutNode?.view
            XCTAssertNotNil(view)
            container.reloadLayout()
        }
        XCTAssertNotNil(container.layoutNode?.view)
        XCTAssertNil(view)
    }

    func testVCViewReleasedAfterReload() {
        weak var view: UIView?
        let vc = TestController()
        autoreleasepool {
            vc.loadLayout(named: "LayoutDidLoad_Valid.xml", bundle: Bundle(for: type(of: self)))
            view = vc.layoutNode?.view
            XCTAssertNotNil(view)
            vc.reloadLayout()
        }
        XCTAssertNotNil(vc.layoutNode?.view)
        XCTAssertNil(view)
    }
}