-
Notifications
You must be signed in to change notification settings - Fork 0
/
server_test.go
58 lines (50 loc) · 1.31 KB
/
server_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package main
import (
"context"
"fmt"
"io"
"net"
"net/http"
"testing"
"golang.org/x/sync/errgroup"
)
func TestServer_Run(t *testing.T) {
l, err := net.Listen("tcp", "localhost:0")
if err != nil {
t.Fatalf("failed to listen port %v", err)
}
ctx, cancel := context.WithCancel(context.Background())
eg, ctx := errgroup.WithContext(ctx)
mux := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Hello, %s", r.URL.Path[1:])
})
// 別ゴルーチンでテスト対象の「run」関数を実行してHTTPサーバーを起動する
eg.Go(func() error {
s := NewServer(l, mux)
return s.Run(ctx)
})
in := "message"
url := fmt.Sprintf("http://%s/%s", l.Addr().String(), in)
// エンドポイントに対してGETリクエストを送信する
t.Logf("try request to %q", url)
rsp, err := http.Get(url)
if err != nil {
t.Errorf("faild to get: %+v", err)
}
defer rsp.Body.Close()
got, err := io.ReadAll(rsp.Body)
if err != nil {
t.Fatalf("failed to read body: %v", err)
}
// HTTPサーバーの戻り値を検証
want := fmt.Sprintf("Hello, %s", in)
if string(got) != want {
t.Errorf("want %q, but got %q", want, got)
}
// run関数に終了通知を送信する
cancel()
// run関数の戻り値を検証する
if err := eg.Wait(); err != nil {
t.Fatal(err)
}
}