Add test for getJenkinsNames and getCurrentJenkins

app/cmd/config_test.go

+package cmd
+
+import (
+	"github.com/golang/mock/gomock"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("Table util test", func() {
+	var (
+		ctrl *gomock.Controller
+	)
+
+	BeforeEach(func() {
+		ctrl = gomock.NewController(GinkgoT())
+		config = nil
+	})
+
+	AfterEach(func() {
+		config = nil
+		ctrl.Finish()
+	})
+
+	Context("basic test", func() {
+		It("getJenkinsNames", func() {
+			config = &Config{
+				JenkinsServers: []JenkinsServer{JenkinsServer{
+					Name: "a",
+				}, JenkinsServer{
+					Name: "b",
+				}},
+			}
+
+			names := getJenkinsNames()
+			Expect(names).To(Equal([]string{"a", "b"}))
+
+			config.JenkinsServers = []JenkinsServer{}
+			names = getJenkinsNames()
+			Expect(names).To(Equal([]string{}))
+		})
+
+		It("getCurrentJenkins", func() {
+			config = &Config{}
+			current := getCurrentJenkins()
+			Expect(current).To(BeNil())
+
+			config.Current = "test"
+			config.JenkinsServers = []JenkinsServer{JenkinsServer{
+				Name: "test",
+			}}
+			current = getCurrentJenkins()
+			Expect(current).To(Equal(&config.JenkinsServers[0]))
+		})
+	})
+})

app/cmd/root_test.go

@@ -22,9 +22,11 @@ var _ = Describe("Root cmd test", func() {
 		rootCmd = &cobra.Command{Use: "root"}
 		successCmd = "echo 1"
 		errorCmd = "exit 1"
+		config = nil
 	})
 
 	AfterEach(func() {
+		config = nil
 		ctrl.Finish()
 	})
 

util/table_test.go

@@ -106,4 +106,24 @@ var _ = Describe("Table util test", func() {
 			Expect(buffer.String()).To(Equal(comp))
 		})
 	})
+
+	Context("basic function", func() {
+		It("shoud success", func() {
+			var buffer bytes.Buffer
+			table := CreateTable(&buffer)
+
+			table.Render()
+			Expect(buffer.String()).To(Equal(""))
+
+			table.AddRow("fake")
+			buffer.Reset()
+			table.Render()
+			Expect(buffer.String()).To(Equal("fake\n"))
+
+			table.Clear()
+			buffer.Reset()
+			table.Render()
+			Expect(buffer.String()).To(Equal(""))
+		})
+	})
 })