 extension BellerophonTests: BellerophonManagerDelegate {
 
     func bellerophonStatus(manager: BellerophonManager, completion: (status: BellerophonObservable?, error: NSError?) -> ()) {
+        completion(status: responseArray[currentIdx], error: nil)
+    }
 
-        switch currentIdx {
-        case 0b00:
-            completion(status: statusArray[0b00], error: nil)
-            XCTAssertEqual(BellerophonManager.sharedInstance.killSwitchWindow.hidden, true)
-        case 0b01:
-            completion(status: statusArray[0b01], error: nil)
-            XCTAssertEqual(shouldForceUpdateIsCalled, true)
-        case 0b10:
-            completion(status: statusArray[0b10], error: nil)
-            XCTAssertEqual(willEngageIsCalled, true)
-        case 0b11:
-            completion(status: statusArray[0b11], error: nil)
-            XCTAssertEqual(willEngageIsCalled, true)
-            XCTAssertEqual(shouldForceUpdateIsCalled, false)
-        default:
-            break
-        }
-    }
-    
     func shouldForceUpdate() {
         shouldForceUpdateIsCalled = true
     }

    }
}

class BellerophonModel: BellerophonObservable {

    init(isAPIInactive: Bool, shouldForceUpdate: Bool, interval: NSTimeInterval, userMessageStr: String) {
        self.isAPIInactive = isAPIInactive

@testable import Bellerophon

class BellerophonTests: XCTestCase {

    enum ResponseCases: Int {
        case KillSwitchOffForceUpdateOff = 0b00
        case KillSwitchOffForceUpdateOn = 0b01
        case KillSwitchOnForceUpdateOff = 0b10
        case KillSwitchOnForceUpdateOn = 0b11
    }

    var responseArray: [BellerophonResponse]!

    var willEngageIsCalled: Bool!
    var willDisengageIsCalled: Bool!
