From 71946bad7ad9866ebce409092094ba6b04875bbf Mon Sep 17 00:00:00 2001
From: Bryan Frimin <bryan@getprobo.com>
Date: Mon, 1 Dec 2025 13:29:40 +0100
Subject: [PATCH] Fix TestOrganization_UpdateContext

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
---
 e2e/console/organization_test.go | 35 +++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/e2e/console/organization_test.go b/e2e/console/organization_test.go
index b248d5416..95556ec7c 100644
--- a/e2e/console/organization_test.go
+++ b/e2e/console/organization_test.go
@@ -137,7 +137,40 @@ func TestOrganization_Update(t *testing.T) {
 }
 
 func TestOrganization_UpdateContext(t *testing.T) {
-	t.Skip("updateOrganizationContext mutation not implemented")
+	t.Parallel()
+	owner := testutil.NewClient(t, testutil.RoleOwner)
+
+	query := `
+		mutation UpdateOrganizationContext($input: UpdateOrganizationContextInput!) {
+			updateOrganizationContext(input: $input) {
+				context {
+					organizationId
+					summary
+				}
+			}
+		}
+	`
+
+	var result struct {
+		UpdateOrganizationContext struct {
+			Context struct {
+				OrganizationID string  `json:"organizationId"`
+				Summary        *string `json:"summary"`
+			} `json:"context"`
+		} `json:"updateOrganizationContext"`
+	}
+
+	err := owner.Execute(query, map[string]any{
+		"input": map[string]any{
+			"organizationId": owner.GetOrganizationID().String(),
+			"summary":        "Our organization provides compliance solutions.",
+		},
+	}, &result)
+	require.NoError(t, err)
+
+	assert.Equal(t, owner.GetOrganizationID().String(), result.UpdateOrganizationContext.Context.OrganizationID)
+	require.NotNil(t, result.UpdateOrganizationContext.Context.Summary)
+	assert.Equal(t, "Our organization provides compliance solutions.", *result.UpdateOrganizationContext.Context.Summary)
 }
 
 func TestOrganization_Get(t *testing.T) {