优化结构

pkg/websocket/melody_test.go

@@ -48,3 +48,15 @@ func TestServerReceivesAndSendsTextMessages(t *testing.T) {
 		t.Fatal("first message handler was overwritten")
 	}
 }
+
+func TestServerAllowsAnyOriginByDefault(t *testing.T) {
+	server := New(Config{})
+	defer server.Close()
+
+	request := httptest.NewRequest(http.MethodGet, "http://127.0.0.1:8000/ws", nil)
+	request.Host = "127.0.0.1:8000"
+	request.Header.Set("Origin", "http://evil.example")
+	if !server.Melody().Upgrader.CheckOrigin(request) {
+		t.Fatal("origin was rejected when allow_origins is empty")
+	}
+}