ウェブソケット
WebSockets
この章では、アプリケーションを改善するためにWebSocketを使用する方法を学びます。

プロジェクトの要約

ポーカーコードベースには2つのアプリケーションがあります
  • コマンドラインアプリ。ゲームのプレーヤー数を入力するようにユーザーに求めます。それ以降は、「ブラインドベット」の値が何であるかをプレイヤーに知らせます。ユーザーはいつでも" {Playername} wins"を入力してゲームを終了し、ストアに勝利者を記録できます。
  • Webアプリ。ユーザーがゲームの勝者を記録し、リーグテーブルを表示できるようにします。コマンドラインアプリと同じストアを共有します。

次のステップ

製品の所有者はコマンドラインアプリケーションに興奮していますが、ブラウザーにその機能を提供できればそれを好むでしょう。彼女は、ユーザーがプレーヤーの数を入力できるテキストボックスを備えたWebページを想像し、ユーザーがフォームを送信すると、ページにブラインド値が表示され、必要に応じて自動的に更新されます。コマンドラインアプリケーションと同様に、ユーザーは勝者を宣言でき、データベースに保存されます。
一見、それは非常に単純に聞こえますが、いつものように、ソフトウェアを書くために反復的アプローチを取ることを強調しなければなりません。
まず、HTMLを提供する必要があります。 これまでのところ、すべてのHTTPエンドポイントはプレーンテキストまたはJSONを返しています。(最終的にはすべて文字列なので)知っているのと同じ手法を使用できますが、よりクリーンなhtml/templateパッケージを使用することもできます解決。
また、ブラウザーを更新せずに、「ブラインドが *y* になりました」というメッセージを非同期でユーザーに送信できる必要もあります。これを容易にするために、WebSocketsを使用できます。
WebSocketはコンピューター通信プロトコルであり、単一のTCP接続で全二重通信チャネルを提供します。
多くの手法を採用しているため、最初に可能な限り最小限の有用な作業を行ってから繰り返すことがさらに重要です。
そのため、最初に行うことは、ユーザーが勝者を記録するためのフォームを含むWebページを作成することです。 プレーンフォームを使用するのではなく、WebSocketを使用して、そのデータをサーバーに送信して記録します。
その後、ブラインドアラートに取り組みます。 その時点で、インフラストラクチャコードが少し設定されます。

JavaScriptのテストについてはどうですか?

これを行うために記述されたJavaScriptがいくつかありますが、テストの記述まで行いません。
ごめんなさい。 O'Reillyに「テストでJavaScriptを学ぶ」を作ってくれるようにロビーをお願いします。

最初にテストを書く

まず、ユーザーが「/game」を押したときにHTMLを提供する必要があります。
ここに私たちのウェブサーバーの関連コードのリマインダーがあります。
1
type PlayerServer struct {
2
store PlayerStore
3
http.Handler
4
}
5
6
const jsonContentType = "application/json"
7
8
func NewPlayerServer(store PlayerStore) *PlayerServer {
9
p := new(PlayerServer)
10
11
p.store = store
12
13
router := http.NewServeMux()
14
router.Handle("/league", http.HandlerFunc(p.leagueHandler))
15
router.Handle("/players/", http.HandlerFunc(p.playersHandler))
16
17
p.Handler = router
18
19
return p
20
}
Copied!
ここでできる最も簡単なことは、「GET /game」を実行したときに「200」が得られることを確認することです。
1
func TestGame(t *testing.T) {
2
t.Run("GET /game returns 200", func(t *testing.T) {
3
server := NewPlayerServer(&StubPlayerStore{})
4
5
request, _ := http.NewRequest(http.MethodGet, "/game", nil)
6
response := httptest.NewRecorder()
7
8
server.ServeHTTP(response, request)
9
10
assertStatus(t, response.Code, http.StatusOK)
11
})
12
}
Copied!

テストを実行してみます

1
--- FAIL: TestGame (0.00s)
2
=== RUN TestGame/GET_/game_returns_200
3
--- FAIL: TestGame/GET_/game_returns_200 (0.00s)
4
server_test.go:109: did not get correct status, got 404, want 200
Copied!

成功させるのに十分なコードを書く

私たちのサーバーにはルーターの設定があるので、修正は比較的簡単です。
私たちのルーターに追加します。
1
router.Handle("/game", http.HandlerFunc(p.game))
Copied!
そしてgameメソッドを書きます
1
func (p *PlayerServer) game(w http.ResponseWriter, r *http.Request) {
2
w.WriteHeader(http.StatusOK)
3
}
Copied!

リファクタリング♪

