From 84d600b8e4745e75c6aad2798ac10d79acd90662 Mon Sep 17 00:00:00 2001
From: Brooklyn Welsh <brooklyn.welsh@caci.com>
Date: Tue, 22 Oct 2024 23:25:51 +0000
Subject: [PATCH] Another mock fix, was returning a func instead of a FF
 struct.

---
 pkg/handlers/apitests.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/handlers/apitests.go b/pkg/handlers/apitests.go
index 6ab3bbd8679..8250facc90e 100644
--- a/pkg/handlers/apitests.go
+++ b/pkg/handlers/apitests.go
@@ -106,7 +106,13 @@ func (suite *BaseHandlerTestSuite) HandlerConfig() *Config {
 		mock.AnythingOfType("string"),
 		mock.Anything,
 	).Return(func(ctx context.Context, appCtx appcontext.AppContext, key string, flagContext map[string]string) (services.FeatureFlag, error) {
-		return mockGetFlagFunc(ctx, appCtx.Logger(), "user@example.com", key, flagContext, "")
+		return services.FeatureFlag{
+			Entity:    "user@example.com",
+			Key:       key,
+			Match:     true,
+			Variant:   "",
+			Namespace: "test",
+		}, nil
 	})
 	return &Config{
 		db:                 suite.DB(),