Merge pull request #8118 from atmire/w2p-85525_pr-use-lenient-stubbing-in-MockSolrLoggerServiceImpl

Use lenient stubbing in MockSolrLoggerServiceImpl
This commit is contained in:
Tim Donohue
2022-01-18 11:36:51 -06:00
committed by GitHub

View File

@@ -9,7 +9,6 @@ package org.dspace.statistics;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.net.InetAddress; import java.net.InetAddress;
import java.util.ArrayList; import java.util.ArrayList;
@@ -29,6 +28,7 @@ import com.maxmind.geoip2.record.Postal;
import com.maxmind.geoip2.record.RepresentedCountry; import com.maxmind.geoip2.record.RepresentedCountry;
import com.maxmind.geoip2.record.Traits; import com.maxmind.geoip2.record.Traits;
import org.dspace.solr.MockSolrServer; import org.dspace.solr.MockSolrServer;
import org.mockito.Mockito;
import org.springframework.beans.factory.DisposableBean; import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean; import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -55,7 +55,7 @@ public class MockSolrLoggerServiceImpl
// Mock GeoIP's DatabaseReader // Mock GeoIP's DatabaseReader
DatabaseReader reader = mock(DatabaseReader.class); DatabaseReader reader = mock(DatabaseReader.class);
// Ensure that any tests requesting a city() get a mock/fake CityResponse // Ensure that any tests requesting a city() get a mock/fake CityResponse
when(reader.city(any(InetAddress.class))).thenReturn(mockCityResponse()); Mockito.lenient().when(reader.city(any(InetAddress.class))).thenReturn(mockCityResponse());
// Save this mock DatabaseReader to be used by SolrLoggerService // Save this mock DatabaseReader to be used by SolrLoggerService
locationService = reader; locationService = reader;
} }