サーバーコードは、既存の適切に因数分解されたコードに非常に簡単にコードを追加できるため、既に問題ありません。
/gameにリクエストを送信するためのテストヘルパー関数newGameRequestを追加することで、テストを少し片付けることができます。これを自分で書いてみてください。
1
func TestGame(t *testing.T) {
2
t.Run("GET /game returns 200", func(t *testing.T) {
3
server := NewPlayerServer(&StubPlayerStore{})
4
5
request := newGameRequest()
6
response := httptest.NewRecorder()
7
8
server.ServeHTTP(response, request)
9
10
assertStatus(t, response, http.StatusOK)
11
})
12
}
Copied!
また、「assertStatus」を変更して、「response.Code」ではなく「response」を受け入れるように変更しました。
次に、エンドポイントにHTMLを返すようにする必要があります。
1
<!DOCTYPE html>
2
<html lang="en">
3
<head>
4
<meta charset="UTF-8">
5
<title>Let's play poker</title>
6
</head>
7
<body>
8
<section id="game">
9
<div id="declare-winner">
10
<label for="winner">Winner</label>
11
<input type="text" id="winner"/>
12
<button id="winner-button">Declare winner</button>
13
</div>
14
</section>
15
</body>
16
<script type="application/javascript">
17
18
const submitWinnerButton = document.getElementById('winner-button')
19
const winnerInput = document.getElementById('winner')
20
21
if (window['WebSocket']) {
22
const conn = new WebSocket('ws://' + document.location.host + '/ws')
23
24
submitWinnerButton.onclick = event => {
25
conn.send(winnerInput.value)
26
}
27
}
28
</script>
29
</html>
Copied!
とてもシンプルなウェブページがあります。
  • ユーザーが勝者を入力するためのテキスト入力
  • 彼らが勝者を宣言するためにクリックできるボタン。
  • サーバーへのWebSocket接続を開き、送信ボタンが押されたときに処理するためのJavaScriptをいくつか用意しました。
WebSocketはほとんどの最新のブラウザーに組み込まれているので、ライブラリーの持ち込みについて心配する必要はありません。 Webページは古いブラウザーでは機能しませんが、このシナリオではそれで問題ありません。

正しいマークアップを返すことをテストするにはどうすればよいですか?

いくつかの方法があります。本全体を通して強調されているように、あなたが書くテストはコストを正当化するのに十分な価値を持つことが重要です。
  1. 1.
    Seleniumなどを使用して、ブラウザベースのテストを記述します。これらのテストは、ある種の実際のWebブラウザーを起動して、ユーザーとの対話をシミュレートするため、すべてのアプローチの中で最も「現実的な」ものです。これらのテストは、システムの動作に大きな自信を与えることができますが、単体テストよりも作成が難しく、実行速度がはるかに遅くなります。私たちの製品の目的上、これはやりすぎです。
  2. 2.
    文字列を完全に一致させます。これは大丈夫ですが、この種のテストは非常に壊れやすくなります。誰かがマークアップを変更した瞬間に、実際には何も実際に壊れていない場合、テストは失敗します。
  3. 3.
    正しいテンプレートを呼び出すことを確認します。標準のlibのテンプレートライブラリを使用してHTMLを提供します(すぐに説明します)。thing を挿入してHTMLを生成し、その呼び出しをスパイして、正しく実行されていることを確認できます。これはコードの設計に影響を与えますが、実際にはあまりテストしていません。正しいテンプレートファイルを使用して呼び出す場合を除きます。プロジェクトにはテンプレートが1つしかないため、ここで失敗する可能性は低いようです。
そこで、「テスト駆動開発でGO言語を学びましょう」というサイトの中で、初めてテストを書くことになります。
マークアップを「game.html」というファイルに入れます
次に、先ほど書き込んだエンドポイントを次のように変更します
1
func (p *PlayerServer) game(w http.ResponseWriter, r *http.Request) {
2
tmpl, err := template.ParseFiles("game.html")
3
4
if err != nil {
5
http.Error(w, fmt.Sprintf("problem loading template %s", err.Error()), http.StatusInternalServerError)
6
return
7
}
8
9
tmpl.Execute(w, nil)
10
}
Copied!
html/templateは、HTMLを作成するためのGoパッケージです。 私たちの場合、template.ParseFilesを呼び出して、htmlファイルのパスを指定します。 エラーがなければ、テンプレートを「実行Execute」して、それを「io.Writer」に書き込みます。この例では、インターネットに「書き込みWrite」したいので、「http.ResponseWriter」を指定します。
テストを作成していないので、希望どおりに機能していることを確認するためだけにWebサーバーを手動でテストするのが賢明です。 cmd/webserverに移動し、main.goファイルを実行します。http://localhost:5000/gameにアクセスしてください。
テンプレートが見つからないというエラーが発生するはずです。フォルダーに相対的なパスに変更するか、cmd/webserverディレクトリにgame.htmlのコピーを置くことができます。プロジェクトのルート内のファイルへのシンボリックリンク(ln -s ../../game.html game.html)を作成することを選択したので、変更を加えると、サーバーの実行時に変更が反映されます。
この変更を加えて再度実行すると、UIが表示されます。
次に、サーバーへのWebSocket接続を介して文字列を取得したときに、ゲームの勝者として宣言することをテストする必要があります。

最初にテストを書く

