Examples with TagStore org.apache.ranger.plugin.store.TagStore used on opensource projects

Example 1 with TagStore

use of org.apache.ranger.plugin.store.TagStore in project ranger by apache.

the class TestServiceTagProcessor method test3process.

@Test
public void test3process() throws Exception {
    serviceTags = new ServiceTags();
    Map<Long, RangerTagDef> fd = new HashMap<>();
    List<RangerTag> associatedTags = new ArrayList<>();
    RangerTagDef rTagDef = Mockito.mock(RangerTagDef.class);
    rTagDef.setGuid("guid");
    rTagDef.setName("rTagDefname");
    fd.put(1l, rTagDef);
    serviceTags.setTagDefinitions(fd);
    List<RangerServiceResource> serviceResources = new ArrayList<RangerServiceResource>();
    RangerServiceResource rserRes = new RangerServiceResource();
    testServiceRest = new TestServiceREST();
    RangerPolicy rp = testServiceRest.rangerPolicy();
    rserRes.setResourceElements(rp.getResources());
    rserRes.setGuid("guId");
    rserRes.setId(1L);
    serviceResources.add(rserRes);
    serviceTags.setServiceResources(serviceResources);
    Map<Long, List<Long>> resourceToTagIds = new HashMap<>();
    resourceToTagIds.put(1L, new ArrayList<Long>(Arrays.asList(22L, 1L, 0L)));
    serviceTags.setResourceToTagIds(resourceToTagIds);
    RangerTag rTag = new RangerTag();
    rTag.setId(22L);
    rTag.setType("type1");
    Map<Long, RangerTag> tags = new HashMap<>();
    rTag.setOwner((short) 0);
    tags.put(1L, rTag);
    serviceTags.setTags(tags);
    associatedTags.add(rTag);
    TagStore tagStore = Mockito.mock(TagStore.class);
    sTagProcessor = new ServiceTagsProcessor(tagStore);
    Mockito.when(tagStore.createTagDef(rTagDef)).thenReturn(rTagDef);
    Mockito.when(tagStore.getServiceResourceByGuid(rserRes.getGuid())).thenReturn(rserRes);
    Mockito.when(tagStore.getTagsForResourceId(rserRes.getId())).thenReturn(associatedTags);
    sTagProcessor.process(serviceTags);
    Mockito.verify(tagStore).createTagDef(rTagDef);
    Mockito.verify(tagStore).getServiceResourceByGuid(rserRes.getGuid());
    Mockito.verify(tagStore).getTagsForResourceId(rserRes.getId());
}

Example 2 with TagStore

use of org.apache.ranger.plugin.store.TagStore in project ranger by apache.

the class TestServiceTagProcessor method test4processDelete.

