sbt/server-test/src/test/scala/testpkg/ResponseTest.scala

/*
 * sbt
 * Copyright 2011 - 2018, Lightbend, Inc.
 * Copyright 2008 - 2010, Mark Harrah
 * Licensed under Apache License 2.0 (see LICENSE)
 */

package testpkg

import scala.concurrent.duration.*

// starts svr using server-test/response and perform custom server tests
class ResponseTest extends AbstractServerTest {
  override val testDirectory: String = "response"

  test("response from a command") {
    svr.sendJsonRpc("""{ "jsonrpc": "2.0", "id": "10", "method": "foo/export", "params": {} }""")
    assert(svr.waitForString(10.seconds) { s =>
      if (!s.contains("systemOut"))
        println(s)
      s.contains(""""id":"10"""") &&
      s.contains("scala-library-2.12.21.jar")
    })
  }

  test("response from a task") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "11", "method": "foo/rootClasspath", "params": {} }"""
    )
    assert(svr.waitForString(10.seconds) { s =>
      if (!s.contains("systemOut")) println(s)
      s.contains(""""id":"11"""") &&
      s.contains("scala-library-2.12.21.jar")
    })
  }

  test("a command failure") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "12", "method": "foo/fail", "params": {} }"""
    )
    assert(svr.waitForString(10.seconds) { s =>
      if (!s.contains("systemOut")) println(s)
      s.contains(""""error":{"code":-33000,"message":"fail message"""")
    })
  }

  test("a command failure with custom code") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "13", "method": "foo/customfail", "params": {} }"""
    )
    assert(svr.waitForString(10.seconds) { s =>
      if (!s.contains("systemOut")) println(s)
      s.contains(""""error":{"code":500,"message":"some error"""")
    })
  }

  test("a command with a notification") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "14", "method": "foo/notification", "params": {} }"""
    )
    assert(svr.waitForString(10.seconds) { s =>
      if (!s.contains("systemOut")) println(s)
      s.contains("""{"jsonrpc":"2.0","method":"foo/something","params":"something"}""")
    })
  }

  test("respond concurrently from a task and the handler") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "15", "method": "foo/respondTwice", "params": {} }"""
    )
    assert {
      svr.waitForString(10.seconds) { s =>
        if (!s.contains("systemOut")) println(s)
        s.contains("\"id\":\"15\"")
      }
    }
    assert {
      // the second response should never be sent
      svr.neverReceive(500.milliseconds) { s =>
        if (!s.contains("systemOut")) println(s)
        s.contains("\"id\":\"15\"")
      }
    }
  }

  test("concurrent result and error") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "id": "16", "method": "foo/resultAndError", "params": {} }"""
    )
    assert {
      svr.waitForString(10.seconds) { s =>
        if (!s.contains("systemOut")) println(s)
        s.contains("\"id\":\"16\"")
      }
    }
    assert {
      // the second response (result or error) should never be sent
      svr.neverReceive(500.milliseconds) { s =>
        if (!s.contains("systemOut")) println(s)
        s.contains("\"id\":\"16\"")
      }
    }
  }

  test("response to a notification should not be sent") {
    svr.sendJsonRpc(
      """{ "jsonrpc": "2.0", "method": "foo/customNotification", "params": {} }"""
    )
    assert {
      svr.neverReceive(500.milliseconds) { s =>
        if (!s.contains("systemOut")) println(s)
        s.contains("\"result\":\"notification result\"")
      }
    }
  }
}