package utils

import (
	"testing"
)

func TestAES(t *testing.T) {
	origin := "foo"
	key := "@MKU^AAOPH%%$#VJ<%G&j*o#$XZSWQ@L"

	encrypted, err := AESEncrypt(origin, key)
	if err != nil {
		t.Fatal(err)
	}

	if encrypted != "gp4ij9IRlDNQLDNQFwPOhg==" {
		t.Fatal("加密密文不正确")
	}

	decrypted, err := AESDecrypt(encrypted, key)
	if err != nil {
		t.Fatal(err)
	}

	if decrypted != origin {
		t.Fatal("解密内容和加密内容不一致")
	}
}

func TestMD5(t *testing.T) {
	origin := "foo"
	salt := "foo-salt"

	md5Str := MD5(origin, "")

	if md5Str != "666f6fd41d8cd98f00b204e9800998ecf8427e" {
		t.Fatal("md5字符串不正确")
	}

	md5WithSaltStr := MD5(origin, salt)

	if md5WithSaltStr != "666f6f666f6f2d73616c74d41d8cd98f00b204e9800998ecf8427e" {
		t.Fatal("md5字符串不正确")
	}
}

func TestSHA256(t *testing.T) {
	origin := "foo"
	salt := "foo-salt"

	sha256Str := SHA256(origin, "")

	if sha256Str != "2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae" {
		t.Fatal("sha256字符串不正确")
	}

	sha256WithSaltStr := SHA256(origin, salt)

	if sha256WithSaltStr != "941189487b27649bf72f232b199c9b859d5e7b26f162f2efe201a9c7a12ce5a2" {
		t.Fatal("sha256字符串不正确")
	}
}