@Test
public void test4processDelete() throws Exception {
    serviceTags = new ServiceTags();
    serviceTags.setOp(ServiceTags.OP_DELETE);
    Map<Long, RangerTagDef> fd = new HashMap<>();
    List<RangerTag> associatedTags = new ArrayList<>();
    RangerTagDef rTagDef = Mockito.mock(RangerTagDef.class);
    rTagDef.setGuid("guid");
    rTagDef.setName("rTagDefName");
    fd.put(1l, rTagDef);
    serviceTags.setTagDefinitions(fd);
    List<RangerServiceResource> serviceResources = new ArrayList<RangerServiceResource>();
    RangerServiceResource rserRes = new RangerServiceResource();
    testServiceRest = new TestServiceREST();
    RangerPolicy rp = testServiceRest.rangerPolicy();
    rserRes.setResourceElements(rp.getResources());
    rserRes.setGuid("guId");
    rserRes.setId(1L);
    rserRes.setServiceName("serviceName1");
    serviceResources.add(rserRes);
    serviceTags.setServiceResources(serviceResources);
    List<RangerTagResourceMap> tagResourceMaps = new ArrayList<RangerTagResourceMap>();
    tagResourceMaps.add(new RangerTagResourceMap());
    Map<Long, List<Long>> resourceToTagIds = new HashMap<>();
    resourceToTagIds.put(1L, new ArrayList<Long>(Arrays.asList(22L, 1L, 0L)));
    serviceTags.setResourceToTagIds(resourceToTagIds);
    RangerTag rTag = new RangerTag();
    rTag.setId(22L);
    rTag.setType("type1");
    rTag.setGuid("tagGuID");
    Map<Long, RangerTag> tags = new HashMap<>();
    rTag.setOwner((short) 0);
    tags.put(1L, rTag);
    serviceTags.setTags(tags);
    RangerServiceResourceSignature serializer = new RangerServiceResourceSignature(rserRes);
    String serviceResourceSignature = serializer.getSignature();
    associatedTags.add(rTag);
    TagStore tagStore = Mockito.mock(TagStore.class);
    sTagProcessor = new ServiceTagsProcessor(tagStore);
    Mockito.when(tagStore.getServiceResourceByGuid(rserRes.getGuid())).thenReturn(null);
    Mockito.when(tagStore.getServiceResourceByServiceAndResourceSignature(rserRes.getServiceName(), serviceResourceSignature)).thenReturn(rserRes);
    Mockito.when(tagStore.getTagResourceMapsForResourceGuid(rserRes.getGuid())).thenReturn(tagResourceMaps);
    Mockito.doNothing().when(tagStore).deleteServiceResource(rserRes.getId());
    Mockito.when(tagStore.getTagByGuid(rTag.getGuid())).thenReturn(rTag);
    Mockito.doNothing().when(tagStore).deleteTag(rTag.getId());
    Mockito.when(tagStore.getTagDefByGuid(rTagDef.getGuid())).thenReturn(rTagDef);
    sTagProcessor.process(serviceTags);
    Mockito.verify(tagStore).getServiceResourceByGuid(rserRes.getGuid());
    Mockito.verify(tagStore).getServiceResourceByServiceAndResourceSignature(rserRes.getServiceName(), serviceResourceSignature);
    Mockito.verify(tagStore).getTagResourceMapsForResourceGuid(rserRes.getGuid());
    Mockito.verify(tagStore).deleteServiceResource(rserRes.getId());
    Mockito.verify(tagStore).getTagByGuid(rTag.getGuid());
    Mockito.verify(tagStore).deleteTag(rTag.getId());
    Mockito.verify(tagStore).getTagDefByGuid(rTagDef.getGuid());
}

Example 3 with TagStore

use of org.apache.ranger.plugin.store.TagStore in project ranger by apache.

the class TestServiceTagProcessor method test5processReplace.

@Test
public void test5processReplace() throws Exception {
    serviceTags = new ServiceTags();
    serviceTags.setOp(ServiceTags.OP_REPLACE);
    Map<Long, RangerTagDef> fd = new HashMap<>();
    List<RangerTag> associatedTags = new ArrayList<>();
    RangerTagDef rTagDef = Mockito.mock(RangerTagDef.class);
    rTagDef.setGuid("guid");
    rTagDef.setName("rTagDefName");
    fd.put(1l, rTagDef);
    serviceTags.setTagDefinitions(fd);
    List<RangerServiceResource> serviceResources = new ArrayList<RangerServiceResource>();
    RangerServiceResource rserRes = new RangerServiceResource();
    testServiceRest = new TestServiceREST();
    RangerPolicy rp = testServiceRest.rangerPolicy();
    rserRes.setResourceElements(rp.getResources());
    rserRes.setGuid("guId");
    rserRes.setId(1L);
    rserRes.setServiceName("serviceName1");
    serviceResources.add(rserRes);
    serviceTags.setServiceResources(serviceResources);
    List<RangerTagResourceMap> tagResourceMaps = new ArrayList<RangerTagResourceMap>();
    serviceTags.setServiceName("tagServiceName");
    RangerTagResourceMap rangerTagRmp = new RangerTagResourceMap();
    rangerTagRmp.setId(2L);
    tagResourceMaps.add(rangerTagRmp);
    Map<Long, List<Long>> resourceToTagIds = new HashMap<>();
    resourceToTagIds.put(1L, new ArrayList<Long>(Arrays.asList(22L, 1L, 0L)));
    serviceTags.setResourceToTagIds(resourceToTagIds);
    RangerTag rTag = new RangerTag();
    rTag.setId(22L);
    rTag.setType("type1");
    rTag.setGuid("tagGuID");
    Map<Long, RangerTag> tags = new HashMap<>();
    rTag.setOwner((short) 0);
    tags.put(1L, rTag);
    serviceTags.setTags(tags);
    associatedTags.add(rTag);
    TagStore tagStore = Mockito.mock(TagStore.class);
    sTagProcessor = new ServiceTagsProcessor(tagStore);
    List<String> serviceResourcesInDb = new ArrayList<>(Arrays.asList("guid"));
    Mockito.when(tagStore.getServiceResourceGuidsByService(serviceTags.getServiceName())).thenReturn(serviceResourcesInDb);
    Mockito.when(tagStore.getTagResourceMapsForResourceGuid(Mockito.anyString())).thenReturn(tagResourceMaps);
    Mockito.doNothing().when(tagStore).deleteTagResourceMap(rangerTagRmp.getId());
    Mockito.doNothing().when(tagStore).deleteServiceResourceByGuid(Mockito.anyString());
    sTagProcessor.process(serviceTags);
    Mockito.verify(tagStore).getServiceResourceGuidsByService(serviceTags.getServiceName());
    Mockito.verify(tagStore).getTagResourceMapsForResourceGuid(Mockito.anyString());
    Mockito.verify(tagStore).deleteTagResourceMap(rangerTagRmp.getId());
    Mockito.verify(tagStore).deleteServiceResourceByGuid(Mockito.anyString());
}

