From 34aa031f1e7e4b3efe9ee9a8610d7199389125e2 Mon Sep 17 00:00:00 2001 From: Shulhan Date: Sun, 4 Feb 2024 16:15:10 +0700 Subject: example: split registering HTTP and WebSocket target into separate methods --- example/example.go | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/example/example.go b/example/example.go index f67d94c..128c78c 100644 --- a/example/example.go +++ b/example/example.go @@ -90,8 +90,14 @@ func New() (ex *Example, err error) { return nil, fmt.Errorf("example: New: %w", err) } - // Register targets for testing HTTP and WebSocket endpoints. - err = ex.registerTargets() + // Register target for testing HTTP endpoints. + err = ex.registerTargetHTTP() + if err != nil { + return nil, fmt.Errorf("example: New: %w", err) + } + + // Register target for testing WebSocket endpoints. + err = ex.registerTargetWebSocket() if err != nil { return nil, fmt.Errorf("example: New: %w", err) } @@ -174,7 +180,7 @@ func (ex *Example) registerWebSocketEndpoints() (err error) { return nil } -func (ex *Example) registerTargets() (err error) { +func (ex *Example) registerTargetHTTP() (err error) { var targetHTTP = &trunks.Target{ Name: "Example HTTP", Hint: "This section provide an example of HTTP endpoints that can be tested and attacked.", @@ -323,7 +329,10 @@ func (ex *Example) registerTargets() (err error) { if err != nil { return err } + return nil +} +func (ex *Example) registerTargetWebSocket() (err error) { targetWebSocket := &trunks.Target{ Name: "Example WebSocket", Hint: "This section provide an example of WebSocket endpoints that can be tested.", @@ -598,7 +607,7 @@ func (ex *Example) runWebSocketGet(rr *trunks.RunRequest) (res interface{}, err wsc := &websocket.Client{ Endpoint: "ws://" + websocketAddress, - HandleText: func(cl *websocket.Client, frame *websocket.Frame) error { + HandleText: func(_ *websocket.Client, frame *websocket.Frame) error { res = frame.Payload() wg.Done() return nil -- cgit v1.3