private AtlasTypesDef getTypeDefs(String ... typeNames) throws AtlasServiceException { final AtlasTypesDef typeDefs = new AtlasTypesDef(); for (int i = 0; i < typeNames.length; i++) { final MultivaluedMap<String, String> searchParams = new MultivaluedMapImpl(); searchParams.add(SearchFilter.PARAM_NAME, typeNames[i]); final AtlasTypesDef typeDef = atlasClient.getAllTypeDefs(new SearchFilter(searchParams)); typeDefs.getEntityDefs().addAll(typeDef.getEntityDefs()); } logger.debug("typeDefs={}", typeDefs); return typeDefs; }
/** * Populate a SearchFilter on the basis of the Query Parameters * @return */ private SearchFilter getSearchFilter(HttpServletRequest httpServletRequest) { SearchFilter ret = new SearchFilter(); Set<String> keySet = httpServletRequest.getParameterMap().keySet(); for (String k : keySet) { String key = String.valueOf(k); String value = String.valueOf(httpServletRequest.getParameter(k)); if (key.equalsIgnoreCase("excludeInternalTypesAndReferences") && value.equalsIgnoreCase("true")) { FilterUtil.addParamsToHideInternalType(ret); } else { ret.setParam(key, value); } } return ret; } }
private void verifyTypesCreated() throws Exception { MultivaluedMap<String, String> searchParams = new MultivaluedMapImpl(); for (String typeName : TYPES) { searchParams.clear(); searchParams.add(SearchFilter.PARAM_NAME, typeName); SearchFilter searchFilter = new SearchFilter(searchParams); AtlasTypesDef searchDefs = atlasClientV2.getAllTypeDefs(searchFilter); assert (!searchDefs.isEmpty()); System.out.println("Created type [" + typeName + "]"); } }
private AtlasTypesDef getTypeDefs(String ... typeNames) throws AtlasServiceException { final AtlasTypesDef typeDefs = new AtlasTypesDef(); for (int i = 0; i < typeNames.length; i++) { final MultivaluedMap<String, String> searchParams = new MultivaluedMapImpl(); searchParams.add(SearchFilter.PARAM_NAME, typeNames[i]); final AtlasTypesDef typeDef = atlasClient.getAllTypeDefs(new SearchFilter(searchParams)); typeDefs.getEntityDefs().addAll(typeDef.getEntityDefs()); } logger.debug("typeDefs={}", typeDefs); return typeDefs; }
@Test public void filterInternalType() throws AtlasBaseException { SearchFilter searchFilter = new SearchFilter(); FilterUtil.addParamsToHideInternalType(searchFilter); AtlasTypesDef filteredTypeDefs = typeDefStore.searchTypesDef(searchFilter); assertNotNull(filteredTypeDefs); Optional<AtlasEntityDef> anyInternal = filteredTypeDefs.getEntityDefs().stream().filter(e -> e.getSuperTypes().contains("__internal")).findAny(); assertFalse(anyInternal.isPresent()); }
@Test public void filterInternalType() throws AtlasBaseException { SearchFilter searchFilter = new SearchFilter(); FilterUtil.addParamsToHideInternalType(searchFilter); AtlasTypesDef filteredTypeDefs = typeDefStore.searchTypesDef(searchFilter); assertNotNull(filteredTypeDefs); Optional<AtlasEntityDef> anyInternal = filteredTypeDefs.getEntityDefs().stream().filter(e -> e.getSuperTypes().contains("__internal")).findAny(); assertFalse(anyInternal.isPresent()); }
@Test(dependsOnMethods = "testGet") public void testSearchFunctionality() { SearchFilter searchFilter = new SearchFilter(); searchFilter.setParam(SearchFilter.PARAM_SUPERTYPE, "Person"); try { AtlasTypesDef typesDef = typeDefStore.searchTypesDef(searchFilter); assertNotNull(typesDef); assertNotNull(typesDef.getEntityDefs()); assertEquals(typesDef.getEntityDefs().size(), 3); } catch (AtlasBaseException e) { fail("Search should've succeeded", e); } }
@Test public void testGet() { try { AtlasTypesDef typesDef = typeDefStore.searchTypesDef(new SearchFilter()); assertNotNull(typesDef.getEnumDefs()); assertEquals(typesDef.getStructDefs().size(), 0); assertNotNull(typesDef.getStructDefs()); assertEquals(typesDef.getClassificationDefs().size(), 0); assertNotNull(typesDef.getClassificationDefs()); assertEquals(typesDef.getEntityDefs().size(), 0); assertNotNull(typesDef.getEntityDefs()); } catch (AtlasBaseException e) { fail("Search of types shouldn't have failed"); } }
@Test(dependsOnMethods = {"testGet"}, dataProvider = "validCreateDeptTypes") public void testCreateDept(AtlasTypesDef atlasTypesDef) { AtlasTypesDef existingTypesDef = null; try { existingTypesDef = typeDefStore.searchTypesDef(new SearchFilter()); } catch (AtlasBaseException e) { // ignore } assertNotEquals(atlasTypesDef, existingTypesDef, "Types to be created already exist in the system"); AtlasTypesDef createdTypesDef = null; try { createdTypesDef = typeDefStore.createTypesDef(atlasTypesDef); assertNotNull(createdTypesDef); assertTrue(createdTypesDef.getEnumDefs().containsAll(atlasTypesDef.getEnumDefs()), "EnumDefs create failed"); assertTrue(createdTypesDef.getClassificationDefs().containsAll(atlasTypesDef.getClassificationDefs()), "ClassificationDef create failed"); assertTrue(createdTypesDef.getStructDefs().containsAll(atlasTypesDef.getStructDefs()), "StructDef creation failed"); Assert.assertEquals(createdTypesDef.getEntityDefs(), atlasTypesDef.getEntityDefs()); } catch (AtlasBaseException e) { fail("Creation of Types should've been a success", e); } }