HasEventsBehavior.swift

import Foundation
import Quick
import Nimble
import RxSwift
import RxCocoa
import RxTest
import WebKit
@testable import RxWebKit

struct HasEventsBehaviorContext<T: Equatable> {
    let scheduler: TestScheduler
    let observable: Observable<T>
    let expected: T?
    init(_ scheduler: TestScheduler, _ observable: Observable<T>, _ expected: T?) {
        self.scheduler = scheduler
        self.observable = observable
        self.expected = expected
    }
}

class HasEventsBehavior<T: Equatable>: Quick.Behavior<HasEventsBehaviorContext<T>> {
    override class func spec(_ context: @escaping () -> HasEventsBehaviorContext<T>) {
        var scheduler: TestScheduler!
        var observable: Observable<T>!
        var expected: T?
        
        beforeEach {
            let cxt = context()
            scheduler = cxt.scheduler
            observable = cxt.observable
            expected = cxt.expected
        }
        
        afterEach {
            scheduler = nil
            observable = nil
        }
        
        describe("Has Events Behavior") {
            it("Actually got the event") {
                SharingScheduler.mock(scheduler: scheduler) {
                    let recorded = scheduler.record(source: observable)
                    scheduler.start()
                    expect(recorded.events.count).to(equal(1))
                    expect(recorded.events[0].value.element).to(equal(expected))
                }
            }
        }
    }
}

extension TestScheduler {
    /// Builds testable observer for s specific observable sequence, binds it's results and sets up disposal.
    /// parameter source: Observable sequence to observe.
    /// returns: Observer that records all events for observable sequence.
    func record<O: ObservableConvertibleType>(source: O) -> TestableObserver<O.Element> {
        let observer = self.createObserver(O.Element.self)
        let disposable = source.asObservable().bind(to: observer)
        self.scheduleAt(100000) {
            disposable.dispose()
        }
        return observer
    }
}