update mock/test for lsd_vectorstores_handler_test.go

Author: jharan1

packages/gen-ai/bff/internal/api/lsd_vectorstores_handler_test.go

@@ -68,16 +68,20 @@ var _ = Describe("LlamaStackListVectorStoresHandler", func() {
 		vectorStores := response.Data.([]interface{})
 		assert.Len(t, vectorStores, 3)
 
-		// Find the auto-provisioned user store by its 32-char hashed username name
-		var userStore map[string]interface{}
-		for _, vs := range vectorStores {
-			store := vs.(map[string]interface{})
-			if name, ok := store["name"].(string); ok && len(name) == 32 {
+		// Find the auto-provisioned user store by contract metadata marker
+ 		var userStore map[string]interface{}
+ 		for _, vs := range vectorStores {
+ 			store := vs.(map[string]interface{})
+			metadata, ok := store["metadata"].(map[string]interface{})
+			if !ok {
+				continue
+			}
+			if createdBy, ok := metadata["created_by"].(string); ok && createdBy == "auto-provisioning" {
 				userStore = store
 				break
 			}
-		}
-		require.NotNil(t, userStore, "expected auto-provisioned user store with 32-char hashed name")
+ 		}
+		require.NotNil(t, userStore, "expected auto-provisioned user store")
 		assert.NotEmpty(t, userStore["id"])
 		assert.Contains(t, []string{"completed", "in_progress"}, userStore["status"])
 	})

packages/gen-ai/bff/internal/integrations/llamastack/lsmocks/llamastack_client_mock.go

@@ -129,6 +129,13 @@ func (m *MockLlamaStackClient) CreateVectorStore(ctx context.Context, params lla
 	}
 
 	mockID := "vs_mock_new123"
+	metadata := map[string]string{
+		"provider_id":           "milvus",
+		"provider_vector_db_id": mockID,
+	}
+	for k, v := range params.Metadata {
+		metadata[k] = v
+	}
 	return &openai.VectorStore{
 		ID:         mockID,
 		Object:     "vector_store",
@@ -144,10 +151,7 @@ func (m *MockLlamaStackClient) CreateVectorStore(ctx context.Context, params lla
 		},
 		Status:       "completed",
 		LastActiveAt: 1755721097,
-		Metadata: map[string]string{
-			"provider_id":           "milvus",
-			"provider_vector_db_id": mockID,
-		},
+		Metadata:     metadata,
 		ExpiresAfter: openai.VectorStoreExpiresAfter{
 			Anchor: "last_active_at",
 			Days:   0,