add test

TRON-US · Dec 3, 2019 · a7ad774 · a7ad774
1 parent 701f21a
commit a7ad774
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/crypto/crypto_test.go b/crypto/crypto_test.go
@@ -8,6 +8,7 @@ import (
 
 const (
 	KeyString = "CAISIJFNZZd5ZSvi9OlJP/mz/vvUobvlrr2//QN4DzX/EShP"
+	EncryptKey = "Tron2theMoon1234"
 )
 
 func TestSignVerify(t *testing.T) {
@@ -44,3 +45,13 @@ func TestSignVerify(t *testing.T) {
 		t.Error("Verify with public key failed")
 	}
 }
+
+func TestEncryptDecrypt(t *testing.T) {
+	origin := "Hello World"
+	key := []byte(EncryptKey)
+	encryptMsg, _ := Encrypt(key, []byte(origin))
+	msg, _ := Decrypt(key, []byte(encryptMsg))
+	if string(msg) != origin {
+		t.Errorf("Decrypt failed")
+	}
+}