You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by lo...@apache.org on 2020/04/08 05:20:12 UTC
[beam] branch lostluck-protosuffix updated: Update stage_test.go
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 <lo...@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)
}