+2 tests

Unleash · Dec 8, 2023 · 4832f4e · 4832f4e
1 parent 96bfb90
commit 4832f4e
Showing 1 changed file with 54 additions and 52 deletions.
diff --git a/client_test.go b/client_test.go
@@ -4,6 +4,8 @@
 package unleash
 
 import (
+	"time"
+
 	"github.com/Unleash/unleash-client-go/v4/api"
 	"github.com/Unleash/unleash-client-go/v4/context"
 
@@ -98,68 +100,68 @@ func TestClient_WithFallbackFunc(t *testing.T) {
 	assert.True(gock.IsDone(), "there should be no more mocks")
 }
 
-// func TestClient_WithResolver(t *testing.T) {
-// 	assert := assert.New(t)
-// 	defer gock.OffAll()
+func TestClient_WithResolver(t *testing.T) {
+	assert := assert.New(t)
+	defer gock.OffAll()
 
-// 	gock.New(mockerServer).
-// 		Post("/client/register").
-// 		MatchHeader("UNLEASH-APPNAME", mockAppName).
-// 		MatchHeader("UNLEASH-INSTANCEID", mockInstanceId).
-// 		Reply(200)
+	gock.New(mockerServer).
+		Post("/client/register").
+		MatchHeader("UNLEASH-APPNAME", mockAppName).
+		MatchHeader("UNLEASH-INSTANCEID", mockInstanceId).
+		Reply(200)
 
-// 	gock.New(mockerServer).
-// 		Get("/client/features").
-// 		Reply(200).
-// 		JSON(api.FeatureResponse{})
+	gock.New(mockerServer).
+		Get("/client/features").
+		Reply(200).
+		JSON(api.FeatureResponse{})
 
-// 	const feature = "some_special_value"
+	const feature = "some_special_value"
 
-// 	mockListener := &MockedListener{}
-// 	mockListener.On("OnReady").Return()
-// 	mockListener.On("OnRegistered", mock.AnythingOfType("ClientData"))
-// 	mockListener.On("OnCount", feature, true).Return()
-// 	mockListener.On("OnError").Return()
+	mockListener := &MockedListener{}
+	mockListener.On("OnReady").Return()
+	mockListener.On("OnRegistered", mock.AnythingOfType("ClientData"))
+	mockListener.On("OnCount", feature, true).Return()
+	mockListener.On("OnError").Return()
 
-// 	client, err := NewClient(
-// 		WithUrl(mockerServer),
-// 		WithAppName(mockAppName),
-// 		WithInstanceId(mockInstanceId),
-// 		WithListener(mockListener),
-// 	)
+	client, err := NewClient(
+		WithUrl(mockerServer),
+		WithAppName(mockAppName),
+		WithInstanceId(mockInstanceId),
+		WithListener(mockListener),
+	)
 
-// 	assert.NoError(err)
+	assert.NoError(err)
 
-// 	client.WaitForReady()
+	client.WaitForReady()
 
-// 	resolver := func(featureName string) *api.Feature {
-// 		if featureName == feature {
-// 			return &api.Feature{
-// 				Name:        "some_special_value-resolved",
-// 				Description: "",
-// 				Enabled:     true,
-// 				Strategies: []api.Strategy{
-// 					{
-// 						Id:   1,
-// 						Name: "default",
-// 					},
-// 				},
-// 				CreatedAt:  time.Time{},
-// 				Strategy:   "default-strategy",
-// 				Parameters: nil,
-// 				Variants:   nil,
-// 			}
-// 		} else {
-// 			t.Fatalf("the feature name passed %s was not the expected one %s", featureName, feature)
-// 			return nil
-// 		}
-// 	}
+	resolver := func(featureName string) *api.Feature {
+		if featureName == feature {
+			return &api.Feature{
+				Name:        "some_special_value-resolved",
+				Description: "",
+				Enabled:     true,
+				Strategies: []api.Strategy{
+					{
+						Id:   1,
+						Name: "default",
+					},
+				},
+				CreatedAt:  time.Time{},
+				Strategy:   "default-strategy",
+				Parameters: nil,
+				Variants:   nil,
+			}
+		} else {
+			t.Fatalf("the feature name passed %s was not the expected one %s", featureName, feature)
+			return nil
+		}
+	}
 
-// 	isEnabled := client.IsEnabled(feature, WithResolver(resolver))
-// 	assert.True(isEnabled)
+	isEnabled := client.IsEnabled(feature, WithResolver(resolver))
+	assert.True(isEnabled)
 
-// 	assert.True(gock.IsDone(), "there should be no more mocks")
-// }
+	assert.True(gock.IsDone(), "there should be no more mocks")
+}
 
 // func TestClient_ListFeatures(t *testing.T) {
 // 	assert := assert.New(t)