x/sync: Auto-generate MockNetworkClient (#2617)

ava-labs · Jan 17, 2024 · 68986d3 · 68986d3
1 parent 003ed7f
commit 68986d3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 25 deletions.
diff --git a/scripts/mocks.mockgen.txt b/scripts/mocks.mockgen.txt
@@ -42,3 +42,4 @@ github.com/ava-labs/avalanchego/vms/registry=VMRegisterer=vms/registry/mock_vm_r
 github.com/ava-labs/avalanchego/vms/registry=VMRegistry=vms/registry/mock_vm_registry.go
 github.com/ava-labs/avalanchego/vms=Factory,Manager=vms/mock_manager.go
 github.com/ava-labs/avalanchego/x/sync=Client=x/sync/mock_client.go
+github.com/ava-labs/avalanchego/x/sync=NetworkClient=x/sync/mock_network_client.go
diff --git a/x/sync/client_test.go b/x/sync/client_test.go
@@ -123,9 +123,6 @@ func sendRangeProofRequest(
 		},
 	).AnyTimes()
 
-	// Handle bandwidth tracking calls from client.
-	networkClient.EXPECT().TrackBandwidth(gomock.Any(), gomock.Any()).AnyTimes()
-
 	// The server should expect to "send" a response to the client.
 	sender.EXPECT().SendAppResponse(
 		gomock.Any(), // ctx

diff --git a/x/sync/mock_network_client.go b/x/sync/mock_network_client.go