feat: add datafile accessor (#290)

* Initial commit.

* fixes.

Co-authored-by: Michael Ng <[email protected]>
Co-authored-by: msohailhussain <[email protected]>

Author: 3 people

pkg/config/datafileprojectconfig/config.go

@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright 2019, Optimizely, Inc. and contributors                        *
+ * Copyright 2019-2020, Optimizely, Inc. and contributors                   *
  *                                                                          *
  * Licensed under the Apache License, Version 2.0 (the "License");          *
  * you may not use this file except in compliance with the License.         *
@@ -32,6 +32,7 @@ var datafileVersions = map[string]struct{}{
 
 // DatafileProjectConfig is a project config backed by a datafile
 type DatafileProjectConfig struct {
+	datafile             string
 	accountID            string
 	projectID            string
 	revision             string
@@ -48,6 +49,11 @@ type DatafileProjectConfig struct {
 	botFiltering         bool
 }
 
+// GetDatafile returns a string representation of the environment's datafile
+func (c DatafileProjectConfig) GetDatafile() string {
+	return c.datafile
+}
+
 // GetProjectID returns projectID
 func (c DatafileProjectConfig) GetProjectID() string {
 	return c.projectID
@@ -196,6 +202,7 @@ func NewDatafileProjectConfig(jsonDatafile []byte, logger logging.OptimizelyLogP
 	mergedAudiences := append(datafile.TypedAudiences, datafile.Audiences...)
 	featureMap := mappers.MapFeatures(datafile.FeatureFlags, rolloutMap, experimentMap)
 	config := &DatafileProjectConfig{
+		datafile:             string(jsonDatafile),
 		accountID:            datafile.AccountID,
 		anonymizeIP:          datafile.AnonymizeIP,
 		attributeKeyToIDMap:  attributeKeyToIDMap,

pkg/config/interface.go

@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright 2019, Optimizely, Inc. and contributors                        *
+ * Copyright 2019-2020, Optimizely, Inc. and contributors                   *
  *                                                                          *
  * Licensed under the Apache License, Version 2.0 (the "License");          *
  * you may not use this file except in compliance with the License.         *
@@ -24,6 +24,7 @@ import (
 
 // ProjectConfig represents the project's experiments and feature flags and contains methods for accessing the them.
 type ProjectConfig interface {
+	GetDatafile() string
 	GetAccountID() string
 	GetAnonymizeIP() bool
 	GetAttributeID(id string) string // returns "" if there is no id

pkg/config/optimizely_config.go

@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright 2019, Optimizely, Inc. and contributors                        *
+ * Copyright 2019-2020, Optimizely, Inc. and contributors                   *
  *                                                                          *
  * Licensed under the Apache License, Version 2.0 (the "License");          *
  * you may not use this file except in compliance with the License.         *
@@ -17,13 +17,21 @@
 // Package config //
 package config
 
-import "github.com/optimizely/go-sdk/pkg/entities"
+import (
+	"github.com/optimizely/go-sdk/pkg/entities"
+)
 
 // OptimizelyConfig is a snapshot of the experiments and features in the project config
 type OptimizelyConfig struct {
 	Revision       string                          `json:"revision"`
 	ExperimentsMap map[string]OptimizelyExperiment `json:"experimentsMap"`
 	FeaturesMap    map[string]OptimizelyFeature    `json:"featuresMap"`
+	datafile       string
+}
+
+// GetDatafile returns a string representation of the environment's datafile
+func (c OptimizelyConfig) GetDatafile() string {
+	return c.datafile
 }
 
 // OptimizelyExperiment has experiment info
@@ -159,6 +167,7 @@ func NewOptimizelyConfig(projConfig ProjectConfig) *OptimizelyConfig {
 	optimizelyConfig.ExperimentsMap = getExperimentMap(featuresList, experimentsList, variableByIDMap)
 	optimizelyConfig.FeaturesMap = getFeatureMap(featuresList, optimizelyConfig.ExperimentsMap)
 	optimizelyConfig.Revision = revision
+	optimizelyConfig.datafile = projConfig.GetDatafile()
 
 	return optimizelyConfig
 }

pkg/config/optimizely_config_test.go

@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright 2019, Optimizely, Inc. and contributors                        *
+ * Copyright 2019-2020, Optimizely, Inc. and contributors                   *
  *                                                                          *
  * Licensed under the Apache License, Version 2.0 (the "License");          *
  * you may not use this file except in compliance with the License.         *
@@ -49,6 +49,7 @@ func (s *OptimizelyConfigTestSuite) SetupTest() {
 	if err != nil {
 		s.Fail("error opening file " + dataFileName)
 	}
+	s.expectedOptimizelyConfig.datafile = string(dataFile)
 
 	projectMgr := NewStaticProjectConfigManagerWithOptions("", WithInitialDatafile(dataFile))
 
@@ -62,17 +63,25 @@ func (s *OptimizelyConfigTestSuite) TestOptlyConfig() {
 	s.Equal(s.expectedOptimizelyConfig.FeaturesMap, optimizelyConfig.FeaturesMap)
 	s.Equal(s.expectedOptimizelyConfig.ExperimentsMap, optimizelyConfig.ExperimentsMap)
 	s.Equal(s.expectedOptimizelyConfig.Revision, optimizelyConfig.Revision)
+	s.Equal(s.expectedOptimizelyConfig.datafile, optimizelyConfig.datafile)
 
 	s.Equal(s.expectedOptimizelyConfig, *optimizelyConfig)
-
 }
 
 func (s *OptimizelyConfigTestSuite) TestOptlyConfigNullProjectConfig() {
 	optimizelyConfig := NewOptimizelyConfig(nil)
 
 	s.Nil(optimizelyConfig)
+}
 
+func (s *OptimizelyConfigTestSuite) TestOptlyConfigGetDatafile() {
+	datafile := []byte(`{"version":"4"}`)
+	projectMgr := NewStaticProjectConfigManagerWithOptions("", WithInitialDatafile(datafile))
+	optimizelyConfig := NewOptimizelyConfig(projectMgr.projectConfig)
+	s.NotNil(optimizelyConfig.datafile)
+	s.Equal(string(datafile), optimizelyConfig.GetDatafile())
 }
+
 func TestOptimizelyConfigTestSuite(t *testing.T) {
 	suite.Run(t, new(OptimizelyConfigTestSuite))
 }

pkg/config/static_manager_test.go

@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright 2019-2020, Optimizely, Inc. and contributors                        *
+ * Copyright 2019-2020, Optimizely, Inc. and contributors                   *
  *                                                                          *
  * Licensed under the Apache License, Version 2.0 (the "License");          *
  * you may not use this file except in compliance with the License.         *
@@ -58,7 +58,7 @@ func TestStaticGetOptimizelyConfig(t *testing.T) {
 	optimizelyConfig := configManager.GetOptimizelyConfig()
 	assert.NotNil(t, configManager.optimizelyConfig)
 	assert.Equal(t, &OptimizelyConfig{ExperimentsMap: map[string]OptimizelyExperiment{},
-		FeaturesMap: map[string]OptimizelyFeature{}}, optimizelyConfig)
+		FeaturesMap: map[string]OptimizelyFeature{}, datafile: "{\"accountId\":\"42\",\"projectId\":\"123\",\"version\":\"4\"}"}, optimizelyConfig)
 }
 func TestNewStaticProjectConfigManagerFromURL(t *testing.T) {