feature/tests: fixes after review

discovery.go

@@ -171,9 +171,6 @@ func (r *Router) DiscoveryAllBuckets(ctx context.Context) error {
 	errGr, ctx := errgroup.WithContext(ctx)
 
 	idToReplicasetRef := r.getIDToReplicaset()
-	if idToReplicasetRef == nil {
-		return fmt.Errorf("smth went wrong; replicasets map is empty")
-	}
 
 	for _, rs := range idToReplicasetRef {
 		rs := rs

discovery_test.go

@@ -36,20 +36,6 @@ func TestSearchLock_WaitOnSearch(t *testing.T) {
 	require.True(t, time.Since(lockStart) < 12*time.Millisecond && time.Since(lockStart) > 9*time.Millisecond)
 }
 
-func TestRouter_DiscoveryAllBuckets(t *testing.T) {
-	ctx := context.TODO()
-
-	r := Router{
-		cfg: Config{
-			TotalBucketCount: uint64(256000),
-			Logger:           &EmptyLogger{},
-		},
-	}
-
-	err := r.DiscoveryAllBuckets(ctx)
-	require.Error(t, err) // replicaset map is empty
-}
-
 func TestRouter_BucketResolve_InvalidBucketID(t *testing.T) {
 	ctx := context.TODO()
 

vshard_shadow_test.go

@@ -12,21 +12,18 @@ import (
 	"github.com/KaymeKaydex/go-vshard-router/providers/static"
 )
 
-// Check that ErrorTopologyProvider implements TopologyProvider interface
-var _ vshardrouter.TopologyProvider = (*ErrorTopologyProvider)(nil)
+type errorTopologyProvider struct{}
 
-type ErrorTopologyProvider struct{}
-
-func (e *ErrorTopologyProvider) Init(_ vshardrouter.TopologyController) error {
+func (e *errorTopologyProvider) Init(_ vshardrouter.TopologyController) error {
 	return fmt.Errorf("test error")
 }
-func (e *ErrorTopologyProvider) Close() {}
+func (e *errorTopologyProvider) Close() {}
 
 func TestNewRouter_ProviderError(t *testing.T) {
 	ctx := context.TODO()
 	_, err := vshardrouter.NewRouter(ctx, vshardrouter.Config{
 		TotalBucketCount: 256000,
-		TopologyProvider: &ErrorTopologyProvider{},
+		TopologyProvider: &errorTopologyProvider{},
 	})
 
 	require.ErrorIs(t, err, vshardrouter.ErrTopologyProvider)