爲jUnit創建匹配程序<Stream>

Question

下面的代碼旨在允許有關流大小的更自然的斷言。

Matcher<Stream> hasCount(int count) { 
    return new TypeSafeDiagnosingMatcher<Stream>() { 
     protected boolean matchesSafely(Stream stream, Description desc) { 
      long streamCount = stream.count(); 
      if (streamCount == count) { 
       return true; 
      } else { 
       desc.appendText("count ").appendValue(streamCount); 
       return false; 
      } 
     } 

     public void describeTo(Description desc) { 
      desc.appendText("count ").appendValue(count); 
     } 
    }; 
} 

這樣的斷言，如：

assertThat(getWidgetStream(), hasCount(52)); 

它工作正常時，斷言道，但是當它失敗（如assertThat(Stream.empty(), hasCount(1));）則返回錯誤 「流已在運行或關閉」而不是預期的描述「預計：計數< 1>具有：計數< 0>」。

當我檢查TypeSafeDiagnosingMatcher的源時，我發現從matches和describeMismatch都調用matchesSafely。所以Hamcrest假設matchesSafely是冪等的，而我的不是。

有沒有辦法解決這個問題？

Answer 1

我發現的一個解決方案是在調用之間存儲結果。

return new TypeSafeDiagnosingMatcher<Stream>() { 
    private Optional<Long> streamCount = Optional.empty(); 

    protected boolean matchesSafely(Stream stream, Description desc) { 
     if (!streamCount.isPresent()) 
      streamCount = Optional.of(stream.count()); 
     if (streamCount.get() == count) { 
      return true; 
     } else { 
      desc.appendText("has count ").appendValue(streamCount.get().intValue()); 
      return false; 
     } 
    } 
}; 

這個工程，但不是特別優雅。