[beam] branch lostluck-protosuffix updated: Update datamgr_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 3c77761  Update datamgr_test.go
3c77761 is described below

commit 3c77761fc6de60c85e5ab85a1a1a4a66791fa595
Author: Robert Burke <lostl...@users.noreply.github.com>
AuthorDate: Tue Apr 7 22:33:34 2020 -0700

    Update datamgr_test.go
---
 sdks/go/pkg/beam/core/runtime/harness/datamgr_test.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/sdks/go/pkg/beam/core/runtime/harness/datamgr_test.go 
b/sdks/go/pkg/beam/core/runtime/harness/datamgr_test.go
index 05354b3..681320c 100644
--- a/sdks/go/pkg/beam/core/runtime/harness/datamgr_test.go
+++ b/sdks/go/pkg/beam/core/runtime/harness/datamgr_test.go
@@ -25,7 +25,7 @@ import (
        "testing"
        "time"
 
-       pb "github.com/apache/beam/sdks/go/pkg/beam/model/fnexecution_v1"
+       fnpb "github.com/apache/beam/sdks/go/pkg/beam/model/fnexecution_v1"
 )
 
 const extraData = 2
@@ -40,16 +40,16 @@ type fakeDataClient struct {
        skipFirstError bool
 }
 
-func (f *fakeDataClient) Recv() (*pb.Elements, error) {
+func (f *fakeDataClient) Recv() (*fnpb.Elements, error) {
        f.calls++
        data := []byte{1, 2, 3, 4}
-       elemData := pb.Elements_Data{
+       elemData := fnpb.Elements_Data{
                InstructionId: "inst_ref",
                Data:          data,
                TransformId:   "ptr",
        }
 
-       msg := pb.Elements{}
+       msg := fnpb.Elements{}
 
        // Send extraData more than the number of elements buffered in the 
channel.
        for i := 0; i < bufElements+extraData; i++ {
@@ -66,7 +66,7 @@ func (f *fakeDataClient) Recv() (*pb.Elements, error) {
                return &msg, f.err
        case 3:
                elemData.Data = []byte{}
-               msg.Data = []*pb.Elements_Data{&elemData}
+               msg.Data = []*fnpb.Elements_Data{&elemData}
                // Broadcasting done here means that this code providing 
messages
                // has not been blocked by the bug blocking the dataReader
                // from getting more messages.
@@ -77,7 +77,7 @@ func (f *fakeDataClient) Recv() (*pb.Elements, error) {
        }
 }
 
-func (f *fakeDataClient) Send(*pb.Elements) error {
+func (f *fakeDataClient) Send(*fnpb.Elements) error {
        // We skip errors on the first call to test that  errors can be returned
        // on the sentinel value send in dataWriter.Close
        // Otherwise, we return an io.EOF similar to semantics documented