Merge pull request #119 from newrelic/unittest/introspector-api

New API in Introspector for unit test cases
newrelic · Oct 9, 2023 · 86e5c9a · 86e5c9a
2 parents 65d6fed + 9e96948
commit 86e5c9a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/...curity-test/src/main/java/com/newrelic/agent/security/introspec/SecurityIntrospector.java b/...curity-test/src/main/java/com/newrelic/agent/security/introspec/SecurityIntrospector.java
@@ -41,5 +41,7 @@ public interface SecurityIntrospector {
 
     void setK2TracingData(String value);
 
+    void setK2ParentId(String value);
+
     void clear();
 }
diff --git a/...rc/main/java/com/newrelic/agent/security/introspec/internal/SecurityIntrospectorImpl.java b/...rc/main/java/com/newrelic/agent/security/introspec/internal/SecurityIntrospectorImpl.java
@@ -3,6 +3,7 @@
 import com.newrelic.agent.security.introspec.SecurityIntrospector;
 import com.newrelic.api.agent.security.Agent;
 import com.newrelic.api.agent.security.NewRelicSecurity;
+import com.newrelic.api.agent.security.instrumentation.helpers.GenericHelper;
 import com.newrelic.api.agent.security.instrumentation.helpers.JdbcHelper;
 import com.newrelic.api.agent.security.schema.AbstractOperation;
 import com.newrelic.api.agent.security.schema.HttpRequest;
@@ -108,6 +109,11 @@ public void setK2TracingData(String value) {
         NewRelicSecurity.getAgent().getSecurityMetaData().setTracingHeaderValue(value);
     }
 
+    @Override
+    public void setK2ParentId(String value) {
+        NewRelicSecurity.getAgent().getSecurityMetaData().addCustomAttribute(GenericHelper.CSEC_PARENT_ID, value);
+    }
+
     @Override
     public void setRequestInputStreamHash(int hashCode) {
         NewRelicSecurity.getAgent().getSecurityMetaData().addCustomAttribute(REQUEST_INPUTSTREAM_HASH, Collections.singleton(hashCode));