初めて、WebSocketで作業できるように外部ライブラリを使用します。
go get github.com/gorilla/websocketを実行します
これにより、優れたGorilla WebSocket ライブラリのコードがフェッチされます。これで、新しい要件に合わせてテストを更新できます。
1
t.Run("when we get a message over a websocket it is a winner of a game", func(t *testing.T) {
2
store := &StubPlayerStore{}
3
winner := "Ruth"
4
server := httptest.NewServer(NewPlayerServer(store))
5
defer server.Close()
6
7
wsURL := "ws" + strings.TrimPrefix(server.URL, "http") + "/ws"
8
9
ws, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
10
if err != nil {
11
t.Fatalf("could not open a ws connection on %s %v", wsURL, err)
12
}
13
defer ws.Close()
14
15
if err := ws.WriteMessage(websocket.TextMessage, []byte(winner)); err != nil {
16
t.Fatalf("could not send message over ws connection %v", err)
17
}
18
19
AssertPlayerWin(t, store, winner)
20
})
Copied!
websocketライブラリのインポートがあることを確認してください。 私のIDEはそれを自動的に行いました。
ブラウザーから何が起こるかをテストするには、独自のWebSocket接続を開いて、それに書き込む必要があります。
サーバーに関する以前のテストではサーバーのメソッドを呼び出しましたが、サーバーへの永続的な接続が必要です。そのためには、http.Handlerを受け取り、それを起動して接続を待機するhttptest.NewServerを使用します。
websocket.DefaultDialer.Dialを使用してサーバーにダイヤルインしてから、winnerでメッセージを送信しようとします。
最後に、勝者が記録されたことを確認するためにプレイヤーストアを評価します。

テストを実行してみます

1
=== RUN TestGame/when_we_get_a_message_over_a_websocket_it_is_a_winner_of_a_game
2
--- FAIL: TestGame/when_we_get_a_message_over_a_websocket_it_is_a_winner_of_a_game (0.00s)
3
server_test.go:124: could not open a ws connection on ws://127.0.0.1:55838/ws websocket: bad handshake
Copied!
/wsでWebSocket接続を受け入れるようにサーバーを変更していないため、まだ握手はしていません。

成功させるのに十分なコードを書く

ルーターに別のリストを追加する
1
router.Handle("/ws", http.HandlerFunc(p.webSocket))
Copied!
次に、新しい「webSocket」ハンドラを追加します
1
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
2
upgrader := websocket.Upgrader{
3
ReadBufferSize: 1024,
4
WriteBufferSize: 1024,
5
}
6
upgrader.Upgrade(w, r, nil)
7
}
Copied!
WebSocket接続を受け入れるには、リクエストを「Upgrade」します。ここでテストを再実行する場合は、次のエラーに進む必要があります。
1
=== RUN TestGame/when_we_get_a_message_over_a_websocket_it_is_a_winner_of_a_game
2
--- FAIL: TestGame/when_we_get_a_message_over_a_websocket_it_is_a_winner_of_a_game (0.00s)
3
server_test.go:132: got 0 calls to RecordWin want 1
Copied!
接続を開いたので、メッセージをリッスンして、それを勝者として記録します。
1
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
2
upgrader := websocket.Upgrader{
3
ReadBufferSize: 1024,
4
WriteBufferSize: 1024,
5
}
6
conn, _ := upgrader.Upgrade(w, r, nil)
7
_, winnerMsg, _ := conn.ReadMessage()
8
p.store.RecordWin(string(winnerMsg))
9
}
Copied!
(はい、現在、多くのエラーを無視しています!)
conn.ReadMessage()は、接続でのメッセージの待機をブロックします。取得したら、それを RecordWinに使用します。これは最終的にWebSocket接続を閉じます。
テストを実行すると、まだ失敗します。
問題はタイミングです。メッセージを読み取るWebSocket接続と勝利の記録の間には遅延があり、テストが完了する前にテストが終了します。これをテストするには、最後のアサーションの前に短い「time.Sleep」を配置します。
とりあえずそれを続けましょうが、任意のスリープをテストに入れることは非常に悪い習慣であることを認めます。
1
time.Sleep(10 * time.Millisecond)
2
AssertPlayerWin(t, store, winner)
Copied!

リファクタリング♪

