From 10266a6b722c6bfedb1cee2ad83c20e1f456f24a Mon Sep 17 00:00:00 2001
From: Ilya Shaisultanov <ilya.shaisultanov@gmail.com>
Date: Fri, 31 Jan 2020 15:35:11 +0100
Subject: [PATCH] Add EnvSettings.SetNamespace method

Signed-off-by: Ilya Shaisultanov <ilya.shaisultanov@gmail.com>
---
 pkg/cli/environment.go      | 5 +++++
 pkg/cli/environment_test.go | 7 +++++++
 2 files changed, 12 insertions(+)

diff --git a/pkg/cli/environment.go b/pkg/cli/environment.go
index 5f947aec7..070b66fa4 100644
--- a/pkg/cli/environment.go
+++ b/pkg/cli/environment.go
@@ -121,6 +121,11 @@ func (s *EnvSettings) Namespace() string {
 	return "default"
 }
 
+//SetNamespace sets the namespace
+func (s *EnvSettings) SetNamespace(ns string) {
+	s.namespace = ns
+}
+
 //RESTClientGetter gets the kubeconfig from EnvSettings
 func (s *EnvSettings) RESTClientGetter() genericclioptions.RESTClientGetter {
 	s.configOnce.Do(func() {
diff --git a/pkg/cli/environment_test.go b/pkg/cli/environment_test.go
index fadc2981e..885c05bcc 100644
--- a/pkg/cli/environment_test.go
+++ b/pkg/cli/environment_test.go
@@ -86,6 +86,13 @@ func TestEnvSettings(t *testing.T) {
 			}
 		})
 	}
+
+	settings := New()
+	settings.SetNamespace("custom-namespace")
+
+	if settings.Namespace() != "custom-namespace" {
+		t.Errorf("expected namespace %s, got %s", "custom-namespace", settings.Namespace())
+	}
 }
 
 func resetEnv() func() {