Go + gRPCのシナリオテストにrunnを使用したいメモ

go install github.com/k1LoW/runn/cmd/runn@latest

runn --version
> runn version 0.67.0

runn new -- grpcurl --d '{"name": "World"}' localhost:8080 hello.GreetingService.Hello

desc: Generated by `runn new`
runners:
  greq: grpc://localhost:8080
steps:
- greq:
    hello.GreetingService.Hello:
      message:
        name: World

runn new --and-run --grpc-no-tls -- grpcurl -d '{"name": "World"}' localhost:8080 
hello.GreetingService/Hello

desc: Generated by `runn new`
runners:
  greq: grpc://localhost:8080
steps:
- greq:
    hello.GreetingService/Hello:
      message:
        name: World
  test: |
    current.res.headers['content-type'][0] == "application/grpc"
    && compare(current.res.message, {"message":"Hello, World!"})
    && current.res.status == 0

runn new --and-run --grpc-no-tls --out ./runbooks/runbook.yml -- grpcurl -d '{"name": "World"}' localhost:8080 hello.GreetingService/Hello

func TestHello(t *testing.T) {
	addr := "127.0.0.1:8080"
	l, err := net.Listen("tcp", addr)
	if err != nil {
		t.Fatal(err)
	}

	s := grpc.NewServer()

	hellopb.RegisterGreetingServiceServer(s, service.NewHelloService())

	reflection.Register(s)

	go func() {
		s.Serve(l)
	}()
	t.Cleanup(func() {
		s.GracefulStop()
	})

	ctx := context.Background()
	opts := []runn.Option{
		runn.T(t),
		runn.Runner("greq", fmt.Sprintf("grpc://%s", addr)),
	}

	o, err := runn.Load("runbooks/*.yml", opts...)
	if err != nil {
		t.Fatal(err)
	}

	if err := o.RunN(ctx); err != nil {
		t.Fatal(err)
	}
}

desc: ログインしてHello Worldする
runners:
  #  gRPCランナーを指定
  greq: grpc://localhost:8080
vars:
#  username: ${USERNAME}
#  password: ${PASSWORD}
  username: yamanaka
  password: pass
steps:
  -
    desc: Step.0 ログインする
    greq:
      hello.GreetingService/Login:
        message:
          name: "{{ vars.username }}"
          pass: "{{ vars.password }}"
    test: |
      current.res.headers['content-type'][0] == "application/grpc"
      && compare(current.res.message, {"id":"U_0001", "name":"yamanaka"})
      && current.res.status == 0
  -
    desc: Step.1 ログインしたユーザー名でハローする
    greq:
      hello.GreetingService/Hello:
        message:
          name: "{{ previous.res.message.name }}"
    test: |
      current.res.headers['content-type'][0] == "application/grpc"
      && compare(current.res.message, {"message":"Hello, yamanaka!"})
      && current.res.status == 0