package security

import (
	"encoding/base64"
	"fmt"
	"reflect"
	"testing"
)

func TestGenAesKey256(t *testing.T) {
	for range 10 {
		key := GenAesKey256()
		fmt.Println(key)
	}
}

// 参考：https://www.yisu.com/zixun/696240.html
func TestAes(t *testing.T) {
	defer func() {
		if r := recover(); r != nil {
			fmt.Println("recover...", r.(error).Error(), reflect.TypeOf(r))
		}
	}()

	// aesKeyStr := GetAesKey256()
	aesKeyStr := "VzKw8Vx+K8k1nO9fmPjhv8o+8l4vqpF+fjsbvXf0j4o="

	fmt.Println(aesKeyStr)
	key, _ := base64.StdEncoding.DecodeString(aesKeyStr)

	text := []byte("hello sig.pub")
	encrypt, _ := EncryptAesCBC(text, key)
	fmt.Println(base64.URLEncoding.EncodeToString(text))

	result, err := DecryptAesCBC(encrypt, key)
	if err != nil {
		t.Errorf("decrypt aes cbc failed: %v", err)
		return
	}
	fmt.Println("result:", string(result))
}