@@ -61,8 +61,8 @@ func TestGetAllFeedsFOk(t *testing.T) {
6161 r := require .New (t )
6262 a := assert .New (t )
6363 rpc , client := newBackendRPCTest (t )
64- streamClient1 := NewMockNeon_StreamEntriesClient (gomock .NewController (t ))
65- streamClient2 := NewMockNeon_StreamEntriesClient (gomock .NewController (t ))
64+ streamClient1 := NewMockServerStreamingClient [api. StreamEntriesResponse ] (gomock .NewController (t ))
65+ streamClient2 := NewMockServerStreamingClient [api. StreamEntriesResponse ] (gomock .NewController (t ))
6666
6767 client .EXPECT ().
6868 ListFeeds (gomock .Any (), gomock .Any (), gomock .Any ()).
@@ -175,7 +175,7 @@ func TestGetAllFeedsFErrStream(t *testing.T) {
175175 r := require .New (t )
176176 a := assert .New (t )
177177 rpc , client := newBackendRPCTest (t )
178- streamClient1 := NewMockNeon_StreamEntriesClient (gomock .NewController (t ))
178+ streamClient1 := NewMockServerStreamingClient [api. StreamEntriesResponse ] (gomock .NewController (t ))
179179
180180 client .EXPECT ().
181181 ListFeeds (gomock .Any (), gomock .Any (), gomock .Any ()).
@@ -229,8 +229,8 @@ func TestGetAllFeedsFErrStreamRecv(t *testing.T) {
229229 r := require .New (t )
230230 a := assert .New (t )
231231 rpc , client := newBackendRPCTest (t )
232- streamClient1 := NewMockNeon_StreamEntriesClient (gomock .NewController (t ))
233- streamClient2 := NewMockNeon_StreamEntriesClient (gomock .NewController (t ))
232+ streamClient1 := NewMockServerStreamingClient [api. StreamEntriesResponse ] (gomock .NewController (t ))
233+ streamClient2 := NewMockServerStreamingClient [api. StreamEntriesResponse ] (gomock .NewController (t ))
234234
235235 client .EXPECT ().
236236 ListFeeds (gomock .Any (), gomock .Any (), gomock .Any ()).
@@ -290,7 +290,7 @@ func TestPullFeedsFExtended(t *testing.T) {
290290 r := require .New (t )
291291 a := assert .New (t )
292292 rpc , client := newBackendRPCTest (t )
293- streamClient := NewMockNeon_PullFeedsClient (gomock .NewController (t ))
293+ streamClient := NewMockServerStreamingClient [api. PullFeedsResponse ] (gomock .NewController (t ))
294294
295295 client .EXPECT ().
296296 PullFeeds (gomock .Any (), gomock .Any ()).
@@ -369,7 +369,7 @@ func TestPullFeedsFErrStream(t *testing.T) {
369369 r := require .New (t )
370370 a := assert .New (t )
371371 rpc , client := newBackendRPCTest (t )
372- streamClient := NewMockNeon_PullFeedsClient (gomock .NewController (t ))
372+ streamClient := NewMockServerStreamingClient [api. PullFeedsResponse ] (gomock .NewController (t ))
373373
374374 client .EXPECT ().
375375 PullFeeds (gomock .Any (), gomock .Any ()).
0 commit comments