vimr/RxPack/RxMsgpackRpcTests/RxMsgpackRpcTests.swift

/**
 * Tae Won Ha - http://taewon.de - @hataewon
 * See LICENSE
 */

import XCTest
import RxSwift

class RxMsgpackRpcTests: XCTestCase {

  private var connection: RxMsgpackRpc!
  private let disposeBag = DisposeBag()

  override func setUp() {
    super.setUp()

    // $ NVIM_LISTEN_ADDRESS=/tmp/nvim.sock nvim --headless $SOMEFILE
    connection = RxMsgpackRpc()
    connection.stream
      .subscribe(onNext: { msg in
        switch msg {
        case let .notification(method, params):
          print("NOTIFICATION: \(method): array of \(params.count) elements")
        case let .error(value, msg):
          print("ERROR: \(msg) with \(value)")
        default:
          print("???")
        }
      }, onError: { print("ERROR: \($0)") })
      .disposed(by: self.disposeBag)

    _ = connection.run(at: "/tmp/nvim.sock")
      .andThen(connection.request(
        method: "nvim_ui_attach",
        params: [.int(40), .int(40), .map([:])],
        expectsReturnValue: true
      ))
      .syncValue()
  }

  override func tearDown() {
    super.tearDown()
    try? self.connection
      .request(
        method: "nvim_command", params: [.string("q!")],
        expectsReturnValue: false
      )
      .asCompletable()
      .wait()
    try? self.connection.stop().wait()
  }

  func testExample() {
    let disposeBag = DisposeBag()

    let lineCount = connection
      .request(
        method: "nvim_buf_line_count",
        params: [.int(0)],
        expectsReturnValue: true
      )
      .syncValue()
    print(lineCount ?? "???")

    let formatter = DateFormatter()
    formatter.dateFormat = "mm:ss.SSS"
    for i in 0...100 {
      let date = Date()
      connection
        .request(
          method: "nvim_command_output",
          params: [.string("echo '\(i) \(formatter.string(from: date))'")],
          expectsReturnValue: true
        )
        .subscribe(
          onSuccess: { response in print(response) },
          onError: { error in print(error) }
        )
        .disposed(by: disposeBag)
    }

    sleep(1)
  }
}