[beam] branch lostluck-protosuffix updated: Update stage_test.go

[lostluck]

This is an automated email from the ASF dual-hosted git repository.

lostluck pushed a commit to branch lostluck-protosuffix
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/lostluck-protosuffix by this 
push:
     new 4a378e9  Update stage_test.go
4a378e9 is described below

commit 4a378e9157d754d7a9021619ceb8776c33385d72
Author: Robert Burke <lostl...@users.noreply.github.com>
AuthorDate: Tue Apr 7 22:20:01 2020 -0700

    Update stage_test.go
---
 sdks/go/pkg/beam/artifact/stage_test.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/sdks/go/pkg/beam/artifact/stage_test.go 
b/sdks/go/pkg/beam/artifact/stage_test.go
index 266ab2b..57121e5 100644
--- a/sdks/go/pkg/beam/artifact/stage_test.go
+++ b/sdks/go/pkg/beam/artifact/stage_test.go
@@ -21,7 +21,7 @@ import (
        "os"
        "testing"
 
-       pb "github.com/apache/beam/sdks/go/pkg/beam/model/jobmanagement_v1"
+       jobpb "github.com/apache/beam/sdks/go/pkg/beam/model/jobmanagement_v1"
        "github.com/apache/beam/sdks/go/pkg/beam/util/grpcx"
        "google.golang.org/grpc"
 )
@@ -30,7 +30,7 @@ import (
 func TestStage(t *testing.T) {
        cc := startServer(t)
        defer cc.Close()
-       client := pb.NewLegacyArtifactStagingServiceClient(cc)
+       client := jobpb.NewLegacyArtifactStagingServiceClient(cc)
 
        ctx := grpcx.WriteWorkerID(context.Background(), "idA")
        keys := []string{"foo", "bar", "baz/baz/baz"}
@@ -40,7 +40,7 @@ func TestStage(t *testing.T) {
        md5s := makeTempFiles(t, src, keys, 300)
 
        st := "whatever"
-       var artifacts []*pb.ArtifactMetadata
+       var artifacts []*jobpb.ArtifactMetadata
        for _, key := range keys {
                a, err := Stage(ctx, client, key, makeFilename(src, key), st)
                if err != nil {
@@ -59,7 +59,7 @@ func TestStage(t *testing.T) {
 func TestStageDir(t *testing.T) {
        cc := startServer(t)
        defer cc.Close()
-       client := pb.NewLegacyArtifactStagingServiceClient(cc)
+       client := jobpb.NewLegacyArtifactStagingServiceClient(cc)
 
        ctx := grpcx.WriteWorkerID(context.Background(), "idB")
        keys := []string{"1", "2", "3", "4", "a/5", "a/6", "a/7", "a/8", 
"a/a/9", "a/a/10", "a/b/11", "a/b/12"}
@@ -81,10 +81,10 @@ func TestStageDir(t *testing.T) {
 }
 
 func validate(ctx context.Context, cc *grpc.ClientConn, t *testing.T, keys, 
sha256s []string, rt string) {
-       rcl := pb.NewLegacyArtifactRetrievalServiceClient(cc)
+       rcl := jobpb.NewLegacyArtifactRetrievalServiceClient(cc)
 
        for i, key := range keys {
-               stream, err := rcl.GetArtifact(ctx, 
&pb.LegacyGetArtifactRequest{Name: key, RetrievalToken: rt})
+               stream, err := rcl.GetArtifact(ctx, 
&jobpb.LegacyGetArtifactRequest{Name: key, RetrievalToken: rt})
                if err != nil {
                        t.Fatalf("failed to get artifact for %v: %v", key, err)
                }