example: split registering HTTP and WebSocket target into separate methods

1 files 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