63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package payment
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
)
|
|
|
|
func TestProvideEncryptionKeySkipsAutoGeneratedTotpKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cfg := &config.Config{
|
|
Totp: config.TotpConfig{
|
|
EncryptionKey: strings.Repeat("a", 64),
|
|
EncryptionKeyConfigured: false,
|
|
},
|
|
}
|
|
|
|
key, err := ProvideEncryptionKey(cfg)
|
|
if err != nil {
|
|
t.Fatalf("ProvideEncryptionKey returned error: %v", err)
|
|
}
|
|
if len(key) != 0 {
|
|
t.Fatalf("encryption key len = %d, want 0", len(key))
|
|
}
|
|
}
|
|
|
|
func TestProvideEncryptionKeyUsesConfiguredTotpKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cfg := &config.Config{
|
|
Totp: config.TotpConfig{
|
|
EncryptionKey: "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef",
|
|
EncryptionKeyConfigured: true,
|
|
},
|
|
}
|
|
|
|
key, err := ProvideEncryptionKey(cfg)
|
|
if err != nil {
|
|
t.Fatalf("ProvideEncryptionKey returned error: %v", err)
|
|
}
|
|
if len(key) != 32 {
|
|
t.Fatalf("encryption key len = %d, want 32", len(key))
|
|
}
|
|
}
|
|
|
|
func TestProvideEncryptionKeyRejectsConfiguredInvalidLength(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cfg := &config.Config{
|
|
Totp: config.TotpConfig{
|
|
EncryptionKey: "abcd",
|
|
EncryptionKeyConfigured: true,
|
|
},
|
|
}
|
|
|
|
_, err := ProvideEncryptionKey(cfg)
|
|
if err == nil {
|
|
t.Fatal("expected error for invalid key length")
|
|
}
|
|
}
|