trace/apiv1/mock_test.go - gocloud - Git at Google

 // Copyright 2019 Google LLC
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     https://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.

 // Code generated by gapic-generator. DO NOT EDIT.

 package trace

 import (
 	"context"
 	"flag"
 	"fmt"
 	"io"
 	"log"
 	"net"
 	"os"
 	"strings"
 	"testing"

 	"github.com/golang/protobuf/proto"
 	"github.com/golang/protobuf/ptypes"
 	emptypb "github.com/golang/protobuf/ptypes/empty"
 	"google.golang.org/api/option"
 	cloudtracepb "google.golang.org/genproto/googleapis/devtools/cloudtrace/v1"

 	status "google.golang.org/genproto/googleapis/rpc/status"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"

 	gstatus "google.golang.org/grpc/status"
 )

 var _ = io.EOF
 var _ = ptypes.MarshalAny
 var _ status.Status

 type mockTraceServer struct {
 	// Embed for forward compatibility.
 	// Tests will keep working if more methods are added
 	// in the future.
 	cloudtracepb.TraceServiceServer

 	reqs []proto.Message

 	// If set, all calls return this error.
 	err error

 	// responses to return if err == nil
 	resps []proto.Message
 }

 func (s *mockTraceServer) ListTraces(ctx context.Context, req *cloudtracepb.ListTracesRequest) (*cloudtracepb.ListTracesResponse, error) {
 	md, _ := metadata.FromIncomingContext(ctx)
 	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
 		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
 	}
 	s.reqs = append(s.reqs, req)
 	if s.err != nil {
 		return nil, s.err
 	}
 	return s.resps[0].(*cloudtracepb.ListTracesResponse), nil
 }

 func (s *mockTraceServer) GetTrace(ctx context.Context, req *cloudtracepb.GetTraceRequest) (*cloudtracepb.Trace, error) {
 	md, _ := metadata.FromIncomingContext(ctx)
 	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
 		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
 	}
 	s.reqs = append(s.reqs, req)
 	if s.err != nil {
 		return nil, s.err
 	}
 	return s.resps[0].(*cloudtracepb.Trace), nil
 }

 func (s *mockTraceServer) PatchTraces(ctx context.Context, req *cloudtracepb.PatchTracesRequest) (*emptypb.Empty, error) {
 	md, _ := metadata.FromIncomingContext(ctx)
 	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
 		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
 	}
 	s.reqs = append(s.reqs, req)
 	if s.err != nil {
 		return nil, s.err
 	}
 	return s.resps[0].(*emptypb.Empty), nil
 }

 // clientOpt is the option tests should use to connect to the test server.
 // It is initialized by TestMain.
 var clientOpt option.ClientOption

 var (
 	mockTrace mockTraceServer
 )

 func TestMain(m *testing.M) {
 	flag.Parse()

 	serv := grpc.NewServer()
 	cloudtracepb.RegisterTraceServiceServer(serv, &mockTrace)

 	lis, err := net.Listen("tcp", "localhost:0")
 	if err != nil {
 		log.Fatal(err)
 	}
 	go serv.Serve(lis)

 	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
 	if err != nil {
 		log.Fatal(err)
 	}
 	clientOpt = option.WithGRPCConn(conn)

 	os.Exit(m.Run())
 }

 func TestTraceServicePatchTraces(t *testing.T) {
 	var expectedResponse *emptypb.Empty = &emptypb.Empty{}

 	mockTrace.err = nil
 	mockTrace.reqs = nil

 	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

 	var projectId string = "projectId-1969970175"
 	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
 	var request = &cloudtracepb.PatchTracesRequest{
 		ProjectId: projectId,
 		Traces:    traces,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	err = c.PatchTraces(context.Background(), request)

 	if err != nil {
 		t.Fatal(err)
 	}

 	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
 		t.Errorf("wrong request %q, want %q", got, want)
 	}

 }

 func TestTraceServicePatchTracesError(t *testing.T) {
 	errCode := codes.PermissionDenied
 	mockTrace.err = gstatus.Error(errCode, "test error")

 	var projectId string = "projectId-1969970175"
 	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
 	var request = &cloudtracepb.PatchTracesRequest{
 		ProjectId: projectId,
 		Traces:    traces,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	err = c.PatchTraces(context.Background(), request)

 	if st, ok := gstatus.FromError(err); !ok {
 		t.Errorf("got error %v, expected grpc error", err)
 	} else if c := st.Code(); c != errCode {
 		t.Errorf("got error code %q, want %q", c, errCode)
 	}
 }
 func TestTraceServiceGetTrace(t *testing.T) {
 	var projectId2 string = "projectId2939242356"
 	var traceId2 string = "traceId2987826376"
 	var expectedResponse = &cloudtracepb.Trace{
 		ProjectId: projectId2,
 		TraceId:   traceId2,
 	}

 	mockTrace.err = nil
 	mockTrace.reqs = nil

 	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

 	var projectId string = "projectId-1969970175"
 	var traceId string = "traceId1270300245"
 	var request = &cloudtracepb.GetTraceRequest{
 		ProjectId: projectId,
 		TraceId:   traceId,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	resp, err := c.GetTrace(context.Background(), request)

 	if err != nil {
 		t.Fatal(err)
 	}

 	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
 		t.Errorf("wrong request %q, want %q", got, want)
 	}

 	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
 		t.Errorf("wrong response %q, want %q)", got, want)
 	}
 }

 func TestTraceServiceGetTraceError(t *testing.T) {
 	errCode := codes.PermissionDenied
 	mockTrace.err = gstatus.Error(errCode, "test error")

 	var projectId string = "projectId-1969970175"
 	var traceId string = "traceId1270300245"
 	var request = &cloudtracepb.GetTraceRequest{
 		ProjectId: projectId,
 		TraceId:   traceId,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	resp, err := c.GetTrace(context.Background(), request)

 	if st, ok := gstatus.FromError(err); !ok {
 		t.Errorf("got error %v, expected grpc error", err)
 	} else if c := st.Code(); c != errCode {
 		t.Errorf("got error code %q, want %q", c, errCode)
 	}
 	_ = resp
 }
 func TestTraceServiceListTraces(t *testing.T) {
 	var nextPageToken string = ""
 	var tracesElement *cloudtracepb.Trace = &cloudtracepb.Trace{}
 	var traces = []*cloudtracepb.Trace{tracesElement}
 	var expectedResponse = &cloudtracepb.ListTracesResponse{
 		NextPageToken: nextPageToken,
 		Traces:        traces,
 	}

 	mockTrace.err = nil
 	mockTrace.reqs = nil

 	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

 	var projectId string = "projectId-1969970175"
 	var request = &cloudtracepb.ListTracesRequest{
 		ProjectId: projectId,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	resp, err := c.ListTraces(context.Background(), request).Next()

 	if err != nil {
 		t.Fatal(err)
 	}

 	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
 		t.Errorf("wrong request %q, want %q", got, want)
 	}

 	want := (interface{})(expectedResponse.Traces[0])
 	got := (interface{})(resp)
 	var ok bool

 	switch want := (want).(type) {
 	case proto.Message:
 		ok = proto.Equal(want, got.(proto.Message))
 	default:
 		ok = want == got
 	}
 	if !ok {
 		t.Errorf("wrong response %q, want %q)", got, want)
 	}
 }

 func TestTraceServiceListTracesError(t *testing.T) {
 	errCode := codes.PermissionDenied
 	mockTrace.err = gstatus.Error(errCode, "test error")

 	var projectId string = "projectId-1969970175"
 	var request = &cloudtracepb.ListTracesRequest{
 		ProjectId: projectId,
 	}

 	c, err := NewClient(context.Background(), clientOpt)
 	if err != nil {
 		t.Fatal(err)
 	}

 	resp, err := c.ListTraces(context.Background(), request).Next()

 	if st, ok := gstatus.FromError(err); !ok {
 		t.Errorf("got error %v, expected grpc error", err)
 	} else if c := st.Code(); c != errCode {
 		t.Errorf("got error code %q, want %q", c, errCode)
 	}
 	_ = resp
 }
	// Copyright 2019 Google LLC
	//
	// Licensed under the Apache License, Version 2.0 (the "License");
	// you may not use this file except in compliance with the License.
	// You may obtain a copy of the License at
	//
	// https://www.apache.org/licenses/LICENSE-2.0
	//
	// Unless required by applicable law or agreed to in writing, software
	// distributed under the License is distributed on an "AS IS" BASIS,
	// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	// See the License for the specific language governing permissions and
	// limitations under the License.

	// Code generated by gapic-generator. DO NOT EDIT.

	package trace

	import (
	"context"
	"flag"
	"fmt"
	"io"
	"log"
	"net"
	"os"
	"strings"
	"testing"

	"github.com/golang/protobuf/proto"
	"github.com/golang/protobuf/ptypes"
	emptypb "github.com/golang/protobuf/ptypes/empty"
	"google.golang.org/api/option"
	cloudtracepb "google.golang.org/genproto/googleapis/devtools/cloudtrace/v1"

	status "google.golang.org/genproto/googleapis/rpc/status"
	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/metadata"

	gstatus "google.golang.org/grpc/status"
	)

	var _ = io.EOF
	var _ = ptypes.MarshalAny
	var _ status.Status

	type mockTraceServer struct {
	// Embed for forward compatibility.
	// Tests will keep working if more methods are added
	// in the future.
	cloudtracepb.TraceServiceServer

	reqs []proto.Message

	// If set, all calls return this error.
	err error

	// responses to return if err == nil
	resps []proto.Message
	}

	func (s mockTraceServer) ListTraces(ctx context.Context, req cloudtracepb.ListTracesRequest) (*cloudtracepb.ListTracesResponse, error) {
	md, _ := metadata.FromIncomingContext(ctx)
	if xg := md["x-goog-api-client"]; len(xg) == 0 \|\| !strings.Contains(xg[0], "gl-go/") {
	return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
	}
	s.reqs = append(s.reqs, req)
	if s.err != nil {
	return nil, s.err
	}
	return s.resps[0].(*cloudtracepb.ListTracesResponse), nil
	}

	func (s mockTraceServer) GetTrace(ctx context.Context, req cloudtracepb.GetTraceRequest) (*cloudtracepb.Trace, error) {
	md, _ := metadata.FromIncomingContext(ctx)
	if xg := md["x-goog-api-client"]; len(xg) == 0 \|\| !strings.Contains(xg[0], "gl-go/") {
	return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
	}
	s.reqs = append(s.reqs, req)
	if s.err != nil {
	return nil, s.err
	}
	return s.resps[0].(*cloudtracepb.Trace), nil
	}

	func (s mockTraceServer) PatchTraces(ctx context.Context, req cloudtracepb.PatchTracesRequest) (*emptypb.Empty, error) {
	md, _ := metadata.FromIncomingContext(ctx)
	if xg := md["x-goog-api-client"]; len(xg) == 0 \|\| !strings.Contains(xg[0], "gl-go/") {
	return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
	}
	s.reqs = append(s.reqs, req)
	if s.err != nil {
	return nil, s.err
	}
	return s.resps[0].(*emptypb.Empty), nil
	}

	// clientOpt is the option tests should use to connect to the test server.
	// It is initialized by TestMain.
	var clientOpt option.ClientOption

	var (
	mockTrace mockTraceServer
	)

	func TestMain(m *testing.M) {
	flag.Parse()

	serv := grpc.NewServer()
	cloudtracepb.RegisterTraceServiceServer(serv, &mockTrace)

	lis, err := net.Listen("tcp", "localhost:0")
	if err != nil {
	log.Fatal(err)
	}
	go serv.Serve(lis)

	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
	if err != nil {
	log.Fatal(err)
	}
	clientOpt = option.WithGRPCConn(conn)

	os.Exit(m.Run())
	}

	func TestTraceServicePatchTraces(t *testing.T) {
	var expectedResponse *emptypb.Empty = &emptypb.Empty{}

	mockTrace.err = nil
	mockTrace.reqs = nil

	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

	var projectId string = "projectId-1969970175"
	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
	var request = &cloudtracepb.PatchTracesRequest{
	ProjectId: projectId,
	Traces: traces,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	err = c.PatchTraces(context.Background(), request)

	if err != nil {
	t.Fatal(err)
	}

	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
	t.Errorf("wrong request %q, want %q", got, want)
	}

	}

	func TestTraceServicePatchTracesError(t *testing.T) {
	errCode := codes.PermissionDenied
	mockTrace.err = gstatus.Error(errCode, "test error")

	var projectId string = "projectId-1969970175"
	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
	var request = &cloudtracepb.PatchTracesRequest{
	ProjectId: projectId,
	Traces: traces,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	err = c.PatchTraces(context.Background(), request)

	if st, ok := gstatus.FromError(err); !ok {
	t.Errorf("got error %v, expected grpc error", err)
	} else if c := st.Code(); c != errCode {
	t.Errorf("got error code %q, want %q", c, errCode)
	}
	}
	func TestTraceServiceGetTrace(t *testing.T) {
	var projectId2 string = "projectId2939242356"
	var traceId2 string = "traceId2987826376"
	var expectedResponse = &cloudtracepb.Trace{
	ProjectId: projectId2,
	TraceId: traceId2,
	}

	mockTrace.err = nil
	mockTrace.reqs = nil

	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

	var projectId string = "projectId-1969970175"
	var traceId string = "traceId1270300245"
	var request = &cloudtracepb.GetTraceRequest{
	ProjectId: projectId,
	TraceId: traceId,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	resp, err := c.GetTrace(context.Background(), request)

	if err != nil {
	t.Fatal(err)
	}

	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
	t.Errorf("wrong request %q, want %q", got, want)
	}

	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
	t.Errorf("wrong response %q, want %q)", got, want)
	}
	}

	func TestTraceServiceGetTraceError(t *testing.T) {
	errCode := codes.PermissionDenied
	mockTrace.err = gstatus.Error(errCode, "test error")

	var projectId string = "projectId-1969970175"
	var traceId string = "traceId1270300245"
	var request = &cloudtracepb.GetTraceRequest{
	ProjectId: projectId,
	TraceId: traceId,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	resp, err := c.GetTrace(context.Background(), request)

	if st, ok := gstatus.FromError(err); !ok {
	t.Errorf("got error %v, expected grpc error", err)
	} else if c := st.Code(); c != errCode {
	t.Errorf("got error code %q, want %q", c, errCode)
	}
	_ = resp
	}
	func TestTraceServiceListTraces(t *testing.T) {
	var nextPageToken string = ""
	var tracesElement *cloudtracepb.Trace = &cloudtracepb.Trace{}
	var traces = []*cloudtracepb.Trace{tracesElement}
	var expectedResponse = &cloudtracepb.ListTracesResponse{
	NextPageToken: nextPageToken,
	Traces: traces,
	}

	mockTrace.err = nil
	mockTrace.reqs = nil

	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)

	var projectId string = "projectId-1969970175"
	var request = &cloudtracepb.ListTracesRequest{
	ProjectId: projectId,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	resp, err := c.ListTraces(context.Background(), request).Next()

	if err != nil {
	t.Fatal(err)
	}

	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
	t.Errorf("wrong request %q, want %q", got, want)
	}

	want := (interface{})(expectedResponse.Traces[0])
	got := (interface{})(resp)
	var ok bool

	switch want := (want).(type) {
	case proto.Message:
	ok = proto.Equal(want, got.(proto.Message))
	default:
	ok = want == got
	}
	if !ok {
	t.Errorf("wrong response %q, want %q)", got, want)
	}
	}

	func TestTraceServiceListTracesError(t *testing.T) {
	errCode := codes.PermissionDenied
	mockTrace.err = gstatus.Error(errCode, "test error")

	var projectId string = "projectId-1969970175"
	var request = &cloudtracepb.ListTracesRequest{
	ProjectId: projectId,
	}

	c, err := NewClient(context.Background(), clientOpt)
	if err != nil {
	t.Fatal(err)
	}

	resp, err := c.ListTraces(context.Background(), request).Next()

	if st, ok := gstatus.FromError(err); !ok {
	t.Errorf("got error %v, expected grpc error", err)
	} else if c := st.Code(); c != errCode {
	t.Errorf("got error code %q, want %q", c, errCode)
	}
	_ = resp
	}