From 00005e72d46e4c193a373fbd2396b6c120e28990 Mon Sep 17 00:00:00 2001 From: trevleon Date: Sat, 1 Jun 2024 23:03:31 -0700 Subject: [PATCH] private --- go/cmd/gcsobjtable/main_test.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/go/cmd/gcsobjtable/main_test.go b/go/cmd/gcsobjtable/main_test.go index c32c388..890eef4 100644 --- a/go/cmd/gcsobjtable/main_test.go +++ b/go/cmd/gcsobjtable/main_test.go @@ -357,22 +357,22 @@ func TestRequestLocation(t *testing.T) { } } -type MockSchedulerClient struct { +type mockSchedulerClient struct { pb.GlobalSchedulerClient requestsReceived map[uint64]*pb.GlobalScheduleRequest } -func (m *MockSchedulerClient) Schedule(ctx context.Context , req *pb.GlobalScheduleRequest, opts ...grpc.CallOption ) (*pb.StatusResponse, error) { +func (m *mockSchedulerClient) Schedule(ctx context.Context , req *pb.GlobalScheduleRequest, opts ...grpc.CallOption ) (*pb.StatusResponse, error) { requestsReceived[req.Uid] = req return nil, nil } -func (m *MockSchedulerClient) Heartbeat(ctx context.Context, req *pb.HeartbeatRequest, opts ...grpc.CallOption ) (*pb.StatusResponse, error) { +func (m *mockSchedulerClient) Heartbeat(ctx context.Context, req *pb.HeartbeatRequest, opts ...grpc.CallOption ) (*pb.StatusResponse, error) { return nil, nil } -func (m *MockSchedulerClient) LiveNodesHeartbeat(ctx context.Context) (error) { +func (m *mockSchedulerClient) LiveNodesHeartbeat(ctx context.Context) (error) { return nil } -func (m *MockSchedulerClient) SendLiveNodes(ctx context.Context) { +func (m *mockSchedulerClient) SendLiveNodes(ctx context.Context) { return nil } func TestNodeDiesWhileGenerating(t *testing.T) { @@ -384,7 +384,7 @@ func TestNodeDiesWhileGenerating(t *testing.T) { */ ctx = context.Background() req := &pb.GlobalScheduleRequest{Uid: 200, Name: "func_name", Args: [], Kwargs: []} - m := MockSchedulerClient { + m := mockSchedulerClient { requestsReceived: make(map[uint64]bool), }