單元測試聲明TreeMap中的對象的順序是正確的？

Question

我想寫一個Junit測試來斷言我的TreeMap中的對象是的正確順序。

我該怎麼做？

我努力做到以下幾點：

assertEquals(treeMapFromMethod.get(0), ValueThatShouldBeFirstEntryInTreeMap); 

但我得到錯誤：

"java.lang.ClassCastException: java.lang.Integer cannot be cast to MyObJect" 

此錯誤與上面的線。

我該如何解決這個問題/是否有更好的方法來測試？

Answer 1

如果使用Hamcrest您可以用contains匹配很容易做到這一點：

TreeMap<String, Integer> map = new TreeMap<>(); 
map.put("one", 1); 
map.put("two", 2); 
map.put("three", 3); 

Assert.assertThat(map.keySet(), Matchers.contains("one", "three", "two")); 

的contains匹配檢查，如果給定的集合具有相同的長度，幷包含以相同的順序給定的項目。