用Mockito模拟一个带有对象参数的方法

Question

在我的单元测试中，我想通过执行以下来模拟与弹性搜索的交互

when(cityDefinitionRepository.findCitiesNearby(geoPoint, SOURCE, 2)).thenReturn(cityDefinitionsArrival);
when(cityDefinitionRepository.findCitiesNearby(geoPoint2, SOURCE, 2)).thenReturn(cityDefinitionsDeparture);
SearchResult results = benerailService.doSearch(interpretation, 2, false);

doSearch方法包含

departureCityDefinitions = cityDefinitionRepository.findCitiesNearby(geo, SOURCE, distance);

当我调试我的代码时，我看到在我的doSearch方法中调用了mockito，但它没有返回cityDefinitionsDarrival对象。这可能是因为geoPoint和geo是两个不同的对象。

地理点和地理对象都是弹性搜索的地理点，包含相同的纬度和经度。

我通过成功地实现了这一点

when(cityDefinitionRepository.findCitiesNearby(any(geoPoint.getClass()), eq(SOURCE), eq(2))).thenReturn(cityDefinitionsArrival);
when(cityDefinitionRepository.findCitiesNearby(any(geoPoint2.getClass()), eq(SOURCE), eq(2))).thenReturn(cityDefinitionsDeparture);

但现在它忽略了我的纬度和经度值，并接受GeoPoint类的任何对象。这是一个问题，因为在我的doSearch方法中，我有两种findCitiesNear的用法，每种用法都有不同的纬度和经度，我需要分别模拟它们。

莫基托可能这样吗？

城市定义到达和城市定义离开都是ArrayLists，SOURCE是一个字符串值，并且geo和geoPoint对象：

GeoPoint geoPoint = new GeoPoint(50.850449999999995, 4.34878);
GeoPoint geoPoint2 = new GeoPoint(48.861710, 2.348923);
double lat = 50.850449999999995;
double lon = 4.34878;
GeoPoint geo = new GeoPoint(lat, lon);
double lat2 = 48.861710;
double lon2 = 2.348923;
GeoPoint geo2 = new GeoPoint(lat2, lon2);

Answer 1

使用argThat:

public final class IsSameLatLong extends ArgumentMatcher<GeoPoint> {
  private final GeoPoint as;
  public IsSameLatLong(GeoPoint as) {
      this.as = as;
  }
  //some sensible value, like 1000th of a second i.e. 0° 0' 0.001"
  private final static double EPSILON = 1.0/(60*60*1000); 
  private static boolean closeEnough(double a, double b) {
     return Math.abs(a - b) < EPSILON;
  }
  public boolean matches(Object point) {
      GeoPoint other = (GeoPoint) point;
      if (other == null) return false;
      return closeEnough(other.getLat(), as.getLat()) &&
             closeEnough(other.getLong(), as.getLong());
  }
}

然后这样使用：

when(cityDefinitionRepository.findCitiesNearby(argThat(new IsSameLatLong(geoPoint)), eq(SOURCE), eq(2))).thenReturn(cityDefinitionsArrival);
when(cityDefinitionRepository.findCitiesNearby(argThat(new IsSameLatLong(geoPoint2)), eq(SOURCE), eq(2))).thenReturn(cityDefinitionsDeparture);