如何存储一个方法,以便在给出一个我不期望的值时,它返回一个默认值?
例如:
Map<String,String> map = mock(Map.class); when(map.get("abcd")).thenReturn("defg"); when(map.get("defg")).thenReturn("ghij"); when(map.get(anyString())).thenReturn("I don't know that string");
第2部分:如上所述但抛出异常:
Map<String,String> map = mock(Map.class); when(map.get("abcd")).thenReturn("defg"); when(map.get("defg")).thenReturn("ghij"); when(map.get(anyString())).thenThrow(new IllegalArgumentException("I don't know that string"));
在上述示例中,最后一个存根优先,因此映射将始终返回默认值.
解决方法
我找到的最好的解决方案是扭转存根的顺序:
Map<String,String> map = mock(Map.class); when(map.get(anyString())).thenReturn("I don't know that string"); when(map.get("abcd")).thenReturn("defg"); when(map.get("defg")).thenReturn("ghij");
当默认是抛出一个异常时,你可以使用doThrow和doReturn
doThrow(new RuntimeException()).when(map).get(anyString()); doReturn("defg").when(map).get("abcd"); doReturn("ghij").when(map).get("defg");
@L_403_0@