test: UNKNOWN or UNAVAILABLE (#1860)

akka · Oct 17, 2023 · b569097 · b569097
1 parent 3a18a25
commit b569097
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/interop-tests/src/test/scala/akka/grpc/scaladsl/LoadBalancingIntegrationSpec.scala b/interop-tests/src/test/scala/akka/grpc/scaladsl/LoadBalancingIntegrationSpec.scala
@@ -154,7 +154,7 @@ class LoadBalancingIntegrationSpec(backend: String, config: Config = ConfigFacto
         client.sayHello(HelloRequest(s"Hello friend")).failed.futureValue.asInstanceOf[StatusRuntimeException]
       // FIXME issue #1857, not sure how this is supposed to be
       if (backend == "netty")
-        failure.getStatus.getCode should be(Code.UNKNOWN)
+        failure.getStatus.getCode should (equal(Code.UNKNOWN).or(equal(Code.UNAVAILABLE)))
       else
         failure.getStatus.getCode should be(Code.UNAVAILABLE)
 
@@ -177,7 +177,7 @@ class LoadBalancingIntegrationSpec(backend: String, config: Config = ConfigFacto
         client.sayHello(HelloRequest(s"Hello friend")).failed.futureValue.asInstanceOf[StatusRuntimeException]
       // FIXME issue #1857, not sure how this is supposed to be
       if (backend == "netty")
-        failure.getStatus.getCode should be(Code.UNKNOWN)
+        failure.getStatus.getCode should (equal(Code.UNKNOWN).or(equal(Code.UNAVAILABLE)))
       else
         failure.getStatus.getCode should be(Code.UNAVAILABLE)
 

diff --git a/interop-tests/src/test/scala/akka/grpc/scaladsl/NonBalancingIntegrationSpec.scala b/interop-tests/src/test/scala/akka/grpc/scaladsl/NonBalancingIntegrationSpec.scala
@@ -177,7 +177,7 @@ class NonBalancingIntegrationSpec(backend: String)
         client.sayHello(HelloRequest(s"Hello friend")).failed.futureValue.asInstanceOf[StatusRuntimeException]
       // FIXME issue #1857, not sure how this is supposed to be
       if (backend == "netty")
-        failure.getStatus.getCode should be(Code.UNKNOWN)
+        failure.getStatus.getCode should (equal(Code.UNKNOWN).or(equal(Code.UNAVAILABLE)))
       else {
         failure.getStatus.getCode should be(Code.UNAVAILABLE)
         client.closed.failed.futureValue shouldBe a[ClientConnectionException]