mirror of
https://github.com/caddyserver/caddy.git
synced 2024-12-23 22:27:38 -05:00
88a2811e2a
* Initial concept for pluggable storage (sans tests and docs) * Add TLS storage docs, test harness, and minor clean up from code review * Fix issue with caddymain's temporary moveStorage * Formatting improvement on struct array literal by removing struct name * Pluggable storage changes: * Change storage interface to persist all site or user data in one call * Add lock/unlock calls for renewal and cert obtaining * Key fields on composite literals
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package caddytls
|
|
|
|
import (
|
|
"errors"
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestStorageForNoURL(t *testing.T) {
|
|
c := &Config{}
|
|
if _, err := c.StorageFor(""); err == nil {
|
|
t.Fatal("Expected error on empty URL")
|
|
}
|
|
}
|
|
|
|
func TestStorageForLowercasesAndPrefixesScheme(t *testing.T) {
|
|
resultStr := ""
|
|
c := &Config{
|
|
StorageCreator: func(caURL *url.URL) (Storage, error) {
|
|
resultStr = caURL.String()
|
|
return nil, nil
|
|
},
|
|
}
|
|
if _, err := c.StorageFor("EXAMPLE.COM/BLAH"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resultStr != "https://example.com/blah" {
|
|
t.Fatalf("Unexpected CA URL string: %v", resultStr)
|
|
}
|
|
}
|
|
|
|
func TestStorageForBadURL(t *testing.T) {
|
|
c := &Config{}
|
|
if _, err := c.StorageFor("http://192.168.0.%31/"); err == nil {
|
|
t.Fatal("Expected error for bad URL")
|
|
}
|
|
}
|
|
|
|
func TestStorageForDefault(t *testing.T) {
|
|
c := &Config{}
|
|
s, err := c.StorageFor("example.com")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if reflect.TypeOf(s).Name() != "FileStorage" {
|
|
t.Fatalf("Unexpected storage type: %v", reflect.TypeOf(s).Name())
|
|
}
|
|
}
|
|
|
|
func TestStorageForCustom(t *testing.T) {
|
|
storage := fakeStorage("fake")
|
|
c := &Config{
|
|
StorageCreator: func(caURL *url.URL) (Storage, error) {
|
|
return storage, nil
|
|
},
|
|
}
|
|
s, err := c.StorageFor("example.com")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if s != storage {
|
|
t.Fatal("Unexpected storage")
|
|
}
|
|
}
|
|
|
|
func TestStorageForCustomError(t *testing.T) {
|
|
c := &Config{
|
|
StorageCreator: func(caURL *url.URL) (Storage, error) {
|
|
return nil, errors.New("some error")
|
|
},
|
|
}
|
|
if _, err := c.StorageFor("example.com"); err == nil {
|
|
t.Fatal("Expecting error")
|
|
}
|
|
}
|
|
|
|
func TestStorageForCustomNil(t *testing.T) {
|
|
// Should fall through to the default
|
|
c := &Config{
|
|
StorageCreator: func(caURL *url.URL) (Storage, error) {
|
|
return nil, nil
|
|
},
|
|
}
|
|
s, err := c.StorageFor("example.com")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if reflect.TypeOf(s).Name() != "FileStorage" {
|
|
t.Fatalf("Unexpected storage type: %v", reflect.TypeOf(s).Name())
|
|
}
|
|
}
|
|
|
|
type fakeStorage string
|
|
|
|
func (s fakeStorage) SiteExists(domain string) bool {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) LoadSite(domain string) (*SiteData, error) {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) StoreSite(domain string, data *SiteData) error {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) DeleteSite(domain string) error {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) LockRegister(domain string) (bool, error) {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) UnlockRegister(domain string) error {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) LoadUser(email string) (*UserData, error) {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) StoreUser(email string, data *UserData) error {
|
|
panic("no impl")
|
|
}
|
|
|
|
func (s fakeStorage) MostRecentUserEmail() string {
|
|
panic("no impl")
|
|
}
|