Example 4 with TagStore

use of org.apache.ranger.plugin.store.TagStore in project ranger by apache.

the class TestServiceTagProcessor method test2processAddOrUpdate.

@Test
public void test2processAddOrUpdate() throws Exception {
    serviceTags = new ServiceTags();
    Map<Long, RangerTagDef> fd = new HashMap<>();
    List<RangerTag> associatedTags = new ArrayList<>();
    RangerTagDef rTagDef = Mockito.mock(RangerTagDef.class);
    rTagDef.setGuid("guid");
    rTagDef.setName("rTagDefname");
    fd.put(1l, rTagDef);
    serviceTags.setTagDefinitions(fd);
    List<RangerServiceResource> serviceResources = new ArrayList<RangerServiceResource>();
    RangerServiceResource rserRes = new RangerServiceResource();
    testServiceRest = new TestServiceREST();
    RangerPolicy rp = testServiceRest.rangerPolicy();
    rserRes.setResourceElements(rp.getResources());
    rserRes.setGuid("guId");
    rserRes.setId(1L);
    serviceResources.add(rserRes);
    serviceTags.setServiceResources(serviceResources);
    Map<Long, List<Long>> resourceToTagIds = new HashMap<>();
    resourceToTagIds.put(1L, new ArrayList<Long>(Arrays.asList(11L, 1L, 13L)));
    serviceTags.setResourceToTagIds(resourceToTagIds);
    RangerTag rTag = new RangerTag();
    rTag.setId(22L);
    rTag.setOwner((short) 1);
    Map<Long, RangerTag> tags = new HashMap<>();
    tags.put(1L, rTag);
    serviceTags.setTags(tags);
    RangerTag rTag2 = new RangerTag();
    rTag2.setId(22L);
    rTag2.setOwner((short) 1);
    Map<Long, RangerTag> tags2 = new HashMap<>();
    tags2.put(1L, rTag2);
    associatedTags.add(rTag2);
    TagStore tagStore = Mockito.mock(TagStore.class);
    sTagProcessor = new ServiceTagsProcessor(tagStore);
    Mockito.when(tagStore.createTagDef(rTagDef)).thenReturn(rTagDef);
    Mockito.when(tagStore.getServiceResourceByGuid(rserRes.getGuid())).thenReturn(rserRes);
    Mockito.when(tagStore.createTag(rTag2)).thenReturn(rTag);
    Mockito.when(tagStore.getTagsForResourceId(rserRes.getId())).thenReturn(associatedTags);
    sTagProcessor.process(serviceTags);
    Mockito.verify(tagStore).createTagDef(rTagDef);
    Mockito.verify(tagStore).getServiceResourceByGuid(rserRes.getGuid());
    Mockito.verify(tagStore).createTag(rTag2);
}