added default log type and updated tests

himanshu219 · himanshu219 · commit 3795aaed4a6a · 2022-07-25T19:28:46.000+05:30
diff --git a/lambda-extensions/config/config.go b/lambda-extensions/config/config.go
@@ -36,7 +36,8 @@ type LambdaExtensionConfig struct {
 	SourceCategoryOverride string
 }
 
-var validLogTypes = []string{"platform", "function"}
+var defaultLogTypes = []string{"platform", "function"}
+var validLogTypes = []string{"platform", "function", "extension"}
 
 // GetConfig to get config instance
 func GetConfig() (*LambdaExtensionConfig, error) {
@@ -93,7 +94,7 @@ func (cfg *LambdaExtensionConfig) setDefaults() {
 		cfg.AWSLambdaRuntimeAPI = "127.0.0.1:9001"
 	}
 	if logTypes == "" {
-		cfg.LogTypes = validLogTypes
+		cfg.LogTypes = defaultLogTypes
 	} else {
 		cfg.LogTypes = strings.Split(logTypes, ",")
 	}
diff --git a/lambda-extensions/sumoclient/sumoclient_test.go b/lambda-extensions/sumoclient/sumoclient_test.go
@@ -31,6 +31,8 @@ func setupEnv() {
 	os.Setenv("SUMO_MAX_DATAQUEUE_LENGTH", "10")
 	os.Setenv("SUMO_MAX_CONCURRENT_REQUESTS", "3")
 	os.Setenv("SUMO_LOG_LEVEL", "DEBUG")
+	os.Setenv("SUMO_RETRY_SLEEP_TIME_MS", "50")
+	os.Setenv("SUMO_LOG_TYPES", "function")
 }
 
 func assertEqual(t *testing.T, a interface{}, b interface{}, message string) {

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,8 @@ type LambdaExtensionConfig struct {`
`36`	`36`	`SourceCategoryOverride string`
`37`	`37`	`}`
`38`	`38`
`39`		`-var validLogTypes = []string{"platform", "function"}`
	`39`	`+var defaultLogTypes = []string{"platform", "function"}`
	`40`	`+var validLogTypes = []string{"platform", "function", "extension"}`
`40`	`41`
`41`	`42`	`// GetConfig to get config instance`
`42`	`43`	`func GetConfig() (*LambdaExtensionConfig, error) {`
`@@ -93,7 +94,7 @@ func (cfg *LambdaExtensionConfig) setDefaults() {`
`93`	`94`	`cfg.AWSLambdaRuntimeAPI = "127.0.0.1:9001"`
`94`	`95`	`}`
`95`	`96`	`if logTypes == "" {`
`96`		`- cfg.LogTypes = validLogTypes`
	`97`	`+ cfg.LogTypes = defaultLogTypes`
`97`	`98`	`} else {`
`98`	`99`	`cfg.LogTypes = strings.Split(logTypes, ",")`
`99`	`100`	`}`
Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,8 @@ func setupEnv() {`
`31`	`31`	`os.Setenv("SUMO_MAX_DATAQUEUE_LENGTH", "10")`
`32`	`32`	`os.Setenv("SUMO_MAX_CONCURRENT_REQUESTS", "3")`
`33`	`33`	`os.Setenv("SUMO_LOG_LEVEL", "DEBUG")`
	`34`	`+ os.Setenv("SUMO_RETRY_SLEEP_TIME_MS", "50")`
	`35`	`+ os.Setenv("SUMO_LOG_TYPES", "function")`
`34`	`36`	`}`
`35`	`37`
`36`	`38`	`func assertEqual(t *testing.T, a interface{}, b interface{}, message string) {`