k8s-tester/nlb-guestbook: initial commit

Signed-off-by: Gyuho Lee <leegyuho@amazon.com>

k8s-tester/README.md

@@ -36,7 +36,7 @@ See example commits:
 
 ### Environmental variables
 
-Total 18 test cases!
+Total 19 test cases!
 
 ```
 *----------------------------------*----------------------*----------------------------------------*---------------*
@@ -156,6 +156,22 @@ Total 18 test cases!
 | K8S_TESTER_ADD_ON_PHP_APACHE_REPOSITORY_IMAGE_TAG  | SETTABLE VIA ENV VAR | *ecr.Repository.ImageTag  | string  |
 *----------------------------------------------------*----------------------*---------------------------*---------*
 
+*----------------------------------------------------------*----------------------*----------------------------------------------*-------------------*
+|                  ENVIRONMENTAL VARIABLE                  |      FIELD TYPE      |                     TYPE                     |      GO TYPE      |
+*----------------------------------------------------------*----------------------*----------------------------------------------*-------------------*
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ENABLE                   | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.Enable                 | bool              |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ACCOUNT_ID               | READ-ONLY            | *nlb_guestbook.Config.AccountID              | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_PARTITION                | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.Partition              | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_REGION                   | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.Region                 | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_MINIMUM_NODES            | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.MinimumNodes           | int               |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_NAMESPACE                | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.Namespace              | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_DEPLOYMENT_NODE_SELECTOR | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.DeploymentNodeSelector | map[string]string |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_DEPLOYMENT_REPLICAS      | SETTABLE VIA ENV VAR | *nlb_guestbook.Config.DeploymentReplicas     | int32             |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ELB_ARN                  | READ-ONLY            | *nlb_guestbook.Config.ELBARN                 | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ELB_NAME                 | READ-ONLY            | *nlb_guestbook.Config.ELBName                | string            |
+| K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ELB_URL                  | READ-ONLY            | *nlb_guestbook.Config.ELBURL                 | string            |
+*----------------------------------------------------------*----------------------*----------------------------------------------*-------------------*
+
 *------------------------------------------------------------*----------------------*------------------------------------------------*-------------------*
 |                   ENVIRONMENTAL VARIABLE                   |      FIELD TYPE      |                      TYPE                      |      GO TYPE      |
 *------------------------------------------------------------*----------------------*------------------------------------------------*-------------------*

k8s-tester/cmd/k8s-tester/go.mod

@@ -10,7 +10,6 @@ require (
 
 replace (
 	github.com/aws/aws-k8s-tester/client => ../../../client
-
 	github.com/aws/aws-k8s-tester/k8s-tester => ../..
 	github.com/aws/aws-k8s-tester/k8s-tester/cloudwatch-agent => ../../cloudwatch-agent
 	github.com/aws/aws-k8s-tester/k8s-tester/clusterloader => ../../clusterloader
@@ -25,6 +24,7 @@ replace (
 	github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi => ../../jobs-pi
 	github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard => ../../kubernetes-dashboard
 	github.com/aws/aws-k8s-tester/k8s-tester/metrics-server => ../../metrics-server
+	github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook => ../../nlb-guestbook
 	github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world => ../../nlb-hello-world
 	github.com/aws/aws-k8s-tester/k8s-tester/php-apache => ../../php-apache
 	github.com/aws/aws-k8s-tester/k8s-tester/secrets => ../../secrets

k8s-tester/cmd/readme-gen/go.mod

@@ -26,7 +26,6 @@ require (
 
 replace (
 	github.com/aws/aws-k8s-tester/client => ../../../client
-
 	github.com/aws/aws-k8s-tester/k8s-tester => ../..
 	github.com/aws/aws-k8s-tester/k8s-tester/cloudwatch-agent => ../../cloudwatch-agent
 	github.com/aws/aws-k8s-tester/k8s-tester/clusterloader => ../../clusterloader
@@ -41,6 +40,7 @@ replace (
 	github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi => ../../jobs-pi
 	github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard => ../../kubernetes-dashboard
 	github.com/aws/aws-k8s-tester/k8s-tester/metrics-server => ../../metrics-server
+	github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook => ../../nlb-guestbook
 	github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world => ../../nlb-hello-world
 	github.com/aws/aws-k8s-tester/k8s-tester/php-apache => ../../php-apache
 	github.com/aws/aws-k8s-tester/k8s-tester/secrets => ../../secrets

k8s-tester/cmd/readme-gen/main.go

@@ -24,6 +24,7 @@ import (
 	jobs_pi "github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi"
 	kubernetes_dashboard "github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard"
 	metrics_server "github.com/aws/aws-k8s-tester/k8s-tester/metrics-server"
+	nlb_guestbook "github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook"
 	nlb_hello_world "github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world"
 	php_apache "github.com/aws/aws-k8s-tester/k8s-tester/php-apache"
 	"github.com/aws/aws-k8s-tester/k8s-tester/secrets"
@@ -101,6 +102,10 @@ func createDoc() string {
 	b.WriteString(es.writeDoc(k8s_tester.ENV_PREFIX+php_apache.EnvRepository()+"_", &aws_v1_ecr.Repository{}))
 	totalTestCases++
 
+	b.WriteByte('\n')
+	b.WriteString(es.writeDoc(k8s_tester.ENV_PREFIX+nlb_guestbook.Env()+"_", &nlb_guestbook.Config{}))
+	totalTestCases++
+
 	b.WriteByte('\n')
 	b.WriteString(es.writeDoc(k8s_tester.ENV_PREFIX+nlb_hello_world.Env()+"_", &nlb_hello_world.Config{}))
 	totalTestCases++

k8s-tester/config.go

@@ -28,6 +28,7 @@ import (
 	jobs_pi "github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi"
 	kubernetes_dashboard "github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard"
 	metrics_server "github.com/aws/aws-k8s-tester/k8s-tester/metrics-server"
+	nlb_guestbook "github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook"
 	nlb_hello_world "github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world"
 	php_apache "github.com/aws/aws-k8s-tester/k8s-tester/php-apache"
 	"github.com/aws/aws-k8s-tester/k8s-tester/secrets"
@@ -132,6 +133,7 @@ type Config struct {
 	AddOnKubernetesDashboard *kubernetes_dashboard.Config `json:"add_on_kubernetes_dashboard"`
 	AddOnFalco               *falco.Config                `json:"add_on_falco"`
 	AddOnPHPApache           *php_apache.Config           `json:"add_on_php_apache"`
+	AddOnNLBGuestbook        *nlb_guestbook.Config        `json:"add_on_nlb_guestbook"`
 	AddOnNLBHelloWorld       *nlb_hello_world.Config      `json:"add_on_nlb_hello_world"`
 	AddOnJobsPi              *jobs_pi.Config              `json:"add_on_jobs_pi"`
 	AddOnJobsEcho            *jobs_echo.Config            `json:"add_on_jobs_echo"`
@@ -204,6 +206,7 @@ func NewDefault() *Config {
 		AddOnKubernetesDashboard: kubernetes_dashboard.NewDefault(),
 		AddOnFalco:               falco.NewDefault(),
 		AddOnPHPApache:           php_apache.NewDefault(),
+		AddOnNLBGuestbook:        nlb_guestbook.NewDefault(),
 		AddOnNLBHelloWorld:       nlb_hello_world.NewDefault(),
 		AddOnJobsPi:              jobs_pi.NewDefault(),
 		AddOnJobsEcho:            jobs_echo.NewDefault("Job"),
@@ -277,6 +280,11 @@ func (cfg *Config) ValidateAndSetDefaults() error {
 			return err
 		}
 	}
+	if cfg.AddOnNLBGuestbook != nil && cfg.AddOnNLBGuestbook.Enable {
+		if err := cfg.AddOnNLBGuestbook.ValidateAndSetDefaults(); err != nil {
+			return err
+		}
+	}
 	if cfg.AddOnNLBHelloWorld != nil && cfg.AddOnNLBHelloWorld.Enable {
 		if err := cfg.AddOnNLBHelloWorld.ValidateAndSetDefaults(); err != nil {
 			return err
@@ -563,6 +571,16 @@ func (cfg *Config) UpdateFromEnvs() (err error) {
 		}
 	}
 
+	vv, err = parseEnvs(ENV_PREFIX+nlb_guestbook.Env()+"_", cfg.AddOnNLBGuestbook)
+	if err != nil {
+		return err
+	}
+	if av, ok := vv.(*nlb_guestbook.Config); ok {
+		cfg.AddOnNLBGuestbook = av
+	} else {
+		return fmt.Errorf("expected *nlb_guestbook.Config, got %T", vv)
+	}
+
 	vv, err = parseEnvs(ENV_PREFIX+nlb_hello_world.Env()+"_", cfg.AddOnNLBHelloWorld)
 	if err != nil {
 		return err

k8s-tester/config_test.go

@@ -330,6 +330,41 @@ func TestEnvAddOnPHPApache(t *testing.T) {
 	}
 }
 
+func TestEnvAddOnNLBGuestbook(t *testing.T) {
+	cfg := NewDefault()
+
+	os.Setenv("K8S_TESTER_CONFIG_PATH", "test.yaml")
+	defer os.Unsetenv("K8S_TESTER_CONFIG_PATH")
+	os.Setenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ENABLE", "true")
+	defer os.Unsetenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_ENABLE")
+	os.Setenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_MINIMUM_NODES", "100")
+	defer os.Unsetenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_MINIMUM_NODES")
+	os.Setenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_NAMESPACE", "hello")
+	defer os.Unsetenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_NAMESPACE")
+	os.Setenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_DEPLOYMENT_NODE_SELECTOR", `{"a":"b","c":"d"}`)
+	defer os.Unsetenv("K8S_TESTER_ADD_ON_NLB_GUESTBOOK_DEPLOYMENT_NODE_SELECTOR")
+
+	if err := cfg.UpdateFromEnvs(); err != nil {
+		t.Fatal(err)
+	}
+
+	if cfg.ConfigPath != "test.yaml" {
+		t.Fatalf("unexpected cfg.ConfigPath %v", cfg.ConfigPath)
+	}
+	if !cfg.AddOnNLBGuestbook.Enable {
+		t.Fatalf("unexpected cfg.AddOnNLBGuestbook.Enable %v", cfg.AddOnNLBGuestbook.Enable)
+	}
+	if cfg.AddOnNLBGuestbook.MinimumNodes != 100 {
+		t.Fatalf("unexpected cfg.AddOnNLBGuestbook.MinimumNodes %v", cfg.AddOnNLBGuestbook.MinimumNodes)
+	}
+	if cfg.AddOnNLBGuestbook.Namespace != "hello" {
+		t.Fatalf("unexpected cfg.AddOnNLBGuestbook.Namespace %v", cfg.AddOnNLBGuestbook.Namespace)
+	}
+	if !reflect.DeepEqual(cfg.AddOnNLBGuestbook.DeploymentNodeSelector, map[string]string{"a": "b", "c": "d"}) {
+		t.Fatalf("unexpected cfg.AddOnNLBGuestbook.DeploymentNodeSelector %v", cfg.AddOnNLBGuestbook.DeploymentNodeSelector)
+	}
+}
+
 func TestEnvAddOnNLBHelloWorld(t *testing.T) {
 	cfg := NewDefault()
 

k8s-tester/fmt.sh

@@ -27,6 +27,12 @@ gofmt -s -w ./conformance
 goimports -w ./csi-ebs
 gofmt -s -w ./csi-ebs
 
+goimports -w ./csrs
+gofmt -s -w ./csrs
+
+goimports -w ./falco
+gofmt -s -w ./falco
+
 goimports -w ./fluent-bit
 gofmt -s -w ./fluent-bit
 
@@ -45,6 +51,9 @@ gofmt -s -w ./kubernetes-dashboard
 goimports -w ./metrics-server
 gofmt -s -w ./metrics-server
 
+goimports -w ./nlb-guestbook
+gofmt -s -w ./nlb-guestbook
+
 goimports -w ./nlb-hello-world
 gofmt -s -w ./nlb-hello-world
 

k8s-tester/go.mod

@@ -16,6 +16,7 @@ require (
 	github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi v0.0.0-00010101000000-000000000000
 	github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard v0.0.0-00010101000000-000000000000
 	github.com/aws/aws-k8s-tester/k8s-tester/metrics-server v0.0.0-00010101000000-000000000000
+	github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook v0.0.0-00010101000000-000000000000
 	github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world v0.0.0-00010101000000-000000000000
 	github.com/aws/aws-k8s-tester/k8s-tester/php-apache v0.0.0-00010101000000-000000000000
 	github.com/aws/aws-k8s-tester/k8s-tester/secrets v0.0.0-00010101000000-000000000000
@@ -34,7 +35,6 @@ require (
 
 replace (
 	github.com/aws/aws-k8s-tester/client => ../client
-
 	github.com/aws/aws-k8s-tester/k8s-tester/cloudwatch-agent => ./cloudwatch-agent
 	github.com/aws/aws-k8s-tester/k8s-tester/clusterloader => ./clusterloader
 	github.com/aws/aws-k8s-tester/k8s-tester/configmaps => ./configmaps
@@ -48,6 +48,7 @@ replace (
 	github.com/aws/aws-k8s-tester/k8s-tester/jobs-pi => ./jobs-pi
 	github.com/aws/aws-k8s-tester/k8s-tester/kubernetes-dashboard => ./kubernetes-dashboard
 	github.com/aws/aws-k8s-tester/k8s-tester/metrics-server => ./metrics-server
+	github.com/aws/aws-k8s-tester/k8s-tester/nlb-guestbook => ./nlb-guestbook
 	github.com/aws/aws-k8s-tester/k8s-tester/nlb-hello-world => ./nlb-hello-world
 	github.com/aws/aws-k8s-tester/k8s-tester/php-apache => ./php-apache
 	github.com/aws/aws-k8s-tester/k8s-tester/secrets => ./secrets