test changes

Author: nhulston

internal/extension/extension.go

@@ -125,6 +125,7 @@ func (em *ExtensionManager) SendStartInvocationRequest(ctx context.Context, even
 			ctx = context.WithValue(ctx, DdParentId, parentId)
 		}
 		samplingPriority := response.Header.Get(string(DdSamplingPriority))
+		fmt.Printf("[DEBUG] [extension.go] [SendStartInvocationRequest()] Sampling priority: %s\n", samplingPriority)
 		if samplingPriority != "" {
 			ctx = context.WithValue(ctx, DdSamplingPriority, samplingPriority)
 		}
@@ -161,6 +162,7 @@ func (em *ExtensionManager) SendEndInvocationRequest(ctx context.Context, functi
 			req.Header.Set(string(DdSpanId), spanId)
 		}
 		if samplingPriority, ok := ctx.Value(DdSamplingPriority).(string); ok {
+			fmt.Printf("[DEBUG] [extension.go] [SendEndInvocationRequest()] Sampling priority: %s\n", samplingPriority)
 			req.Header.Set(string(DdSamplingPriority), samplingPriority)
 		}
 	} else {

internal/trace/context.go

@@ -76,6 +76,7 @@ func contextWithRootTraceContext(ctx context.Context, ev json.RawMessage, mergeX
 	mergedTraceContext := TraceContext{}
 	mergedTraceContext[traceIDHeader] = datadogTraceContext[traceIDHeader]
 	mergedTraceContext[samplingPriorityHeader] = datadogTraceContext[samplingPriorityHeader]
+	fmt.Printf("[DEBUG] [context.go] [contextWithRootTraceContext()] Sampling priority: %s\n", datadogTraceContext[samplingPriorityHeader])
 	mergedTraceContext[parentIDHeader] = xrayTraceContext[parentIDHeader]
 	return context.WithValue(ctx, traceContextKey, mergedTraceContext), nil
 }
@@ -97,6 +98,7 @@ func ConvertCurrentXrayTraceContext(ctx context.Context) TraceContext {
 		newTraceContext := map[string]string{}
 		newTraceContext[traceIDHeader] = xrayTraceContext[traceIDHeader]
 		newTraceContext[samplingPriorityHeader] = xrayTraceContext[samplingPriorityHeader]
+		fmt.Printf("[DEBUG] [context.go] [ConvertCurrentXrayTraceContext()] Sampling priority: %s\n", xrayTraceContext[samplingPriorityHeader])
 		newTraceContext[parentIDHeader] = parentID
 
 		return newTraceContext
@@ -113,6 +115,7 @@ func createDummySubsegmentForXrayConverter(ctx context.Context, traceCtx TraceCo
 	traceID := traceCtx[traceIDHeader]
 	parentID := traceCtx[parentIDHeader]
 	sampled := traceCtx[samplingPriorityHeader]
+	fmt.Printf("[DEBUG] [context.go] [createDummySubsegmentForXrayConverter()] Sampling priority: %s\n", sampled)
 	metadata := map[string]string{
 		"trace-id":          traceID,
 		"parent-id":         parentID,
@@ -151,13 +154,16 @@ func getTraceContext(ctx context.Context, headers map[string]string) (TraceConte
 	}
 
 	samplingPriority := headers[samplingPriorityHeader]
+	fmt.Printf("[DEBUG] [context.go] [getTraceContext() 1] Sampling priority: %s\n", samplingPriority)
 	if samplingPriority == "" {
 		if val, ok := ctx.Value(extension.DdSamplingPriority).(string); ok {
 			samplingPriority = val
+			fmt.Printf("[DEBUG] [context.go] [getTraceContext() 2] Sampling priority: %s\n", samplingPriority)
 		}
 	}
 	if samplingPriority == "" {
 		samplingPriority = "1" //sampler-keep
+		fmt.Printf("[DEBUG] [context.go] [getTraceContext() 3] Sampling priority empty, keeping.\n")
 	}
 
 	tc[samplingPriorityHeader] = samplingPriority
@@ -205,6 +211,7 @@ func convertXrayTraceContextFromLambdaContext(ctx context.Context) (TraceContext
 		return traceCtx, fmt.Errorf("Couldn't read parent id from X-Ray: %v", err)
 	}
 	samplingPriority := convertXRaySamplingDecision(header.SamplingDecision)
+	fmt.Printf("[DEBUG] [context.go] [convertXrayTraceContextFromLambdaContext()] Sampling priority: %s\n", samplingPriority)
 
 	traceCtx[traceIDHeader] = traceID
 	traceCtx[parentIDHeader] = parentID