From acf672efcd74ae1ee75c41bfe1a92ebf10cea83e Mon Sep 17 00:00:00 2001
From: "valexz@gmail.com" <avyssochin@gmail.com>
Date: Sun, 8 Oct 2023 19:56:41 +0500
Subject: [PATCH] test(cmd/root):Added support for kubernetes provider config

---
 cmd/root.go                          | 4 ++--
 cmd/root_test.go                     | 2 ++
 cmd/testdata/config.env              | 2 ++
 cmd/testdata/config.yml              | 3 +++
 cmd/testdata/config_cli_wanted.json  | 6 +++++-
 cmd/testdata/config_env_wanted.json  | 6 +++++-
 cmd/testdata/config_yaml_wanted.json | 6 +++++-
 7 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index 46b2cc7..576b75b 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -46,10 +46,10 @@ It provides an integrations with multiple reverse proxies and different loading
 	// Provider flags
 	startCmd.Flags().StringVar(&conf.Provider.Name, "provider.name", "docker", fmt.Sprintf("Provider to use to manage containers %v", config.GetProviders()))
 	viper.BindPFlag("provider.name", startCmd.Flags().Lookup("provider.name"))
-	startCmd.Flags().Float32Var(&conf.Provider.Kubernetes.QPS, "provider.kubernetes.QPS", 5, fmt.Sprintf("QPS limit for K8S API access client-side throttling"))
+	startCmd.Flags().Float32Var(&conf.Provider.Kubernetes.QPS, "provider.kubernetes.qps", 5, fmt.Sprintf("QPS limit for K8S API access client-side throttling"))
 	viper.BindPFlag("provider.kubernetes.qps", startCmd.Flags().Lookup("provider.kubernetes.qps"))
 	startCmd.Flags().IntVar(&conf.Provider.Kubernetes.Burst, "provider.kubernetes.burst", 10, fmt.Sprintf("Maximum burst for K8S API acees client-side throttling"))
-	viper.BindPFlag("provider.kubernetes.burst", startCmd.Flags().Lookup("provider.kubernetes.Burst"))
+	viper.BindPFlag("provider.kubernetes.burst", startCmd.Flags().Lookup("provider.kubernetes.burst"))
 	// Server flags
 	startCmd.Flags().IntVar(&conf.Server.Port, "server.port", 10000, "The server port to use")
 	viper.BindPFlag("server.port", startCmd.Flags().Lookup("server.port"))
diff --git a/cmd/root_test.go b/cmd/root_test.go
index e3cdc1e..5b22fe7 100644
--- a/cmd/root_test.go
+++ b/cmd/root_test.go
@@ -80,6 +80,8 @@ func TestPrecedence(t *testing.T) {
 			"--configFile", filepath.Join(testDir, "testdata", "config.yml"),
 			"start",
 			"--provider.name", "cli",
+			"--provider.kubernetes.qps", "256",
+			"--provider.kubernetes.burst", "512",
 			"--server.port", "3333",
 			"--server.base-path", "/cli/",
 			"--storage.file", "/tmp/cli.json",
diff --git a/cmd/testdata/config.env b/cmd/testdata/config.env
index 05468ba..be4bd08 100644
--- a/cmd/testdata/config.env
+++ b/cmd/testdata/config.env
@@ -1,4 +1,6 @@
 PROVIDER_NAME=envvar
+PROVIDER_KUBERNETES_QPS=16
+PROVIDER_KUBERNETES_BURST=32
 SERVER_PORT=2222
 SERVER_BASE_PATH=/envvar/
 STORAGE_FILE=/tmp/envvar.json
diff --git a/cmd/testdata/config.yml b/cmd/testdata/config.yml
index 683ae90..a23bdfc 100644
--- a/cmd/testdata/config.yml
+++ b/cmd/testdata/config.yml
@@ -1,5 +1,8 @@
 provider:
   name: configfile
+  kubernetes:
+    qps: 64
+    burst: 128
 server:
   port: 1111
   base-path: /configfile/
diff --git a/cmd/testdata/config_cli_wanted.json b/cmd/testdata/config_cli_wanted.json
index 9827aa5..cea8b24 100644
--- a/cmd/testdata/config_cli_wanted.json
+++ b/cmd/testdata/config_cli_wanted.json
@@ -7,7 +7,11 @@
     "File": "/tmp/cli.json"
   },
   "Provider": {
-    "Name": "cli"
+    "Name": "cli",
+    "Kubernetes": {
+      "QPS": 256,
+      "Burst": 512
+    }
   },
   "Sessions": {
     "DefaultDuration": 10800000000000,
diff --git a/cmd/testdata/config_env_wanted.json b/cmd/testdata/config_env_wanted.json
index 12c031e..3b5afa7 100644
--- a/cmd/testdata/config_env_wanted.json
+++ b/cmd/testdata/config_env_wanted.json
@@ -7,7 +7,11 @@
     "File": "/tmp/envvar.json"
   },
   "Provider": {
-    "Name": "envvar"
+    "Name": "envvar",
+    "Kubernetes": {
+      "QPS": 16,
+      "Burst": 32
+    }
   },
   "Sessions": {
     "DefaultDuration": 7200000000000,
diff --git a/cmd/testdata/config_yaml_wanted.json b/cmd/testdata/config_yaml_wanted.json
index cec2578..bfa9eeb 100644
--- a/cmd/testdata/config_yaml_wanted.json
+++ b/cmd/testdata/config_yaml_wanted.json
@@ -7,7 +7,11 @@
     "File": "/tmp/configfile.json"
   },
   "Provider": {
-    "Name": "configfile"
+    "Name": "configfile",
+    "Kubernetes": {
+      "QPS": 64,
+      "Burst": 128
+    }
   },
   "Sessions": {
     "DefaultDuration": 3600000000000,