fix unittest

pg-sharding · Sep 14, 2024 · 0908ff5 · 0908ff5
1 parent 00a5627
commit 0908ff5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/router/frontend/frontend_test.go b/router/frontend/frontend_test.go
@@ -422,7 +422,7 @@ func TestFrontendSimpleCopyIn(t *testing.T) {
 
 	srv.EXPECT().Send(query).Times(1).Return(nil)
 
-	sh1.EXPECT().Send(&pgproto3.CopyData{Data: []byte("1\n1\n1\n1\n")}).Times(1).Return(nil)
+	sh1.EXPECT().Send(&pgproto3.CopyData{Data: []byte("1\n")}).Times(4).Return(nil)
 	srv.EXPECT().Send(&pgproto3.CopyDone{}).Times(1).Return(nil)
 
 	srv.EXPECT().Receive().Times(1).Return(&pgproto3.CopyInResponse{}, nil)