mirror of
https://github.com/caddyserver/caddy.git
synced 2025-01-06 10:58:49 +03:00
a798e0c951
- Server types no longer need to store their own contexts; they are stored on the caddy.Instance, which means each context will be properly GC'ed when the instance is stopped. Server types should use type assertions to convert from caddy.Context to their concrete context type when they need to use it. - Pass the entire context into httpserver.GetConfig instead of only the Key field. - caddy.NewTestController now requires a server type string so it can create a controller with the proper concrete context associated with that server type. Tests still need more attention so that we can test the proper creation of startup functions, etc.
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package startupshutdown
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mholt/caddy"
|
|
)
|
|
|
|
// The Startup function's tests are symmetrical to Shutdown tests,
|
|
// because the Startup and Shutdown functions share virtually the
|
|
// same functionality
|
|
func TestStartup(t *testing.T) {
|
|
tempDirPath := os.TempDir()
|
|
|
|
testDir := filepath.Join(tempDirPath, "temp_dir_for_testing_startupshutdown")
|
|
defer func() {
|
|
// clean up after non-blocking startup function quits
|
|
time.Sleep(500 * time.Millisecond)
|
|
os.RemoveAll(testDir)
|
|
}()
|
|
osSenitiveTestDir := filepath.FromSlash(testDir)
|
|
os.RemoveAll(osSenitiveTestDir) // start with a clean slate
|
|
|
|
var registeredFunction func() error
|
|
fakeRegister := func(fn func() error) {
|
|
registeredFunction = fn
|
|
}
|
|
|
|
tests := []struct {
|
|
input string
|
|
shouldExecutionErr bool
|
|
shouldRemoveErr bool
|
|
}{
|
|
// test case #0 tests proper functionality blocking commands
|
|
{"startup mkdir " + osSenitiveTestDir, false, false},
|
|
|
|
// test case #1 tests proper functionality of non-blocking commands
|
|
{"startup mkdir " + osSenitiveTestDir + " &", false, true},
|
|
|
|
// test case #2 tests handling of non-existent commands
|
|
{"startup " + strconv.Itoa(int(time.Now().UnixNano())), true, true},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
c := caddy.NewTestController("http", test.input)
|
|
err := registerCallback(c, fakeRegister)
|
|
if err != nil {
|
|
t.Errorf("Expected no errors, got: %v", err)
|
|
}
|
|
if registeredFunction == nil {
|
|
t.Fatalf("Expected function to be registered, but it wasn't")
|
|
}
|
|
err = registeredFunction()
|
|
if err != nil && !test.shouldExecutionErr {
|
|
t.Errorf("Test %d recieved an error of:\n%v", i, err)
|
|
}
|
|
err = os.Remove(osSenitiveTestDir)
|
|
if err != nil && !test.shouldRemoveErr {
|
|
t.Errorf("Test %d recieved an error of:\n%v", i, err)
|
|
}
|
|
}
|
|
}
|