NetNewsWire/Frameworks/Account/AccountTests/Feedly/FeedlyOperationTests.swift

//
//  FeedlyOperationTests.swift
//  AccountTests
//
//  Created by Kiel Gillard on 21/10/19.
//  Copyright © 2019 Ranchero Software, LLC. All rights reserved.
//

import XCTest
@testable import Account

class FeedlyOperationTests: XCTestCase {
	
	enum TestOperationError: Error, Equatable {
		case mockError
		case anotherMockError
	}
	
	final class TestOperation: FeedlyOperation {
		var didCallMainExpectation: XCTestExpectation?
		var mockError: Error?
		
		override func main() {
			// Should always call on main thread.
			XCTAssertTrue(Thread.isMainThread)
			
			didCallMainExpectation?.fulfill()
			
			if let error = mockError {
				didFinish(error)
			} else {
				didFinish()
			}
		}
	}
	
	final class TestDelegate: FeedlyOperationDelegate {
		
		var error: Error?
		var didFailExpectation: XCTestExpectation?
		
		func feedlyOperation(_ operation: FeedlyOperation, didFailWith error: Error) {
			didFailExpectation?.fulfill()
			self.error = error
		}
	}

    func testDoesCallMain() {
        let testOperation = TestOperation()
		testOperation.didCallMainExpectation = expectation(description: "Did Call Main")
		
		OperationQueue.main.addOperation(testOperation)
		
		waitForExpectations(timeout: 2)
    }

	func testDoesFail() {
        let testOperation = TestOperation()
		testOperation.didCallMainExpectation = expectation(description: "Did Call Main")
		testOperation.mockError = TestOperationError.mockError
		
		let delegate = TestDelegate()
		delegate.didFailExpectation = expectation(description: "Operation Failed As Expected")
		
		testOperation.delegate = delegate
		
		OperationQueue.main.addOperation(testOperation)
		
		waitForExpectations(timeout: 2)
		
		if let error = delegate.error as? TestOperationError {
			XCTAssertEqual(error, TestOperationError.mockError)
		} else {
			XCTFail("Expected \(TestOperationError.self) but got \(String(describing: delegate.error)).")
		}
    }
	
	func testOperationFlags() {
        let testOperation = TestOperation()
		testOperation.didCallMainExpectation = expectation(description: "Did Call Main")
		
		let completionExpectation = expectation(description: "Operation Completed")
		testOperation.completionBlock = {
			completionExpectation.fulfill()
		}
		
		XCTAssertTrue(testOperation.isReady)
		XCTAssertFalse(testOperation.isFinished)
		XCTAssertFalse(testOperation.isExecuting)
		XCTAssertFalse(testOperation.isCancelled)
		
		OperationQueue.main.addOperation(testOperation)
		
		waitForExpectations(timeout: 2)
		
		XCTAssertTrue(testOperation.isReady)
		XCTAssertTrue(testOperation.isFinished)
		XCTAssertFalse(testOperation.isExecuting)
		XCTAssertFalse(testOperation.isCancelled)
    }
	
	func testOperationCancellationFlags() {
        let testOperation = TestOperation()
		testOperation.didCallMainExpectation = expectation(description: "Did Call Main")
		
		let completionExpectation = expectation(description: "Operation Completed")
		testOperation.completionBlock = {
			completionExpectation.fulfill()
		}
		
		XCTAssertTrue(testOperation.isReady)
		XCTAssertFalse(testOperation.isFinished)
		XCTAssertFalse(testOperation.isExecuting)
		XCTAssertFalse(testOperation.isCancelled)
		
		OperationQueue.main.addOperation(testOperation)
		
		testOperation.cancel()
		
		waitForExpectations(timeout: 2)
		
		XCTAssertTrue(testOperation.isReady)
		XCTAssertTrue(testOperation.isFinished)
		XCTAssertFalse(testOperation.isExecuting)
		XCTAssertTrue(testOperation.isCancelled)
    }
	
	func testDependency() {
        let testOperation = TestOperation()
		testOperation.didCallMainExpectation = expectation(description: "Did Call Main")
		
		let dependencyExpectation = expectation(description: "Did Call Dependency")
		let blockOperation = BlockOperation {
			dependencyExpectation.fulfill()
		}
		
		blockOperation.addDependency(testOperation)
		
		XCTAssertTrue(blockOperation.dependencies.contains(testOperation))
		
		OperationQueue.main.addOperations([testOperation, blockOperation], waitUntilFinished: false)
		
		waitForExpectations(timeout: 2)
    }

}