From 02e66db07332b196be5620bbcb6a1a57bd113adb Mon Sep 17 00:00:00 2001
From: Chris Ainsworth-Patrick <chris.ainsworth-patrick@docker.com>
Date: Thu, 7 Mar 2024 13:22:36 +0000
Subject: [PATCH] need to pass in skill execution context

---
 policy/policy_handler/mocks/base_image_details.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/policy/policy_handler/mocks/base_image_details.go b/policy/policy_handler/mocks/base_image_details.go
index 28dd197..5b0b8c8 100644
--- a/policy/policy_handler/mocks/base_image_details.go
+++ b/policy/policy_handler/mocks/base_image_details.go
@@ -93,7 +93,9 @@ func mockBaseImageDetails(ctx context.Context, req skill.RequestContext, sb *typ
 
 	var queryResponse BaseImagesByDigestResponse
 
-	queryVariables := map[string]interface{}{"digest": baseImageDigest}
+	queryVariables := map[string]interface{}{
+		"digest":  baseImageDigest,
+		"context": data.GqlContext(req)}
 	_, err = ds.Query(ctx, baseImagesByDigestQueryName, baseImagesByDigestQuery, queryVariables, &queryResponse)
 	if err != nil {
 		return ImageDetailsByDigestResponse{}, err