RANGER-5324: replace iterations with streams in RangerRequestScriptEvaluator - 3

  updated getAllTagTypes() to call init() which was removed in earlier commit

Author: mneethiraj

agents-common/src/main/java/org/apache/ranger/plugin/policyengine/RangerRequestScriptEvaluator.java

@@ -432,6 +432,8 @@ public String getCurrentTagType() {
     }
 
     public Set<String> getAllTagTypes() {
+        init();
+
         return Collections.unmodifiableSet(tags.keySet());
     }
 

agents-common/src/test/java/org/apache/ranger/plugin/conditionevaluator/RangerRequestScriptEvaluatorTest.java

@@ -513,6 +513,24 @@ public void testMultipleTagInstancesOfType() {
         }
     }
 
+    @Test
+    public void testGetAllTagTypes() {
+        RangerAccessRequest          request   = createRequest("test-user", Collections.emptySet(), Collections.emptySet(), Collections.emptyList());
+        RangerRequestScriptEvaluator evaluator = new RangerRequestScriptEvaluator(request, scriptEngine, false);
+
+        Assert.assertEquals(Collections.emptySet(), evaluator.evaluateScript("ctx.getAllTagTypes()"));
+
+        request   = createRequest("test-user", Collections.emptySet(), Collections.emptySet(), Collections.singletonList(new RangerTag("PII", Collections.emptyMap())));
+        evaluator = new RangerRequestScriptEvaluator(request, scriptEngine, false);
+
+        Assert.assertEquals(Collections.singleton("PII"), evaluator.evaluateScript("ctx.getAllTagTypes()"));
+
+        request   = createRequest("test-user", Collections.emptySet(), Collections.emptySet(), Arrays.asList(new RangerTag("PII", Collections.emptyMap()), new RangerTag("PCI", Collections.emptyMap())));
+        evaluator = new RangerRequestScriptEvaluator(request, scriptEngine, false);
+
+        Assert.assertEquals(new HashSet<>(Arrays.asList("PCI", "PII")), evaluator.evaluateScript("ctx.getAllTagTypes()"));
+    }
+
     RangerAccessRequest createRequest(String userName, Set<String> userGroups, Set<String> userRoles, List<RangerTag> resourceTags) {
         RangerAccessResource resource = mock(RangerAccessResource.class);