add more test cases

Signed-off-by: Max Peng <[email protected]>

Author: pengisgood

key_test.go

@@ -11,26 +11,66 @@ func TestKeyGen(t *testing.T) {
 	is := is.New(t)
 	s := NewShell(shellUrl)
 
-	key, err := s.KeyGen(context.Background(), "testKey", KeyGen.Type("ed25519"), KeyGen.Size(2048))
+	defer func() {
+		_, err := s.KeyRm(context.Background(), "testKey1")
+		is.Nil(err)
+	}()
+	key1, err := s.KeyGen(context.Background(), "testKey1", KeyGen.Type("ed25519"), KeyGen.Size(2048))
 	is.Nil(err)
-
-	is.Equal(key.Name, "testKey")
-	is.NotNil(key.Id)
-
-	_, err = s.KeyRm(context.Background(), "testKey")
+	is.Equal(key1.Name, "testKey1")
+	is.NotNil(key1.Id)
+
+	defer func() {
+		_, err = s.KeyRm(context.Background(), "testKey2")
+		is.Nil(err)
+	}()
+	key2, err := s.KeyGen(context.Background(), "testKey2", KeyGen.Type("ed25519"))
+	is.Nil(err)
+	is.Equal(key2.Name, "testKey2")
+	is.NotNil(key2.Id)
+
+	defer func() {
+		_, err = s.KeyRm(context.Background(), "testKey3")
+		is.Nil(err)
+	}()
+	key3, err := s.KeyGen(context.Background(), "testKey3", KeyGen.Type("rsa"))
 	is.Nil(err)
+	is.Equal(key3.Name, "testKey3")
+	is.NotNil(key3.Id)
+
+	defer func() {
+		_, err = s.KeyRm(context.Background(), "testKey4")
+		is.Nil(err)
+	}()
+	key4, err := s.KeyGen(context.Background(), "testKey4", KeyGen.Type("rsa"), KeyGen.Size(4096))
+	is.Nil(err)
+	is.Equal(key4.Name, "testKey4")
+	is.NotNil(key4.Id)
+
+	_, err = s.KeyGen(context.Background(), "testKey5", KeyGen.Type("rsa"), KeyGen.Size(1024))
+	is.NotNil(err)
+	is.Equal(err.Error(), "key/gen: rsa keys must be >= 2048 bits to be useful")
 }
 
 func TestKeyList(t *testing.T) {
 	is := is.New(t)
 	s := NewShell(shellUrl)
 
+	defer func() {
+		_, err := s.KeyRm(context.Background(), "testKey")
+		is.Nil(err)
+	}()
+	key, err := s.KeyGen(context.Background(), "testKey")
+	is.Nil(err)
+
 	keys, err := s.KeyList(context.Background())
 	is.Nil(err)
 
-	is.Equal(len(keys), 1)
+	is.Equal(len(keys), 2)
 	is.Equal(keys[0].Name, "self")
 	is.NotNil(keys[0].Id)
+	is.NotNil(keys[1].Id, key.Id)
+	is.NotNil(keys[1].Name, key.Name)
 }
 
 func TestKeyRename(t *testing.T) {
@@ -50,6 +90,10 @@ func TestKeyRename(t *testing.T) {
 
 	_, err = s.KeyRm(context.Background(), "test2")
 	is.Nil(err)
+
+	_, err = s.KeyRename(context.Background(), "test2", "test1", false)
+	is.NotNil(err)
+	is.Equal(err.Error(), "key/rename: no key named test2 was found")
 }
 
 func TestKeyRm(t *testing.T) {
@@ -63,6 +107,10 @@ func TestKeyRm(t *testing.T) {
 	is.Nil(err)
 
 	is.Equal(len(keys), 1)
-	is.Equal(keys[0].Name, "testKey")
+	is.Equal(keys[0].Name, key.Name)
 	is.Equal(keys[0].Id, key.Id)
+
+	_, err = s.KeyRm(context.Background(), "testKey")
+	is.NotNil(err)
+	is.Equal(err.Error(), "key/rm: no key named testKey was found")
 }