80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package certificate
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/go-acme/lego/v5/acme"
|
|
)
|
|
|
|
type memoryAccountStore struct {
|
|
items map[string]map[string]string
|
|
}
|
|
|
|
func (s *memoryAccountStore) Put(_ context.Context, reference string, value map[string]string) error {
|
|
if s.items == nil {
|
|
s.items = map[string]map[string]string{}
|
|
}
|
|
copyValue := make(map[string]string, len(value))
|
|
for key, item := range value {
|
|
copyValue[key] = item
|
|
}
|
|
s.items[reference] = copyValue
|
|
return nil
|
|
}
|
|
|
|
func (s *memoryAccountStore) Get(_ context.Context, reference string) (map[string]string, error) {
|
|
return s.items[reference], nil
|
|
}
|
|
|
|
func (s *memoryAccountStore) Has(_ context.Context, reference string) (bool, error) {
|
|
_, ok := s.items[reference]
|
|
return ok, nil
|
|
}
|
|
|
|
func TestAccountRepositoryPersistsKeyAndRegistration(t *testing.T) {
|
|
store := &memoryAccountStore{}
|
|
repository := NewAccountRepository(store)
|
|
user, err := repository.LoadOrCreate(context.Background(), "staging", "admin@example.com")
|
|
if err != nil {
|
|
t.Fatalf("创建 ACME 账户失败：%v", err)
|
|
}
|
|
if user.PrivateKey == nil {
|
|
t.Fatal("ACME 账户私钥未生成")
|
|
}
|
|
user.Registration = &acme.ExtendedAccount{
|
|
Account: acme.Account{Status: acme.StatusValid, TermsOfServiceAgreed: true},
|
|
Location: "https://acme.example.test/account/1",
|
|
}
|
|
if err := repository.Save(context.Background(), "staging", user); err != nil {
|
|
t.Fatalf("保存 ACME 账户失败：%v", err)
|
|
}
|
|
loaded, err := repository.LoadOrCreate(context.Background(), "staging", "")
|
|
if err != nil {
|
|
t.Fatalf("重新读取 ACME 账户失败：%v", err)
|
|
}
|
|
if loaded.Email != "admin@example.com" {
|
|
t.Fatalf("ACME 账户邮箱错误：%s", loaded.Email)
|
|
}
|
|
if loaded.Registration == nil || loaded.Registration.Location != user.Registration.Location {
|
|
t.Fatalf("ACME 注册状态未持久化：%+v", loaded.Registration)
|
|
}
|
|
if loaded.PrivateKey == nil {
|
|
t.Fatal("ACME 账户私钥未恢复")
|
|
}
|
|
}
|
|
|
|
func TestAccountRepositorySeparatesEnvironments(t *testing.T) {
|
|
store := &memoryAccountStore{}
|
|
repository := NewAccountRepository(store)
|
|
if _, err := repository.LoadOrCreate(context.Background(), "staging", ""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := repository.LoadOrCreate(context.Background(), "production", ""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(store.items) != 2 {
|
|
t.Fatalf("测试环境与生产环境应使用独立账户：%d", len(store.items))
|
|
}
|
|
}
|