grpc-swift/Tests/gRPCTests/GRPCTests.swift

import XCTest
import Foundation
import Dispatch
@testable import gRPC

func Log(_ message : String) {
  FileHandle.standardError.write((message + "\n").data(using:.utf8)!)
}

class gRPCTests: XCTestCase {
  var count : Int = 0

  func testBasicSanity() {
    gRPC.initialize()
    let done = NSCondition()
    self.count = 2
    DispatchQueue.global().async() {
      server()
      Log("server finished")
      done.lock()
      self.count = self.count - 1
      done.signal()
      done.unlock()
    }
    DispatchQueue.global().async() {
      client()
      Log("client finished")
      done.lock()
      self.count = self.count - 1
      done.signal()
      done.unlock()
    }
    var running = true
    while (running) {
      Log("waiting")
      done.lock()
      done.wait()
      if (self.count == 0) {
        running = false
      }
      Log("count \(self.count)")
      done.unlock()
    }
  }
}

extension gRPCTests {
  static var allTests : [(String, (gRPCTests) -> () throws -> Void)] {
    return [
      ("testBasicSanity", testBasicSanity),
    ]
  }
}

let address = "localhost:8999"
let host = "foo.test.google.fr"
let clientText = "hello, server!"
let serverText = "hello, client!"
let initialClientMetadata =
  ["x": "xylophone",
   "y": "yu",
   "z": "zither"]
let initialServerMetadata =
  ["a": "Apple",
   "b": "Banana",
   "c": "Cherry"]
let trailingServerMetadata =
  ["0": "zero",
   "1": "one",
   "2": "two"]
let steps = 30
let hello = "/hello"
let goodbye = "/goodbye"
let statusCode = 0
let statusMessage = "OK"

func verify_metadata(_ metadata: Metadata, expected: [String:String]) {
  XCTAssertGreaterThanOrEqual(metadata.count(), expected.count)
  for i in 0..<metadata.count() {
    if expected[metadata.key(i)] != nil {
      XCTAssertEqual(metadata.value(i), expected[metadata.key(i)])
    }
  }
}

func client() {
  let message = clientText.data(using: .utf8)
  let c = gRPC.Channel(address:address)
  c.host = host
  let done = NSCondition()
  var running = true
  for i in 0..<steps {
    var call : Call
    do {
      let method = (i < steps-1) ? hello : goodbye
      call = c.makeCall(method)
      let metadata = Metadata(initialClientMetadata)
      try call.start(.unary, metadata:metadata, message:message) {
        (response) in
        // verify the basic response from the server
        XCTAssertEqual(response.statusCode, statusCode)
        XCTAssertEqual(response.statusMessage, statusMessage)
        // verify the message from the server
        let resultData = response.resultData
        let messageString = String(data: resultData!, encoding: .utf8)
        XCTAssertEqual(messageString, serverText)
        // verify the initial metadata from the server
        let initialMetadata = response.initialMetadata!
        verify_metadata(initialMetadata, expected: initialServerMetadata)
        // verify the trailing metadata from the server
        let trailingMetadata = response.trailingMetadata!
        verify_metadata(trailingMetadata, expected: trailingServerMetadata)
        done.lock()
        running = false
        done.signal()
        done.unlock()
      }
    } catch (let error) {
        XCTFail("error \(error)")
    }
    // wait for the call to complete
    var finished = false
    while (!finished) {
      done.lock()
      done.wait()
      if (!running) {
        finished = true
      }
      done.unlock()
    }
    Log("finished client call \(i)")
  }
  Log("client done")
}

func server() {
  let server = gRPC.Server(address:address)
  var requestCount = 0
  let done = NSCondition()
  var running = true
  server.run() {(requestHandler) in
    do {
      requestCount += 1
      XCTAssertEqual(requestHandler.host, host)
      if (requestCount < steps) {
        XCTAssertEqual(requestHandler.method, hello)
      } else {
        XCTAssertEqual(requestHandler.method, goodbye)
      }
      let initialMetadata = requestHandler.requestMetadata
      verify_metadata(initialMetadata, expected: initialClientMetadata)
      let initialMetadataToSend = Metadata(initialServerMetadata)
      try requestHandler.receiveMessage(initialMetadata:initialMetadataToSend)
      {(messageData) in
        let messageString = String(data: messageData!, encoding: .utf8)
        XCTAssertEqual(messageString, clientText)
      }
      if requestHandler.method == goodbye {
        server.stop()
      }
      let replyMessage = serverText
      let trailingMetadataToSend = Metadata(trailingServerMetadata)
      try requestHandler.sendResponse(message:replyMessage.data(using: .utf8)!,
                                      statusCode:statusCode,
                                      statusMessage:statusMessage,
                                      trailingMetadata:trailingMetadataToSend)
    } catch (let error) {
      XCTFail("error \(error)")
    }
  }
  server.onCompletion() {
      // exit the server thread
      Log("signaling completion")
      done.lock()
      running = false
      done.signal()
      done.unlock()
  }
  // wait for the server to exit
  var finished = false
  while !finished {
    done.lock()
    done.wait()
    if (!running) {
      finished = true
    }
    done.unlock()
  }
  Log("server done")
}