fix(_tests): get keys tests working

Author: mboersma

_tests/keys_test.go

@@ -2,6 +2,8 @@ package _tests_test
 
 import (
 	"os"
+	"os/user"
+	"path"
 
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
@@ -10,38 +12,40 @@ import (
 var _ = Describe("Keys", func() {
 	key := testUser
 
-	BeforeEach(func() {
-		addKey(key)
-	})
-
-	AfterEach(func() {
-		// remove the generated ssh key
-		key := testUser
-		execute("deis keys:remove %s", key)
-		err := os.Remove(key)
-		Expect(err).NotTo(HaveOccurred())
-		err = os.Remove(key + ".pub")
-		Expect(err).NotTo(HaveOccurred())
-	})
+	Context("when logged in as a normal user", func() {
+		BeforeEach(func() {
+			login(url, testUser, testPassword)
+			createKey(testUser)
+		})
 
-	Context("with a new key", func() {
+		AfterEach(func() {
+			// remove the generated ssh key
+			var home string
+			if user, err := user.Current(); err != nil {
+				home = "~"
+			} else {
+				home = user.HomeDir
+			}
+			path := path.Join(home, ".ssh", testUser)
+			err := os.Remove(path)
+			Expect(err).NotTo(HaveOccurred())
+			err = os.Remove(path + ".pub")
+			Expect(err).NotTo(HaveOccurred())
+		})
 
-		It("can add the key", func() {
-			output, err := execute("deis keys:add %s.pub", key)
+		It("can add, list, and remove a key", func() {
+			output, err := execute("deis keys:add ~/.ssh/%s.pub", key)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(output).To(ContainSubstring("Uploading %s.pub to deis... done", key))
 			output, err = execute("deis keys:list")
 			Expect(err).NotTo(HaveOccurred())
-			Expect(output).To(ContainSubstring("%s@deis.com", key))
-		})
-
-		It("can remove the key", func() {
-			output, err := execute("deis keys:remove %s@deis.com", key)
+			Expect(output).To(ContainSubstring("%s ssh-rsa", key))
+			output, err = execute("deis keys:remove %s", key)
 			Expect(err).NotTo(HaveOccurred())
-			Expect(output).To(ContainSubstring("Removing %s@deis.com SSH Key... done", key))
+			Expect(output).To(ContainSubstring("Removing %s SSH Key... done", key))
 			output, err = execute("deis keys")
 			Expect(err).NotTo(HaveOccurred())
-			Expect(output).NotTo(ContainSubstring("%s@deis.com", key))
+			Expect(output).NotTo(ContainSubstring("%s ssh-rsa", key))
 		})
 	})
 })

_tests/tests_suite_test.go

@@ -53,7 +53,10 @@ var _ = BeforeSuite(func() {
 
 	// register the test user and add a key
 	register(url, testUser, testPassword, testEmail)
-	addKey("deis-test")
+	createKey("deis-test")
+	output, err := execute("deis keys:add ~/.ssh/deis-test.pub")
+	Expect(err).NotTo(HaveOccurred())
+	Expect(output).To(ContainSubstring("Uploading deis-test.pub to deis... done"))
 })
 
 var _ = AfterSuite(func() {
@@ -108,7 +111,7 @@ func execute(cmdLine string, args ...interface{}) (string, error) {
 	return stdout.String(), nil
 }
 
-func addKey(name string) {
+func createKey(name string) {
 	var home string
 	if user, err := user.Current(); err != nil {
 		home = "~"
@@ -118,16 +121,13 @@ func addKey(name string) {
 	path := path.Join(home, ".ssh", name)
 	// create the key under ~/.ssh/<name> if it doesn't already exist
 	if _, err := os.Stat(path); os.IsNotExist(err) {
-		cmd := "ssh-keygen -q -t rsa -b 4096 -C otto.test@deis.com -f %s -N ''"
-		_, err := execute(cmd, path)
+		cmd := "ssh-keygen -q -t rsa -b 4096 -C %s -f %s -N ''"
+		_, err := execute(cmd, name, path)
 		Expect(err).NotTo(HaveOccurred())
 	}
 	// add the key to ssh-agent
 	_, err := execute("eval $(ssh-agent) && ssh-add %s", path)
 	Expect(err).NotTo(HaveOccurred())
-	// add the public key to deis (assumes the user is logged in)
-	_, err = execute("deis keys:add %s.pub", path)
-	Expect(err).NotTo(HaveOccurred())
 }
 
 func getController() string {