chore(test): Add test cases for pkg/home/auth.go (#1009)

* add auth tests Signed-off-by: Yves Tumushimire <[email protected]> * fix lint Signed-off-by: Yves Tumushimire <[email protected]> * Trigger Build Signed-off-by: Yves Tumushimire <[email protected]> Signed-off-by: Yves Tumushimire <[email protected]>
tensorchord · Oct 12, 2022 · 46168da · 46168da
1 parent d87afda
commit 46168da
Showing 1 changed file with 68 additions and 0 deletions.
diff --git a/pkg/home/auth_test.go b/pkg/home/auth_test.go
@@ -0,0 +1,68 @@
+// Copyright 2022 The envd Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package home
+
+import (
+	"github.com/tensorchord/envd/pkg/types"
+
+	. "github.com/onsi/ginkgo/v2"
+
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("auth test", func() {
+	defaultAuthName := "auth_name"
+	defaultIdentityToken := "default_token"
+	ac := types.AuthConfig{
+		Name:          defaultAuthName,
+		IdentityToken: defaultIdentityToken,
+	}
+
+	Describe("create with use", Ordered, func() {
+		BeforeAll(func() {
+			err := GetManager().AuthCreate(ac, true)
+			Expect(err).NotTo(HaveOccurred())
+		})
+
+		It("current Auth config should be the new one", func() {
+			authConf, err := GetManager().AuthGetCurrent()
+			Expect(err).NotTo(HaveOccurred())
+			Expect(authConf.Name).To(Equal(ac.Name))
+		})
+
+		AfterAll(func() {
+			Expect(GetManager().AuthUse(defaultAuthName)).To(Succeed())
+			auth, err := GetManager().AuthGetCurrent()
+			Expect(err).NotTo(HaveOccurred())
+			Expect(auth.Name).To(Equal(defaultAuthName))
+		})
+	})
+
+	Describe("create without use", Ordered, func() {
+		BeforeAll(func() {
+			err := GetManager().AuthCreate(ac, false)
+			Expect(err).NotTo(HaveOccurred())
+		})
+
+		It("should not be able to create the same context", func() {
+			err := GetManager().AuthCreate(ac, false)
+			Expect(err).NotTo(HaveOccurred())
+		})
+
+		It("should use the default authConfig", func() {
+			_, err := GetManager().AuthGetCurrent()
+			Expect(err).NotTo(HaveOccurred())
+		})
+	})
+})