pubsub apiv1 publisher test

pubsub_apiv1_publisher_test.go

package pubsub

import (
	"context"
	"testing"

	"cloud.google.com/go/pubsub/pstest"
	"google.golang.org/api/option"
	pubsubpb "google.golang.org/genproto/googleapis/pubsub/v1"
	"google.golang.org/grpc"
)

func TestPublisher_Publish(t *testing.T) {
	t.Helper()

	// テストケース作成
	cases := map[string]struct {
		topic      string
		msg        string
		createFlag bool
		expected   bool
	}{
		"topicが存在しない場合失敗": {
			topic:      "projects/apiv1-sample/topics/xxxxx",
			msg:        "message-1",
			createFlag: false,
			expected:   true,
		},
		"成功": {
			topic:      "projects/apiv1-sample/topics/test-topic",
			msg:        "message-2",
			createFlag: true,
			expected:   false,
		},
	}

	for k, c := range cases {
		srv := pstest.NewServer()
		conn, _ := grpc.Dial(srv.Addr, grpc.WithInsecure())
		defer srv.Close()
		defer conn.Close()

		t.Run(k, func(t *testing.T) {
			ctx := context.Background()
			p, _ := NewPublisher(ctx, option.WithGRPCConn(conn))
			if c.createFlag {
				req := &pubsubpb.Topic{
					Name: c.topic,
				}
				p.Client.CreateTopic(ctx, req)
			}

			err := p.Publish(ctx, c.topic, c.msg)
			if (err != nil) != c.expected {
				t.Errorf("TestPublisher_Publish error. got: %v, expected: %v", err, c.expected)
			}
		})
	}
}