如何junit这个可观察的异步方法

Question

我有一个使用 java8 和响应式测试异步执行的存储库：

zadd 的接口：

Observable<Long> zadd(K key, double score, V member);



 public class TargetClass()
{
..
     public void executeMethod(List<String> input) {
            input.forEach(item -> redisConnection.reactive().zadd(...).subscribe(
                    result -> log.info("Success")
                    error -> log.error("Failed...")
            ));
        }
..
}

我的junit测试代码：

@Test
public void testMethod() {
  TargetClass targetClass=new TargetClass();
    targetClass.executeMethod(Arrays.asList("input1", "input2", "input3"));

    //as you can see I must put here Thread.sleep in order to let my execution to finish before continue since we have dependency on it

    try {
        Thread.sleep(4000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }

    //continue with testing..

如何修改我的代码部分权限。添加Thread.sleep让异步方法代码发生感觉不太对

谢谢， 射线。

Answer 1

如果我是你，我会通过添加 2 个新参数来重载方法 executeMethod，一个参数将是在 success 上执行的函数，另一个是在 failure 上执行的函数。我将创建新方法protected 而不是public，我将在我的单元测试中测试这个新方法而不是另一个。你的public 方法可以用这个executeMethod(input, result -> log.info("Success"), error -> log.error("Failed...")) 调用新的方法。在我的单元测试中，我可以提供一种方法来获得这样的结果：

@Test
public void testMethod() {
    TargetClass targetClass = new TargetClass();
    AtomicBoolean success = new AtomicBoolean();
    synchronized (success) {
        targetClass.executeMethod(
            list, 
            result -> {
                synchronized (success) {
                    success.set(true);
                    success.notify();
                } 
            }, 
            error -> {
                synchronized (success) {
                    success.set(false);
                    success.notify();
                } 
            }
        );
        success.wait();
    }
    // The rest of your unit test here
}