Go WebSocket の単体テストを行うにはどうすればよいですか?-Golang-php.cn

Go WebSocket の単体テストを行うにはどうすればよいですか?

WBOY

リリース： 2024-06-01 22:16:00

オリジナル

1191 人が閲覧しました

Go WebSocket 単体テストでは、WebSocket サーバーの正確性と堅牢性を検証できます。 Ginkgo および Gomega テストフレームワークをインストールします。 WebSocket サーバーを作成して起動します。単体テストの作成: WebSocket 接続の確立。メッセージを送受信します。応答を確認します。テストを実行してサーバーの機能を確認します。

Go WebSocket 如何进行单元测试？

Go WebSocket 単体テストガイド

WebSocket は、クライアントとサーバーの間に双方向の永続的な接続を確立するネットワークプロトコルです。 Go では、[gorilla/websocket](https://godoc.org/github.com/gorilla/websocket) は、WebSocket 接続を作成および処理するための一般的なライブラリです。単体テストは、コードの正確さと堅牢性を検証するため、WebSocket サーバーを作成する場合には不可欠です。この記事では、Ginkgo と Gomega を使用して Go WebSocket の単体テストを行うためのステップバイステップのガイドを、実践的な例とともに提供します。

ステップ 1: テストフレームワークをインストールします

Go モジュールを使用して依存関係を管理し、次のコマンドで Ginkgo と Gomega をインストールします:

go get -u github.com/onsi/ginkgo/v2
go get -u github.com/onsi/gomega

ログイン後にコピー

ステップ 2: WebSocket サーバーを作成する

新しい Go モジュールを作成して書き込みます単純な WebSocket サーバー、例:

package main

import (
    "github.com/gorilla/websocket"
    "log"
    "net/http"
)

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        conn, err := websocket.Upgrade(w, r, nil, 1024, 1024)
        if err != nil {
            log.Println(err)
            http.Error(w, "Internal Server Error", http.StatusInternalServerError)
            return
        }
        log.Printf("WebSocket connection established")

        // Handle incoming messages and send responses...
    })

    log.Println("Listening on port 8080")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Println(err)
    }
}

ログイン後にコピー

ステップ 3: 単体テストを作成する

Ginkgo および Gomega フレームワークを使用して単体テストを作成します。例:

package main_test

import (
    "bytes"
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/gorilla/websocket"
    "github.com/onsi/ginkgo/v2"
    "github.com/onsi/gomega"
)

var _ = ginkgo.Describe("WebSocket Server", func() {        
    var (
        ts         *httptest.Server
        clientConn *websocket.Conn
    )
    
    ginkgo.BeforeEach(func() {
        ts = httptest.NewServer(http.HandlerFunc(wsHandler))
        var err error    
        clientConn, _, err = websocket.DefaultDialer.Dial(ts.URL, nil)
        gomega.Expect(err).To(gomega.BeNil())
    })
    
    ginkgo.AfterEach(func() {
        ts.Close()
        clientConn.Close()
    })
    
    ginkgo.It("should establish a WebSocket connection", func() {
        gomega.Expect(clientConn).NotTo(gomega.BeNil())
    })
    
    ginkgo.It("should send and receive messages", func() {
        // Send a message to the server
        err := clientConn.WriteMessage(websocket.TextMessage, []byte("Hello WebSocket!"))
        gomega.Expect(err).To(gomega.BeNil())

        // Read the response from the server
        _, msg, err := clientConn.ReadMessage()
        gomega.Expect(err).To(gomega.BeNil())

        // Verify the response 
        gomega.Expect(string(msg)).To(gomega.Equal("Hello Client!"))
    })
})

ログイン後にコピー

実用的なケース

この例は、Ginkgo の使用方法を示しています。 Gomega は、WebSocket サーバーの基本機能をテストして、接続を確立し、メッセージを送受信します。これらのテストを拡張して、認証、接続制限、エラー処理などの他の使用例を検証できます。

注: