mirror of
https://github.com/caddyserver/caddy.git
synced 2024-12-11 12:47:57 +08: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.
104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package websocket
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mholt/caddy"
|
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
|
)
|
|
|
|
func TestWebSocket(t *testing.T) {
|
|
c := caddy.NewTestController("http", `websocket cat`)
|
|
err := setup(c)
|
|
if err != nil {
|
|
t.Errorf("Expected no errors, got: %v", err)
|
|
}
|
|
mids := httpserver.GetConfig(c).Middleware()
|
|
if len(mids) == 0 {
|
|
t.Fatal("Expected middleware, got 0 instead")
|
|
}
|
|
|
|
handler := mids[0](httpserver.EmptyNext)
|
|
myHandler, ok := handler.(WebSocket)
|
|
|
|
if !ok {
|
|
t.Fatalf("Expected handler to be type WebSocket, got: %#v", handler)
|
|
}
|
|
|
|
if myHandler.Sockets[0].Path != "/" {
|
|
t.Errorf("Expected / as the default Path")
|
|
}
|
|
if myHandler.Sockets[0].Command != "cat" {
|
|
t.Errorf("Expected %s as the command", "cat")
|
|
}
|
|
|
|
}
|
|
func TestWebSocketParse(t *testing.T) {
|
|
tests := []struct {
|
|
inputWebSocketConfig string
|
|
shouldErr bool
|
|
expectedWebSocketConfig []Config
|
|
}{
|
|
{`websocket /api1 cat`, false, []Config{{
|
|
Path: "/api1",
|
|
Command: "cat",
|
|
}}},
|
|
|
|
{`websocket /api3 cat
|
|
websocket /api4 cat `, false, []Config{{
|
|
Path: "/api3",
|
|
Command: "cat",
|
|
}, {
|
|
Path: "/api4",
|
|
Command: "cat",
|
|
}}},
|
|
|
|
{`websocket /api5 "cmd arg1 arg2 arg3"`, false, []Config{{
|
|
Path: "/api5",
|
|
Command: "cmd",
|
|
Arguments: []string{"arg1", "arg2", "arg3"},
|
|
}}},
|
|
|
|
// accept respawn
|
|
{`websocket /api6 cat {
|
|
respawn
|
|
}`, false, []Config{{
|
|
Path: "/api6",
|
|
Command: "cat",
|
|
}}},
|
|
|
|
// invalid configuration
|
|
{`websocket /api7 cat {
|
|
invalid
|
|
}`, true, []Config{}},
|
|
}
|
|
for i, test := range tests {
|
|
c := caddy.NewTestController("http", test.inputWebSocketConfig)
|
|
actualWebSocketConfigs, err := webSocketParse(c)
|
|
|
|
if err == nil && test.shouldErr {
|
|
t.Errorf("Test %d didn't error, but it should have", i)
|
|
} else if err != nil && !test.shouldErr {
|
|
t.Errorf("Test %d errored, but it shouldn't have; got '%v'", i, err)
|
|
}
|
|
if len(actualWebSocketConfigs) != len(test.expectedWebSocketConfig) {
|
|
t.Fatalf("Test %d expected %d no of WebSocket configs, but got %d ",
|
|
i, len(test.expectedWebSocketConfig), len(actualWebSocketConfigs))
|
|
}
|
|
for j, actualWebSocketConfig := range actualWebSocketConfigs {
|
|
|
|
if actualWebSocketConfig.Path != test.expectedWebSocketConfig[j].Path {
|
|
t.Errorf("Test %d expected %dth WebSocket Config Path to be %s , but got %s",
|
|
i, j, test.expectedWebSocketConfig[j].Path, actualWebSocketConfig.Path)
|
|
}
|
|
|
|
if actualWebSocketConfig.Command != test.expectedWebSocketConfig[j].Command {
|
|
t.Errorf("Test %d expected %dth WebSocket Config Command to be %s , but got %s",
|
|
i, j, test.expectedWebSocketConfig[j].Command, actualWebSocketConfig.Command)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
}
|