mirror of
https://github.com/restic/rest-server.git
synced 2025-10-19 15:43:21 +00:00
1f593fafaf
This contains all the glue to make Server use the new repo.Handler: - Remove all old handlers - Add ServeHTTP to make Server a single http.Handler - Remove Goji routing and replace by net/http and custom routing logic Additionally, this implements two-level backup repositories.
121 lines
3 KiB
Go
121 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
restserver "github.com/restic/rest-server"
|
|
)
|
|
|
|
func TestTLSSettings(t *testing.T) {
|
|
type expected struct {
|
|
TLSKey string
|
|
TLSCert string
|
|
Error bool
|
|
}
|
|
type passed struct {
|
|
Path string
|
|
TLS bool
|
|
TLSKey string
|
|
TLSCert string
|
|
}
|
|
|
|
var tests = []struct {
|
|
passed passed
|
|
expected expected
|
|
}{
|
|
{passed{TLS: false}, expected{"", "", false}},
|
|
{passed{TLS: true}, expected{"/tmp/restic/private_key", "/tmp/restic/public_key", false}},
|
|
{passed{Path: "/tmp", TLS: true}, expected{"/tmp/private_key", "/tmp/public_key", false}},
|
|
{passed{Path: "/tmp", TLS: true, TLSKey: "/etc/restic/key", TLSCert: "/etc/restic/cert"}, expected{"/etc/restic/key", "/etc/restic/cert", false}},
|
|
{passed{Path: "/tmp", TLS: false, TLSKey: "/etc/restic/key", TLSCert: "/etc/restic/cert"}, expected{"", "", true}},
|
|
{passed{Path: "/tmp", TLS: false, TLSKey: "/etc/restic/key"}, expected{"", "", true}},
|
|
{passed{Path: "/tmp", TLS: false, TLSCert: "/etc/restic/cert"}, expected{"", "", true}},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
|
|
t.Run("", func(t *testing.T) {
|
|
// defer func() { restserver.Server = defaultConfig }()
|
|
if test.passed.Path != "" {
|
|
server.Path = test.passed.Path
|
|
}
|
|
server.TLS = test.passed.TLS
|
|
server.TLSKey = test.passed.TLSKey
|
|
server.TLSCert = test.passed.TLSCert
|
|
|
|
gotTLS, gotKey, gotCert, err := tlsSettings()
|
|
if err != nil && !test.expected.Error {
|
|
t.Fatalf("tls_settings returned err (%v)", err)
|
|
}
|
|
if test.expected.Error {
|
|
if err == nil {
|
|
t.Fatalf("Error not returned properly (%v)", test)
|
|
} else {
|
|
return
|
|
}
|
|
}
|
|
if gotTLS != test.passed.TLS {
|
|
t.Errorf("TLS enabled, want (%v), got (%v)", test.passed.TLS, gotTLS)
|
|
}
|
|
wantKey := test.expected.TLSKey
|
|
if gotKey != wantKey {
|
|
t.Errorf("wrong TLSPrivPath path, want (%v), got (%v)", wantKey, gotKey)
|
|
}
|
|
|
|
wantCert := test.expected.TLSCert
|
|
if gotCert != wantCert {
|
|
t.Errorf("wrong TLSCertPath path, want (%v), got (%v)", wantCert, gotCert)
|
|
}
|
|
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetHandler(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "rest-server-test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
err := os.Remove(dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
getHandler := restserver.NewHandler
|
|
|
|
// With NoAuth = false and no .htpasswd
|
|
_, err = getHandler(&restserver.Server{Path: dir})
|
|
if err == nil {
|
|
t.Errorf("NoAuth=false: expected error, got nil")
|
|
}
|
|
|
|
// With NoAuth = true and no .htpasswd
|
|
_, err = getHandler(&restserver.Server{NoAuth: true, Path: dir})
|
|
if err != nil {
|
|
t.Errorf("NoAuth=true: expected no error, got %v", err)
|
|
}
|
|
|
|
// Create .htpasswd
|
|
htpasswd := filepath.Join(dir, ".htpasswd")
|
|
err = ioutil.WriteFile(htpasswd, []byte(""), 0644)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() {
|
|
err := os.Remove(htpasswd)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
// With NoAuth = false and with .htpasswd
|
|
_, err = getHandler(&restserver.Server{Path: dir})
|
|
if err != nil {
|
|
t.Errorf("NoAuth=false with .htpasswd: expected no error, got %v", err)
|
|
}
|
|
}
|