add unit tests

pkg/client/client_test.go

@@ -1378,6 +1378,15 @@ func TestGetOptimizelyConfig(t *testing.T) {
 	assert.Equal(t, &config.OptimizelyConfig{Revision: "232"}, optimizelyConfig)
 }
 
+func TestGetNotificationCenter(t *testing.T) {
+	nc := &MockNotificationCenter{}
+	client := OptimizelyClient{
+		notificationCenter: nc,
+	}
+
+	assert.Equal(t, client.GetNotificationCenter(), nc)
+}
+
 func TestGetFeatureDecisionValid(t *testing.T) {
 	testFeatureKey := "test_feature_key"
 	testVariableKey := "test_feature_flag_key"

pkg/client/factory_test.go

@@ -179,6 +179,14 @@ func TestClientWithDefaultSDKSettings(t *testing.T) {
 	assert.NotNil(t, optimizelyClient.OdpManager)
 }
 
+func TestClientWithNotificationCenterInOptions(t *testing.T) {
+	factory := OptimizelyFactory{SDKKey: "1212"}
+	nc := &MockNotificationCenter{}
+	optimizelyClient, err := factory.Client(WithNotificationCenter(nc))
+	assert.NoError(t, err)
+	assert.Equal(t, nc, optimizelyClient.notificationCenter)
+}
+
 func TestDummy(t *testing.T) {
 	factory := OptimizelyFactory{}
 	configManager := config.NewPollingProjectConfigManager("123")