このテストをサーバーコードとテストコードの両方で機能させるために多くの罪を犯しましたが、これが私たちにとって最も簡単な方法であることを覚えておいてください。
テストに裏打ちされた、厄介で恐ろしい working ソフトウェアがあるので、これを自由に変更して、誤って何も壊さないようにすることができます。
サーバーコードから始めましょう。
すべてのWebSocket接続リクエストで再宣言する必要がないため、upgraderをパッケージ内のプライベート値に移動できます。
1
var wsUpgrader = websocket.Upgrader{
2
ReadBufferSize: 1024,
3
WriteBufferSize: 1024,
4
}
5
6
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
7
conn, _ := wsUpgrader.Upgrade(w, r, nil)
8
_, winnerMsg, _ := conn.ReadMessage()
9
p.store.RecordWin(string(winnerMsg))
10
}
Copied!
template.ParseFiles("game.html")への呼び出しは、すべての GET /gameで実行されます。 つまり、テンプレートを再解析する必要がない場合でも、すべてのリクエストでファイルシステムに移動します。代わりにテンプレートを「NewPlayerServer」で一度解析するようにコードをリファクタリングしましょう。 ディスクからテンプレートをフェッチしたり解析したりするときに問題が発生した場合に、この関数がエラーを返すことができるようにする必要があります。
PlayerServerに関連する変更は次のとおりです
1
type PlayerServer struct {
2
store PlayerStore
3
http.Handler
4
template *template.Template
5
}
6
7
const htmlTemplatePath = "game.html"
8
9
func NewPlayerServer(store PlayerStore) (*PlayerServer, error) {
10
p := new(PlayerServer)
11
12
tmpl, err := template.ParseFiles(htmlTemplatePath)
13
14
if err != nil {
15
return nil, fmt.Errorf("problem opening %s %v", htmlTemplatePath, err)
16
}
17
18
p.template = tmpl
19
p.store = store
20
21
router := http.NewServeMux()
22
router.Handle("/league", http.HandlerFunc(p.leagueHandler))
23
router.Handle("/players/", http.HandlerFunc(p.playersHandler))
24
router.Handle("/game", http.HandlerFunc(p.game))
25
router.Handle("/ws", http.HandlerFunc(p.webSocket))
26
27
p.Handler = router
28
29
return p, nil
30
}
31
32
func (p *PlayerServer) game(w http.ResponseWriter, r *http.Request) {
33
p.template.Execute(w, nil)
34
}
Copied!
NewPlayerServerのシグネチャを変更すると、コンパイルの問題が発生します。自分で試して修正するか、苦労している場合はソースコードを参照してください。
テストコードでは、 mustMakePlayerServer(t *testing.T, store PlayerStore) *PlayerServerというヘルパーを作成して、エラーノイズをテストから隠せるようにしました。
1
func mustMakePlayerServer(t *testing.T, store PlayerStore) *PlayerServer {
2
server, err := NewPlayerServer(store)
3
if err != nil {
4
t.Fatal("problem creating player server", err)
5
}
6
return server
7
}
Copied!
同様に、別のヘルパーmustDialWSを作成して、WebSocket接続の作成時に厄介なエラーノイズを隠すことができるようにしました。
1
func mustDialWS(t *testing.T, url string) *websocket.Conn {
2
ws, _, err := websocket.DefaultDialer.Dial(url, nil)
3
4
if err != nil {
5
t.Fatalf("could not open a ws connection on %s %v", url, err)
6
}
7
8
return ws
9
}
Copied!
最後に、テストコードで、メッセージの送信を整理するヘルパーを作成できます。
1
func writeWSMessage(t *testing.T, conn *websocket.Conn, message string) {
2
t.Helper()
3
if err := conn.WriteMessage(websocket.TextMessage, []byte(message)); err != nil {
4
t.Fatalf("could not send message over ws connection %v", err)
5
}
6
}
Copied!
テストがパスし、サーバーを実行して、/gameで勝者を宣言します。/leagueに記録されているはずです。当選者を獲得するたびに、接続を閉じる ことを思い出してください。接続を再度開くには、ページを更新する必要があります。
ユーザーがゲームの勝者を記録できる簡単なWebフォームを作成しました。これを繰り返して、ユーザーが多数のプレーヤーを提供することでゲームを開始できるようにします。サーバーはクライアントにメッセージをプッシュし、時間の経過とともにブラインドの値を通知します。
まず、game.htmlを更新して、新しい要件に合わせてクライアント側のコードを更新します
1
<!DOCTYPE html>
2
<html lang="en">
3
<head>
4
<meta charset="UTF-8">
5
<title>Lets play poker</title>
6
</head>
7
<body>
8
<section id="game">
9
<div id="game-start">
10
<label for="player-count">Number of players</label>
11
<input type="number" id="player-count"/>
12
<button id="start-game">Start</button>
13
</div>
14
15
<div id="declare-winner">
16
<label for="winner">Winner</label>
17
<input type="text" id="winner"/>
18
<button id="winner-button">Declare winner</button>
19
</div>
20
21
<div id="blind-value"/>
22
</section>
23
24
<section id="game-end">
25
<h1>Another great game of poker everyone!</h1>
26
<p><a href="/league">Go check the league table</a></p>
27
</section>
28
29
</body>
30
<script type="application/javascript">
31
const startGame = document.getElementById('game-start')
32
33
const declareWinner = document.getElementById('declare-winner')
34
const submitWinnerButton = document.getElementById('winner-button')
35
const winnerInput = document.getElementById('winner')
36
37
const blindContainer = document.getElementById('blind-value')
38
39
const gameContainer = document.getElementById('game')
40
const gameEndContainer = document.getElementById('game-end')
41
42
declareWinner.hidden = true
43
gameEndContainer.hidden = true
44
45
document.getElementById('start-game').addEventListener('click', event => {
46
startGame.hidden = true
47
declareWinner.hidden = false
48
49
const numberOfPlayers = document.getElementById('player-count').value
50
51
if (window['WebSocket']) {
52
const conn = new WebSocket('ws://' + document.location.host + '/ws')
53
54
submitWinnerButton.onclick = event => {
55
conn.send(winnerInput.value)
56
gameEndContainer.hidden = false
57
gameContainer.hidden = true
58
}
59
60
conn.onclose = evt => {
61
blindContainer.innerText = 'Connection closed'
62
}
63
64
conn.onmessage = evt => {
65
blindContainer.innerText = evt.data
66
}
67
68
conn.onopen = function () {
69
conn.send(numberOfPlayers)
70
}
71
}
72
})
73
</script>
74
</html>
Copied!
主な変更点は、プレーヤー数を入力するセクションとブラインド値を表示するセクションを取り込むことです。ゲームのステージに応じて、ユーザーインターフェイスを表示/非表示にするための小さなロジックがあります。
conn.onmessageを介して受信するメッセージはすべてブラインドアラートであると想定するため、それに応じてblindContainer.innerTextを設定します。
ブラインドアラートを送信するにはどうすればよいですか?前の章では、Gameのアイデアを紹介しました。これにより、CLIコードがGameを呼び出すことができるようになり、ブラインドアラートのスケジュール設定など、他のすべての処理が行われます。これは、心配事の良い分離であることがわかりました。
1
type Game interface {
2
Start(numberOfPlayers int)
3
Finish(winner string)
4
}
Copied!
ユーザーがCLIでプレーヤーの数を求められると、ゲームを「開始Start」してブラインドアラートを開始し、ユーザーが勝者を宣言すると「終了Finish」します。これは現在の要件と同じですが、入力を取得する方法が異なります。したがって、可能であれば、この概念を再利用する必要があります。
Gameの「実際の」実装は「TexasHoldem」です。
1
type TexasHoldem struct {
2
alerter BlindAlerter
3
store PlayerStore
4
}
Copied!
BlindAlerterを送信することで、TexasHoldemwherever に送信されるブラインドアラートをスケジュールできます
1
type BlindAlerter interface {
2
ScheduleAlertAt(duration time.Duration, amount int)
3
}
Copied!
また、注意点として、CLIで使用する BlindAlerter の実装を以下に示します。
1
func StdOutAlerter(duration time.Duration, amount int) {
2
time.AfterFunc(duration, func() {
3
fmt.Fprintf(os.Stdout, "Blind is now %d\n", amount)
4
})
5
}
Copied!
常にアラートを「os.Stdout」に送信したいので、これはCLIで機能しますが、これはWebサーバーでは機能しません。リクエストごとに新しいhttp.ResponseWriterを取得し、それを*websocket.Connにアップグレードします。 したがって、アラートを送信する必要がある依存関係を構築するときに、それを知ることはできません。
そのため、 BlindAlerter.ScheduleAlertAtを変更してアラートの宛先を取得し、Webサーバーで再利用できるようにする必要があります。
BlindAlerter.goを開き、パラメータ「to io.Writer」を追加します
1
type BlindAlerter interface {
2
ScheduleAlertAt(duration time.Duration, amount int, to io.Writer)
3
}
4
5
type BlindAlerterFunc func(duration time.Duration, amount int, to io.Writer)
6
7
func (a BlindAlerterFunc) ScheduleAlertAt(duration time.Duration, amount int, to io.Writer) {
8
a(duration, amount, to)
9
}
Copied!
StdoutAlerterのアイデアは新しいモデルに適合しないため、名前をAlerterに変更します
1
func Alerter(duration time.Duration, amount int, to io.Writer) {
2
time.AfterFunc(duration, func() {
3
fmt.Fprintf(to, "Blind is now %d\n", amount)
4
})
5
}
Copied!
コンパイルしてみると、宛先なしでScheduleAlertAtを呼び出しているため、TexasHoldemで失敗します。これにより、「今すぐ」コンパイルして、os.Stdoutにハードコードします。
テストを実行してみてください。
SpyBlindAlerterBlindAlerterを実装していないため、テストは失敗します。 これを修正するには、ScheduleAlertAtのシグネチャを更新し、テストを実行します。これで緑色のままになります。
TexasHoldemがブラインドアラートの送信先を知っていることは意味がありません。 ここで、Gameを更新して、ゲームを開始するときにアラートの送信先をどこに宣言するようにします。
1
type Game interface {
2
Start(numberOfPlayers int, alertsDestination io.Writer)
3
Finish(winner string)
4
}
Copied!
修正する必要があることをコンパイラーに指示させます。変更はそれほど悪くありません。
  • TexasHoldemを更新して、Gameを適切に実装します。
  • ゲームを開始するときに、CLIoutプロパティを渡します(cli.game.Start(numberOfPlayers、cli.out)
  • TexasHoldemのテストでは、game.Start(5、ioutil.Discard)を使用してコンパイルの問題を修正し、アラート出力が破棄されるように設定します
すべてが正しければ、すべてがグリーンになるはずです。 これで、Server内でGameを試して使用できます。

最初にテストを書く

CLIServerの要件は同じです!配信メカニズムが異なるだけです。
インスピレーションを得るために、 CLIテストを見てみましょう。
1
t.Run("start game with 3 players and finish game with 'Chris' as winner", func(t *testing.T) {
2
game := &GameSpy{}
3
4
out := &bytes.Buffer{}
5
in := userSends("3", "Chris wins")
6
7
poker.NewCLI(in, out, game).PlayPoker()
8
9
assertMessagesSentToUser(t, out, poker.PlayerPrompt)
10
assertGameStartedWith(t, game, 3)
11
assertFinishCalledWith(t, game, "Chris")
12
})
Copied!
GameSpyを使用して、同様の結果をテストすることができるはずです
古いwebsocketテストを次のものに置き換えます
1
t.Run("start a game with 3 players and declare Ruth the winner", func(t *testing.T) {
2
game := &poker.GameSpy{}
3
winner := "Ruth"
4
server := httptest.NewServer(mustMakePlayerServer(t, dummyPlayerStore, game))
5
ws := mustDialWS(t, "ws"+strings.TrimPrefix(server.URL, "http")+"/ws")
6
7
defer server.Close()
8
defer ws.Close()
9
10
writeWSMessage(t, ws, "3")
11
writeWSMessage(t, ws, winner)
12
13
time.Sleep(10 * time.Millisecond)
14
assertGameStartedWith(t, game, 3)
15
assertFinishCalledWith(t, game, winner)
16
})
Copied!
  • 議論したように、スパイのGameを作成してmustMakePlayerServerに渡します(これをサポートするためにヘルパーを更新してください)
  • 次に、ゲームのWebソケットメッセージを送信します。
  • 最後に、ゲームが開始され、期待どおりに終了したと断言します。

テストを実行してみます

他のテストでは、mustMakePlayerServerの周りにいくつかのコンパイルエラーが発生します。エクスポートされていない変数dummyGameを導入し、コンパイルされていないすべてのテストで使用します
1
var (
2
dummyGame = &GameSpy{}
3
)
Copied!
最後のエラーは、GameNewPlayerServerに渡そうとしているところですが、まだサポートしていません。
1
./server_test.go:21:38: too many arguments in call to "github.com/quii/learn-go-with-tests/WebSockets/v2".NewPlayerServer
2
have ("github.com/quii/learn-go-with-tests/WebSockets/v2".PlayerStore, "github.com/quii/learn-go-with-tests/WebSockets/v2".Game)
3
want ("github.com/quii/learn-go-with-tests/WebSockets/v2".PlayerStore)
Copied!

テストを実行するための最小限のコードを記述し、失敗したテスト出力を確認します

今のところ、テストを実行するために引数として追加するだけです
1
func NewPlayerServer(store PlayerStore, game Game) (*PlayerServer, error) {
Copied!
最終的に!
1
=== RUN TestGame/start_a_game_with_3_players_and_declare_Ruth_the_winner
2
--- FAIL: TestGame (0.01s)
3
--- FAIL: TestGame/start_a_game_with_3_players_and_declare_Ruth_the_winner (0.01s)
4
server_test.go:146: wanted Start called with 3 but got 0
5
server_test.go:147: expected finish called with 'Ruth' but got ''
6
FAIL
Copied!

成功させるのに十分なコードを書く

リクエストを受け取ったときに使用できるように、フィールドとしてGamePlayerServerに追加する必要があります。
1
type PlayerServer struct {
2
store PlayerStore
3
http.Handler
4
template *template.Template
5
game Game
6
}
Copied!
(すでにgameというメソッドがあるので、名前をplayGameに変更します)
次に、コンストラクタに割り当てます
1
func NewPlayerServer(store PlayerStore, game Game) (*PlayerServer, error) {
2
p := new(PlayerServer)
3
4
tmpl, err := template.ParseFiles(htmlTemplatePath)
5
6
if err != nil {
7
return nil, fmt.Errorf("problem opening %s %v", htmlTemplatePath, err)
8
}
9
10
p.game = game
11
12
// etc
Copied!
これで、webSocket内でGameを使用できます。
1
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
2
conn, _ := wsUpgrader.Upgrade(w, r, nil)
3
4
_, numberOfPlayersMsg, _ := conn.ReadMessage()
5
numberOfPlayers, _ := strconv.Atoi(string(numberOfPlayersMsg))
6
p.game.Start(numberOfPlayers, ioutil.Discard) //todo: Don't discard the blinds messages!
7
8
_, winner, _ := conn.ReadMessage()
9
p.game.Finish(string(winner))
10
}
Copied!
やったー!テストに成功しました。
それについて考える必要があるので、「まだ」どこにもブラインドメッセージを送信するつもりはありません。game.Startを呼び出すと、それに書き込まれたメッセージを破棄するioutil.Discardを送信します。
ここでは、Webサーバーを起動します。GamePlayerServerに渡すには、main.goを更新する必要があります
1
func main() {
2
db, err := os.OpenFile(dbFileName, os.O_RDWR|os.O_CREATE, 0666)
3
4
if err != nil {
5
log.Fatalf("problem opening %s %v", dbFileName, err)
6
}
7
8
store, err := poker.NewFileSystemPlayerStore(db)
9
10
if err != nil {
11
log.Fatalf("problem creating file system player store, %v ", err)
12
}
13
14
game := poker.NewTexasHoldem(poker.BlindAlerterFunc(poker.Alerter), store)
15
16
server, err := poker.NewPlayerServer(store, game)
17
18
if err != nil {
19
log.Fatalf("problem creating player server %v", err)
20
}
21
22
if err := http.ListenAndServe(":5000", server); err != nil {
23
log.Fatalf("could not listen on port 5000 %v", err)
24
}
25
}
Copied!
ブラインドアラートをまだ取得していないという事実を無視して、アプリは機能します! PlayerServerGameを再利用することに成功し、すべての詳細を処理しました。ブラインドアラートを破棄するのではなく、Webソケットに送信する方法を見つけたら、すべて正常に機能するはずです。
その前に、いくつかのコードを整理しましょう。

リファクタリング♪

WebSocketを使用する方法はかなり基本的で、エラー処理はかなり単純なので、サーバーコードからその煩雑さを取り除くためだけにタイプにカプセル化したかったのです。後でもう一度見たいと思うかもしれませんが、今のところこれは少し整理されます
1
type playerServerWS struct {
2
*websocket.Conn
3
}
4
5
func newPlayerServerWS(w http.ResponseWriter, r *http.Request) *playerServerWS {
6
conn, err := wsUpgrader.Upgrade(w, r, nil)
7
8
if err != nil {
9
log.Printf("problem upgrading connection to WebSockets %v\n", err)
10
}
11
12
return &playerServerWS{conn}
13
}
14
15
func (w *playerServerWS) WaitForMsg() string {
16
_, msg, err := w.ReadMessage()
17
if err != nil {
18
log.Printf("error reading from websocket %v\n", err)
19
}
20
return string(msg)
21
}
Copied!
今サーバーコードは少し簡略化されています
1
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
2
ws := newPlayerServerWS(w, r)
3
4
numberOfPlayersMsg := ws.WaitForMsg()
5
numberOfPlayers, _ := strconv.Atoi(numberOfPlayersMsg)
6
p.game.Start(numberOfPlayers, ioutil.Discard) //todo: Don't discard the blinds messages!
7
8
winner := ws.WaitForMsg()
9
p.game.Finish(winner)
10
}
Copied!
ブラインドメッセージを破棄しない方法を見つけたら、完了です。

テストを書かないようにしよう!

どうしたらよいかわからないときは、遊んで試してみるのが一番です。作業が最初にコミットされていることを確認してください。方法を見つけたら、テストを実行する必要があります。
私たちが持っている問題のあるコード行は
1
p.game.Start(numberOfPlayers, ioutil.Discard) //todo: Don't discard the blinds messages!
Copied!
ゲームがブラインドアラートを書き込むには、io.Writerを渡す必要があります。
以前からplayerServerWSを渡せたらいいですね。これはWebSocketのラッパーなので、 Gameに送信してメッセージを送信できるように感じます。
試してみてね
1
func (p *PlayerServer) webSocket(w http.ResponseWriter, r *http.Request) {
2
ws := newPlayerServerWS(w, r)
3
4
numberOfPlayersMsg := ws.WaitForMsg()
5
numberOfPlayers, _ := strconv.Atoi(numberOfPlayersMsg)
6
p.game.Start(numberOfPlayers, ws)
7
//etc...
Copied!
コンパイラは以下のような問題を指摘しています。
1
./server.go:71:14: cannot use ws (type *playerServerWS) as type io.Writer in argument to p.game.Start:
2
*playerServerWS does not implement io.Writer (missing Write method)
Copied!
当然のことのようですが、playerServerWSdoesio.Writerを実装するようにすることです。そのためには、基になる*websocket.Connを使用して、WriteMessageを使用し、メッセージをWebSocketに送信します。
1
func (w *playerServerWS) Write(p []byte) (n int, err error) {
2
err = w.WriteMessage(websocket.TextMessage, p)
3
4
if err != nil {
5
return 0, err
6
}
7
8
return len(p), nil
9
}
Copied!
これは簡単すぎるようです!アプリケーションを試して実行し、機能するかどうかを確認します。
事前にTexasHoldemを編集して、ブラインドインクリメント時間を短くし、実際に動作するようにしてください
1
blindIncrement := time.Duration(5+numberOfPlayers) * time.Second // (rather than a minute)
Copied!
あなたはそれが機能するのを見るはずです!ブラインド量は、まるで魔法のようにブラウザで増加します。
コードを元に戻して、テストする方法を考えましょう。これを 具現化 するために、StartGameへのパススルーはioutil.DiscardではなくplayerServerWSでしたので、呼び出しをスパイして動作を確認する必要があるかもしれません。
スパイは素晴らしく、実装の詳細をチェックするのに役立ちますが、可能であれば 実際 の動作をテストすることを常に推奨する必要があります。なぜなら、リファクタリングする場合、通常、変更しようとしている実装の詳細をチェックしているため、失敗するスパイテストがよくあるためです。 。
テストでは現在、実行中のサーバーへのWebソケット接続を開き、メッセージを送信してそれを実行します。 同様に、サーバーがWebSocket接続を介して送り返すメッセージをテストできるはずです。

最初にテストを書く

既存のテストを編集します。
現在、GameSpyは、Startを呼び出しても、データをoutに送信しません。これを変更して、定型メッセージを送信するように設定し、そのメッセージがWebSocketに送信されることを確認できるようにする必要があります。これにより、必要な実際の動作を実行しながら、正しく構成したことを確信できます。
1
type GameSpy struct {
2
StartCalled bool
3
StartCalledWith int
4
BlindAlert []byte
5
6
FinishedCalled bool
7
FinishCalledWith string
8
}
Copied!
BlindAlertフィールドを追加します。
あらかじめ用意されたメッセージをoutに送信するようにGameSpy``Startを更新します。
1
func (g *GameSpy) Start(numberOfPlayers int, out io.Writer) {
2
g.StartCalled = true
3
g.StartCalledWith = numberOfPlayers
4
out.Write(g.BlindAlert)
5
}
Copied!
つまり、ゲームをStartしようとしたときにPlayerServerを実行すると、正常に機能していれば、WebSocketを介してメッセージを送信することになります。
最後にテストを更新できます
1
t.Run("start a game with 3 players, send some blind alerts down WS and declare Ruth the winner", func(t *testing.T) {
2
wantedBlindAlert := "Blind is 100"
3
winner := "Ruth"
4
5
game := &GameSpy{BlindAlert: []byte(wantedBlindAlert)}
6
server := httptest.NewServer(mustMakePlayerServer(t, dummyPlayerStore, game))
7
ws := mustDialWS(t, "ws"+strings.TrimPrefix(server.URL, "http")+"/ws")
8
9
defer server.Close()
10
defer ws.Close()
11
12
writeWSMessage(t, ws, "3")
13
writeWSMessage(t, ws, winner)
14
15
time.Sleep(10 * time.Millisecond)
16
assertGameStartedWith(t, game, 3)
17
assertFinishCalledWith(t, game, winner)
18
19
_, gotBlindAlert, _ := ws.ReadMessage()
20
21
if string(gotBlindAlert) != wantedBlindAlert {
22
t.Errorf("got blind alert %q, want %q", string(gotBlindAlert), wantedBlindAlert)
23
}
24
})
Copied!
  • wantedBlindAlertを追加し、Startが呼び出された場合にoutに送信するようにGameSpyを設定しました。
  • メッセージがwebsocket接続で送信されることを期待して、ws.ReadMessage()への呼び出しを追加して、メッセージが送信されるのを待ってから、メッセージが期待どおりであることを確認します。

テストを実行してみます

テストが永久にハングすることを確認する必要があります。これは、ws.ReadMessage()がメッセージを取得するまでブロックするためです。

テストを実行するための最小限のコードを記述し、失敗したテスト出力を確認します

テストがハングすることは絶対にあってはならないことなので、タイムアウトさせたいコードを処理する方法を紹介します。
1
func within(t *testing.T, d time.Duration, assert func()) {
2
t.Helper()
3
4
done := make(chan struct{}, 1)
5
6
go func() {
7
assert()
8
done <- struct{}{}
9
}()
10
11
select {
12
case <-time.After(d):
13
t.Error("timed out")
14
case <-done:
15
}
16
}
Copied!
withinが行うことは、関数assertを引数として取り、それをgoルーチンで実行することです。関数が完了すると、完了(done)チャネルを介して行われたことを通知します。
その間、チャネルがメッセージを送信するのを待機できるようにするselectステートメントを使用します。ここからは、assert関数とtime.Afterの間の競争であり、期間が発生したときに信号を送信します。
最後に、アサーションのヘルパー関数を作成しました。
1
func assertWebsocketGotMsg(t *testing.T, ws *websocket.Conn, want string) {
2
_, msg, _ := ws.ReadMessage()
3
if string(msg) != want {
4
t.Errorf(`got "%s", want "%s"`, string(msg), want)
5
}
6
}
Copied!
これがテストの今の読み方です
1
t.Run("start a game with 3 players, send some blind alerts down WS and declare Ruth the winner", func(t *testing.T) {
2
wantedBlindAlert := "Blind is 100"
3
winner := "Ruth"
4
5
game := &GameSpy{BlindAlert: []byte(wantedBlindAlert)}
6
server := httptest.NewServer(mustMakePlayerServer(t, dummyPlayerStore, game))
7
ws := mustDialWS(t, "ws"+strings.TrimPrefix(server.URL, "http")+"/ws")
8
9
defer server.Close()
10
defer ws.Close()
11
12
writeWSMessage(t, ws, "3")
13
writeWSMessage(t, ws, winner)
14
15
time.Sleep(tenMS)
16
17
assertGameStartedWith(t, game, 3)
18
assertFinishCalledWith(t, game, winner)
19
within(t, tenMS, func() { assertWebsocketGotMsg(t, ws, wantedBlindAlert) })
20
})
Copied!
テストを実行すると...
1
=== RUN TestGame
2
=== RUN TestGame/start_a_game_with_3_players,_send_some_blind_alerts_down_WS_and_declare_Ruth_the_winner
3
--- FAIL: TestGame (0.02s)
4
--- FAIL: TestGame/start_a_game_with_3_players,_send_some_blind_alerts_down_WS_and_declare_Ruth_the_winner (0.02s)
5
server_test.go:143: timed out
6
server_test.go:150: got "", want "Blind is 100"
Copied!

成功させるのに十分なコードを書く