架构师训练营 11 周作业

package sha
import (
	"crypto/sha256"
	"encoding/hex"
	"fmt"
)
const SALT  = "geeklangzhe"
var  DB  =  make(map[string]string)
func Register(userId, password string)  {
	password = Encode(password)
	WriteDB(userId, password)
}
func Login(userId, password string)  bool {
	password = Encode(password)
	dbPassword := ReadDB(userId)
	return password==dbPassword
}
func Encode(str string) string {
	s1 := sha256.Sum256([]byte(str+SALT))
	s := hex.EncodeToString(s1[:])
	fmt.Println(s)
	return s
}
func WriteDB(userId, password string) {
	DB[userId]=password
}
func ReadDB(userId string) string  {
	return DB[userId]
}
package sha
import (
	"testing"
)
func TestEncode(t *testing.T) {
	type args struct {
		str string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{"1", args{"lang"}, "de90ffc82aa7b263680efc54e111bda228f368174ece54af102f593710dd4451"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := Encode(tt.args.str); got != tt.want {
				t.Errorf("Encode() = %v, want %v", got, tt.want)
			}
		})
	}
}
func TestLogin(t *testing.T) {
	Register("langzhe","geek")
	type args struct {
		userId   string
		password string
	}
	tests := []struct {
		name string
		args args
		want bool
	}{
		{"1",args{"langzhe","geek"},true},
		{"2",args{"langzhe1","geek"},false},
		{"3",args{"langzhe","geek1"},false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := Login(tt.args.userId, tt.args.password); got != tt.want {
				t.Errorf("Login() = %v, want %v", got, tt.want)
			}
		})
	}
}