Skip to content
Snippets Groups Projects

Policy as a feature

Merged Policy as a feature
feature-policy into master
Merged Riabokon Stanislav(EPAM)[GCP] requested to merge feature-policy into master
Compare
and
Show latest version
4 files
+ 154
− 176
Compare changes
  • Side-by-side
  • Inline
Files
4
search-core/src/test/java/org/opengroup/osdu/search/provider/impl/ScrollCoreQueryServiceImplIPolicyServiceTest.java
+ 150
− 173
@@ -60,8 +60,8 @@ import org.opengroup.osdu.search.util.CrossTenantUtils;
import org.opengroup.osdu.search.util.ElasticClientHandler;
import org.opengroup.osdu.search.util.GeoQueryBuilder;
import org.opengroup.osdu.search.util.IDetailedBadRequestMessageUtil;
import org.opengroup.osdu.search.util.IQueryPerformanceLogger;
import org.opengroup.osdu.search.util.IQueryParserUtil;
import org.opengroup.osdu.search.util.IQueryPerformanceLogger;
import org.opengroup.osdu.search.util.ISortParserUtil;
import org.opengroup.osdu.search.util.ResponseExceptionParser;
import org.opengroup.osdu.search.util.SuggestionsQueryUtil;
@@ -69,176 +69,153 @@ import org.opengroup.osdu.search.util.SuggestionsQueryUtil;
@RunWith(MockitoJUnitRunner.class)
public class ScrollCoreQueryServiceImplIPolicyServiceTest {
private final String DATA_GROUPS = "X-Data-Groups";
private final String DATA_GROUP_1 = "data.welldb.viewers@common.evd.cloud.slb-ds.com";
private final String DATA_GROUP_2 = "data.npd.viewers@common.evd.cloud.slb-ds.com";
private final String PARTITION_ID = "opendes";
@InjectMocks
ScrollCoreQueryServiceImpl scrollQueryServiceAws;
@Mock
private ElasticClientHandler elasticClientHandler;
@Mock
private JaxRsDpsLog log;
@Mock
private IProviderHeaderService providerHeaderService;
@Mock
private CrossTenantUtils crossTenantUtils;
@Mock
private IFieldMappingTypeService fieldMappingTypeService;
@Mock
private IQueryParserUtil queryParserUtil;
@Mock
private ISortParserUtil sortParserUtil;
@Mock
private IDetailedBadRequestMessageUtil detailedBadRequestMessageUtil;
@Mock
private DpsHeaders dpsHeaders;
@Mock
private AuditLogger auditLogger;
@Mock
private CursorCache cursorCache;
@Mock
private ResponseExceptionParser exceptionParser;
@Mock
private IPolicyService iPolicyService;
@Mock
private GeoQueryBuilder geoQueryBuilder;
@Mock
private ElasticLoggingConfig elasticLoggingConfig;
@Mock
private IQueryPerformanceLogger searchDependencyLogger;
@Mock
private SuggestionsQueryUtil suggestionsQueryUtil;
@Mock
private IFeatureFlag featureFlag;
@Before
public void setup() {
MockitoAnnotations.openMocks(this);
Map<String, String> HEADERS = new HashMap<>();
HEADERS.put(DpsHeaders.ACCOUNT_ID, "tenant1");
HEADERS.put(DpsHeaders.AUTHORIZATION, "Bearer blah");
HEADERS.put(DATA_GROUPS, String.format("%s,%s", DATA_GROUP_1, DATA_GROUP_2));
when(elasticLoggingConfig.getEnabled()).thenReturn(false);
when(elasticLoggingConfig.getThreshold()).thenReturn(200L);
}
@Test
public void should_return_CorrectQueryResponse_NullResult_noCursorSet_QueryBuilder() throws Exception {
// arrange
// create query request according to this example query:
// {
// "kind": "osdu:wks:reference-data--CoordinateTransformation:1.0.0",
// "query": "data.ID:\"EPSG::1078\"",
// "spatialFilter": {
// "field": "data.Wgs84Coordinates",
// "byIntersection": {
// "polygons": [
// {
// "points": [
// {
// "latitude": 10.75,
// "longitude": -8.61
// }
// ]
// }
// ]
// }
// }
// }
CursorQueryRequest queryRequest = new CursorQueryRequest();
queryRequest.setQuery("data.ID:\"EPSG::1078\"");
SpatialFilter spatialFilter = new SpatialFilter();
spatialFilter.setField("data.Wgs84Coordinates");
SpatialFilter.ByIntersection byIntersection = new SpatialFilter.ByIntersection();
Polygon polygon = new Polygon();
Point point = new Point(1.02, -8.61);
Point point1 = new Point(1.02, -2.48);
Point point2 = new Point(10.74, -2.48);
Point point3 = new Point(10.74, -8.61);
Point point4 = new Point(1.02, -8.61);
List<Point> points = new ArrayList<>();
points.add(point);
points.add(point1);
points.add(point2);
points.add(point3);
points.add(point4);
polygon.setPoints(points);
List<Polygon> polygons = new ArrayList<>();
polygons.add(polygon);
byIntersection.setPolygons(polygons);
spatialFilter.setByIntersection(byIntersection);
queryRequest.setSpatialFilter(spatialFilter);
// mock out elastic client handler
RestHighLevelClient client = Mockito.mock(RestHighLevelClient.class, Mockito.RETURNS_DEEP_STUBS);
SearchResponse searchResponse = Mockito.mock(SearchResponse.class);
QueryBuilder textQueryBuilder = Mockito.mock(QueryBuilder.class);
when(iPolicyService.getCompiledPolicy(any())).thenReturn("PolicyString");
when(featureFlag.isFeatureEnabled(POLICY_FEATURE_NAME)).thenReturn(true);
when(queryParserUtil.buildQueryBuilderFromQueryString(anyString())).thenReturn(textQueryBuilder);
when(searchResponse.status())
.thenReturn(RestStatus.OK);
SearchHits searchHits = Mockito.mock(SearchHits.class);
when(searchHits.getHits())
.thenReturn(new SearchHit[]{});
when(searchResponse.getHits())
.thenReturn(searchHits);
when(client.search(Mockito.any(SearchRequest.class), Mockito.eq(RequestOptions.DEFAULT)))
.thenReturn(searchResponse);
when(elasticClientHandler.createRestClient())
.thenReturn(client);
String index = "some-index";
when(crossTenantUtils.getIndexName(Mockito.any()))
.thenReturn(index);
Set<String> indexedTypes = new HashSet<>();
indexedTypes.add("geo_shape");
Map<String, String> headers = new HashMap<>();
headers.put("groups", "[]");
String expectedSource = "{\"size\":10,\"timeout\":\"1m\",\"query\":{\"bool\":{\"must\":[{\"wrapper\":{\"query\":\"UG9saWN5U3RyaW5n\"}}],\"adjust_pure_negative\":true,\"boost\":1.0}},\"_source\":{\"includes\":[],\"excludes\":[\"x-acl\",\"index\"]},\"sort\":[{\"_score\":{\"order\":\"desc\"}},{\"_doc\":{\"order\":\"asc\"}}],\"highlight\":{}}";
// act
scrollQueryServiceAws.queryIndex(queryRequest);
// assert
ArgumentCaptor<SearchRequest> searchRequestArg = ArgumentCaptor.forClass(SearchRequest.class);
Mockito.verify(client, Mockito.times(1)).search(searchRequestArg.capture(), Mockito.any());
SearchRequest searchRequest = searchRequestArg.getValue();
String actualSource = searchRequest.source().toString();
assertEquals(expectedSource, actualSource);
}
private final String DATA_GROUPS = "X-Data-Groups";
private final String DATA_GROUP_1 = "data.welldb.viewers@common.evd.cloud.slb-ds.com";
private final String DATA_GROUP_2 = "data.npd.viewers@common.evd.cloud.slb-ds.com";
private final String PARTITION_ID = "opendes";
@InjectMocks ScrollCoreQueryServiceImpl scrollQueryServiceAws;
@Mock private ElasticClientHandler elasticClientHandler;
@Mock private JaxRsDpsLog log;
@Mock private IProviderHeaderService providerHeaderService;
@Mock private CrossTenantUtils crossTenantUtils;
@Mock private IFieldMappingTypeService fieldMappingTypeService;
@Mock private IQueryParserUtil queryParserUtil;
@Mock private ISortParserUtil sortParserUtil;
@Mock private IDetailedBadRequestMessageUtil detailedBadRequestMessageUtil;
@Mock private DpsHeaders dpsHeaders;
@Mock private AuditLogger auditLogger;
@Mock private CursorCache cursorCache;
@Mock private ResponseExceptionParser exceptionParser;
@Mock private IPolicyService iPolicyService;
@Mock private GeoQueryBuilder geoQueryBuilder;
@Mock private ElasticLoggingConfig elasticLoggingConfig;
@Mock private IQueryPerformanceLogger searchDependencyLogger;
@Mock private SuggestionsQueryUtil suggestionsQueryUtil;
@Mock public IFeatureFlag iFeatureFlag;
@Before
public void setup() {
MockitoAnnotations.openMocks(this);
Map<String, String> HEADERS = new HashMap<>();
HEADERS.put(DpsHeaders.ACCOUNT_ID, "tenant1");
HEADERS.put(DpsHeaders.AUTHORIZATION, "Bearer blah");
HEADERS.put(DATA_GROUPS, String.format("%s,%s", DATA_GROUP_1, DATA_GROUP_2));
when(elasticLoggingConfig.getEnabled()).thenReturn(false);
when(elasticLoggingConfig.getThreshold()).thenReturn(200L);
}
@Test
public void should_return_CorrectQueryResponse_NullResult_noCursorSet_QueryBuilder()
throws Exception {
// arrange
// create query request according to this example query:
// {
// "kind": "osdu:wks:reference-data--CoordinateTransformation:1.0.0",
// "query": "data.ID:\"EPSG::1078\"",
// "spatialFilter": {
// "field": "data.Wgs84Coordinates",
// "byIntersection": {
// "polygons": [
// {
// "points": [
// {
// "latitude": 10.75,
// "longitude": -8.61
// }
// ]
// }
// ]
// }
// }
// }
CursorQueryRequest queryRequest = new CursorQueryRequest();
queryRequest.setQuery("data.ID:\"EPSG::1078\"");
SpatialFilter spatialFilter = new SpatialFilter();
spatialFilter.setField("data.Wgs84Coordinates");
SpatialFilter.ByIntersection byIntersection = new SpatialFilter.ByIntersection();
Polygon polygon = new Polygon();
Point point = new Point(1.02, -8.61);
Point point1 = new Point(1.02, -2.48);
Point point2 = new Point(10.74, -2.48);
Point point3 = new Point(10.74, -8.61);
Point point4 = new Point(1.02, -8.61);
List<Point> points = new ArrayList<>();
points.add(point);
points.add(point1);
points.add(point2);
points.add(point3);
points.add(point4);
polygon.setPoints(points);
List<Polygon> polygons = new ArrayList<>();
polygons.add(polygon);
byIntersection.setPolygons(polygons);
spatialFilter.setByIntersection(byIntersection);
queryRequest.setSpatialFilter(spatialFilter);
// mock out elastic client handler
RestHighLevelClient client =
Mockito.mock(RestHighLevelClient.class, Mockito.RETURNS_DEEP_STUBS);
SearchResponse searchResponse = Mockito.mock(SearchResponse.class);
QueryBuilder textQueryBuilder = Mockito.mock(QueryBuilder.class);
when(iPolicyService.getCompiledPolicy(any())).thenReturn("PolicyString");
when(iFeatureFlag.isFeatureEnabled(POLICY_FEATURE_NAME)).thenReturn(true);
when(queryParserUtil.buildQueryBuilderFromQueryString(anyString()))
.thenReturn(textQueryBuilder);
when(searchResponse.status()).thenReturn(RestStatus.OK);
SearchHits searchHits = Mockito.mock(SearchHits.class);
when(searchHits.getHits()).thenReturn(new SearchHit[] {});
when(searchResponse.getHits()).thenReturn(searchHits);
when(client.search(Mockito.any(SearchRequest.class), Mockito.eq(RequestOptions.DEFAULT)))
.thenReturn(searchResponse);
when(elasticClientHandler.createRestClient()).thenReturn(client);
String index = "some-index";
when(crossTenantUtils.getIndexName(Mockito.any())).thenReturn(index);
Set<String> indexedTypes = new HashSet<>();
indexedTypes.add("geo_shape");
Map<String, String> headers = new HashMap<>();
headers.put("groups", "[]");
String expectedSource =
"{\"size\":10,\"timeout\":\"1m\",\"query\":{\"bool\":{\"must\":[{\"wrapper\":{\"query\":\"UG9saWN5U3RyaW5n\"}}],\"adjust_pure_negative\":true,\"boost\":1.0}},\"_source\":{\"includes\":[],\"excludes\":[\"x-acl\",\"index\"]},\"sort\":[{\"_score\":{\"order\":\"desc\"}},{\"_doc\":{\"order\":\"asc\"}}],\"highlight\":{}}";
// act
scrollQueryServiceAws.queryIndex(queryRequest);
// assert
ArgumentCaptor<SearchRequest> searchRequestArg = ArgumentCaptor.forClass(SearchRequest.class);
Mockito.verify(client, Mockito.times(1)).search(searchRequestArg.capture(), Mockito.any());
SearchRequest searchRequest = searchRequestArg.getValue();
String actualSource = searchRequest.source().toString();
assertEquals(expectedSource, actualSource);
}
}