Fix HttpServer AppSec Flaky Tests (#9956)

Author: zarirhamza

dd-java-agent/instrumentation-testing/src/main/groovy/datadog/trace/agent/test/base/HttpServerTest.groovy

@@ -2432,15 +2432,18 @@ abstract class HttpServerTest<SERVER> extends WithHttpServer<SERVER> {
 
     final BiFunction<RequestContext, IGSpanInfo, Flow<Void>> requestEndedCb =
     ({ RequestContext rqCtxt, IGSpanInfo info ->
-      Context context = rqCtxt.getData(RequestContextSlot.APPSEC)
-      if (context.responseBodyTag) {
-        rqCtxt.traceSegment.setTagTop('response.body', context.responseBody)
-      }
-      if (context.extraSpanName) {
-        runUnderTrace(context.extraSpanName, false) {
-          def span = activeSpan()
-          context.tags.each { key, val ->
-            span.setTag(key, val)
+      Object contextObj = rqCtxt.getData(RequestContextSlot.APPSEC)
+      if (contextObj instanceof Context) {
+        Context context = (Context) contextObj
+        if (context.responseBodyTag) {
+          rqCtxt.traceSegment.setTagTop('response.body', context.responseBody)
+        }
+        if (context.extraSpanName) {
+          runUnderTrace(context.extraSpanName, false) {
+            def span = activeSpan()
+            context.tags.each { key, val ->
+              span.setTag(key, val)
+            }
           }
         }
       }
@@ -2649,8 +2652,9 @@ abstract class HttpServerTest<SERVER> extends WithHttpServer<SERVER> {
 
     final BiFunction<RequestContext, String, Flow<Void>> requestSessionCb =
     { RequestContext rqCtxt, String sessionId ->
-      Context context = rqCtxt.getData(RequestContextSlot.APPSEC)
-      if (sessionId != null) {
+      Object contextObj = rqCtxt.getData(RequestContextSlot.APPSEC)
+      if (contextObj instanceof Context && sessionId != null) {
+        Context context = (Context) contextObj
         context.extraSpanName = 'appsec-span'
         context.tags.put(IG_SESSION_ID_TAG, sessionId)
       }