mock

Documentation

Index

• Variables
• func CreateNodesFromShortConfig(nodeGroups [][]uint32, nodeInfo map[uint32]TestNodeInfo) []*Ydb_Maintenance.Node
• func MakeActionGroupsFromHostFQDNs(hostFQDNs ...string) []*Ydb_Maintenance.ActionGroup
• func MakeActionGroupsFromNodeIds(nodeIDs ...uint32) []*Ydb_Maintenance.ActionGroup
• type AdditionalMockBehaviour
• type TestNodeInfo
• type YdbMock
  • func NewYdbMockServer() *YdbMock
  • func (s *YdbMock) CompleteAction(ctx context.Context, req *CompleteActionRequest) (*ManageActionResponse, error)
  • func (s *YdbMock) CreateMaintenanceTask(ctx context.Context, req *CreateMaintenanceTaskRequest) (*MaintenanceTaskResponse, error)
  • func (s *YdbMock) DropMaintenanceTask(ctx context.Context, req *DropMaintenanceTaskRequest) (*ManageMaintenanceTaskResponse, error)
  • func (s *YdbMock) GetMaintenanceTask(ctx context.Context, req *GetMaintenanceTaskRequest) (*GetMaintenanceTaskResponse, error)
  • func (s *YdbMock) ListClusterNodes(ctx context.Context, req *ListClusterNodesRequest) (*ListClusterNodesResponse, error)
  • func (s *YdbMock) ListDatabases(ctx context.Context, req *Ydb_Cms.ListDatabasesRequest) (*Ydb_Cms.ListDatabasesResponse, error)
  • func (s *YdbMock) ListMaintenanceTasks(ctx context.Context, req *ListMaintenanceTasksRequest) (*ListMaintenanceTasksResponse, error)
  • func (s *YdbMock) Login(ctx context.Context, req *Ydb_Auth.LoginRequest) (*Ydb_Auth.LoginResponse, error)
  • func (s *YdbMock) RefreshMaintenanceTask(ctx context.Context, req *RefreshMaintenanceTaskRequest) (*MaintenanceTaskResponse, error)
  • func (s *YdbMock) SetMockBehaviour(additionalMockBehaviour AdditionalMockBehaviour)
  • func (s *YdbMock) SetNodeConfiguration(nodeGroups [][]uint32, nodeInfo map[uint32]TestNodeInfo)
  • func (s *YdbMock) SetupSimpleTLS(caFile, keyFile string)
  • func (s *YdbMock) StartOn(port int)
  • func (s *YdbMock) Teardown()
  • func (s *YdbMock) WhoAmI(ctx context.Context, req *Ydb_Discovery.WhoAmIRequest) (*Ydb_Discovery.WhoAmIResponse, error)

Variables

var (
	TestUser             = "test-user"
	TestPassword         = "test-password"
	TokenForTestUsername = "test-token-here-you-go"
)

Functions

func CreateNodesFromShortConfig

func CreateNodesFromShortConfig(nodeGroups [][]uint32, nodeInfo map[uint32]TestNodeInfo) []*Ydb_Maintenance.Node

func MakeActionGroupsFromHostFQDNs

func MakeActionGroupsFromHostFQDNs(hostFQDNs ...string) []*Ydb_Maintenance.ActionGroup

func MakeActionGroupsFromNodeIds

func MakeActionGroupsFromNodeIds(nodeIDs ...uint32) []*Ydb_Maintenance.ActionGroup

Types

type AdditionalMockBehaviour

type AdditionalMockBehaviour struct {
	RestartNodesOnNewVersion string
}

type TestNodeInfo

type TestNodeInfo struct {
	StartTime  time.Time
	IsDynnode  bool
	TenantName string
	Version    string
	Datacenter string
	State      Ydb_Maintenance.ItemState
}

type YdbMock

type YdbMock struct {
	Ydb_Maintenance_V1.UnimplementedMaintenanceServiceServer
	Ydb_Cms_V1.UnimplementedCmsServiceServer
	Ydb_Auth_V1.UnimplementedAuthServiceServer
	Ydb_Discovery_V1.UnimplementedDiscoveryServiceServer

	// This is just a log of all requests that rolling-restart has sent to
	// the CMS. It is populated during the test and then used only once to
	// check against the expected messages.
	RequestLog []proto.Message
	// contains filtered or unexported fields
}

func NewYdbMockServer

func NewYdbMockServer() *YdbMock

func (*YdbMock) CompleteAction

func (s *YdbMock) CompleteAction(ctx context.Context, req *CompleteActionRequest) (*ManageActionResponse, error)

func (*YdbMock) CreateMaintenanceTask

func (s *YdbMock) CreateMaintenanceTask(ctx context.Context, req *CreateMaintenanceTaskRequest) (*MaintenanceTaskResponse, error)

func (*YdbMock) DropMaintenanceTask

func (s *YdbMock) DropMaintenanceTask(ctx context.Context, req *DropMaintenanceTaskRequest) (*ManageMaintenanceTaskResponse, error)

func (*YdbMock) GetMaintenanceTask

func (s *YdbMock) GetMaintenanceTask(
	ctx context.Context,
	req *GetMaintenanceTaskRequest,
) (*GetMaintenanceTaskResponse, error)

func (*YdbMock) ListClusterNodes

func (s *YdbMock) ListClusterNodes(ctx context.Context, req *ListClusterNodesRequest) (*ListClusterNodesResponse, error)

func (*YdbMock) ListDatabases

func (s *YdbMock) ListDatabases(ctx context.Context, req *Ydb_Cms.ListDatabasesRequest) (*Ydb_Cms.ListDatabasesResponse, error)

func (*YdbMock) ListMaintenanceTasks

func (s *YdbMock) ListMaintenanceTasks(ctx context.Context, req *ListMaintenanceTasksRequest) (*ListMaintenanceTasksResponse, error)

func (*YdbMock) Login

func (s *YdbMock) Login(ctx context.Context, req *Ydb_Auth.LoginRequest) (*Ydb_Auth.LoginResponse, error)

func (*YdbMock) RefreshMaintenanceTask

func (s *YdbMock) RefreshMaintenanceTask(ctx context.Context, req *RefreshMaintenanceTaskRequest) (*MaintenanceTaskResponse, error)

func (*YdbMock) SetMockBehaviour

func (s *YdbMock) SetMockBehaviour(additionalMockBehaviour AdditionalMockBehaviour)

func (*YdbMock) SetNodeConfiguration

func (s *YdbMock) SetNodeConfiguration(nodeGroups [][]uint32, nodeInfo map[uint32]TestNodeInfo)

func (*YdbMock) SetupSimpleTLS

func (s *YdbMock) SetupSimpleTLS(caFile, keyFile string)

func (*YdbMock) StartOn

func (s *YdbMock) StartOn(port int)

func (*YdbMock) Teardown

func (s *YdbMock) Teardown()

func (*YdbMock) WhoAmI

func (s *YdbMock) WhoAmI(ctx context.Context, req *Ydb_Discovery.WhoAmIRequest) (*Ydb_Discovery.WhoAmIResponse, error)