github.com/weaviate/weaviate@v1.24.6/usecases/traverser/hybrid/fakes_for_test.go (about)

     1  //                           _       _
     2  // __      _____  __ ___   ___  __ _| |_ ___
     3  // \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
     4  //  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
     5  //   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
     6  //
     7  //  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
     8  //
     9  //  CONTACT: hello@weaviate.io
    10  //
    11  
    12  package hybrid
    13  
    14  import (
    15  	"context"
    16  
    17  	"github.com/weaviate/weaviate/entities/backup"
    18  	"github.com/weaviate/weaviate/entities/models"
    19  	"github.com/weaviate/weaviate/entities/schema"
    20  	"github.com/weaviate/weaviate/usecases/sharding"
    21  )
    22  
    23  func newFakeTargetVectorParamHelper() *fakeTargetVectorParamHelper {
    24  	return &fakeTargetVectorParamHelper{}
    25  }
    26  
    27  func newFakeSchemaManager() *fakeSchemaManager {
    28  	return &fakeSchemaManager{
    29  		schema: schema.Schema{Objects: &models.Schema{}},
    30  	}
    31  }
    32  
    33  type fakeTargetVectorParamHelper struct{}
    34  
    35  func (*fakeTargetVectorParamHelper) GetTargetVectorOrDefault(sch schema.Schema, className, targetVector string) (string, error) {
    36  	return targetVector, nil
    37  }
    38  
    39  type fakeSchemaManager struct {
    40  	schema schema.Schema
    41  }
    42  
    43  func (f *fakeSchemaManager) GetSchemaSkipAuth() schema.Schema {
    44  	return f.schema
    45  }
    46  
    47  func (f *fakeSchemaManager) CopyShardingState(class string) *sharding.State {
    48  	return nil
    49  }
    50  
    51  func (f *fakeSchemaManager) ShardOwner(class, shard string) (string, error) {
    52  	return "", nil
    53  }
    54  
    55  func (f *fakeSchemaManager) ShardReplicas(class, shard string) ([]string, error) {
    56  	return []string{}, nil
    57  }
    58  
    59  func (f *fakeSchemaManager) RestoreClass(ctx context.Context, d *backup.ClassDescriptor, nodeMapping map[string]string) error {
    60  	return nil
    61  }
    62  
    63  func (f *fakeSchemaManager) Nodes() []string {
    64  	return []string{"NOT SET"}
    65  }
    66  
    67  func (f *fakeSchemaManager) NodeName() string {
    68  	return ""
    69  }
    70  
    71  func (f *fakeSchemaManager) ClusterHealthScore() int {
    72  	return 0
    73  }
    74  
    75  func (f *fakeSchemaManager) ResolveParentNodes(_ string, shard string,
    76  ) (map[string]string, error) {
    77  	return nil, nil
    78  }
    79  
    80  func (f *fakeSchemaManager) ShardFromUUID(class string, uuid []byte) string {
    81  	return ""
    82  }
    83  
    84  func (f *fakeSchemaManager) TenantShard(class, tenant string) (string, string) {
    85  	return "", ""
    86  }