cc.otavia.testkit.TestProbe.scala Maven / Gradle / Ivy
/*
* Copyright 2022 Yan Kun
*
* This file fork from netty.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cc.otavia.testkit
import cc.otavia.core.address.Address
import cc.otavia.core.message.{Ask, Reply, ReplyOf}
import cc.otavia.core.stack.MessageFuture
import cc.otavia.core.system.ActorSystem
import cc.otavia.core.testkit.ProbeActor
import scala.concurrent.duration.*
import scala.concurrent.{Await, Promise}
import scala.reflect.ClassTag
class TestProbe(system: ActorSystem) {
def askAndExpect[M <: Ask[? <: Reply], R <: ReplyOf[M]: ClassTag](
actor: Address[M],
ask: M,
expect: MessageFuture[R] => Boolean
): Boolean = {
val result = Promise.apply[Boolean]()
system.buildActor(() => new ProbeActor(actor, ask, expect, result))
Await.result(result.future, Duration.Inf)
}
}