storj/satellite/repair/queue/mock.go
Márton Elek c44e3d78d8 satellite/satellitedb: repairqueue.Select uses placement constraints
Change-Id: I59739926f8f6c5eaca3199369d4c5d88a9c08be8
2023-09-25 10:14:25 +00:00

76 lines
2.1 KiB
Go

// Copyright (C) 2023 Storj Labs, Inc.
// See LICENSE for copying information.
package queue
import (
"context"
"time"
"storj.io/common/storj"
"storj.io/common/uuid"
"storj.io/storj/satellite/metabase"
)
// MockRepairQueue helps testing RepairQueue.
type MockRepairQueue struct {
Segments []*InjuredSegment
}
// Insert implements RepairQueue.
func (m *MockRepairQueue) Insert(ctx context.Context, s *InjuredSegment) (alreadyInserted bool, err error) {
for _, alreadyAdded := range m.Segments {
if alreadyAdded.StreamID == s.StreamID {
return true, nil
}
}
m.Segments = append(m.Segments, s)
return false, nil
}
// InsertBatch implements RepairQueue.
func (m *MockRepairQueue) InsertBatch(ctx context.Context, segments []*InjuredSegment) (newlyInsertedSegments []*InjuredSegment, err error) {
for _, segment := range segments {
inserted, err := m.Insert(ctx, segment)
if err != nil {
return nil, err
}
if inserted {
newlyInsertedSegments = append(newlyInsertedSegments, segment)
}
}
return newlyInsertedSegments, nil
}
// Select implements RepairQueue.
func (m *MockRepairQueue) Select(context.Context, []storj.PlacementConstraint, []storj.PlacementConstraint) (*InjuredSegment, error) {
panic("implement me")
}
// Delete implements RepairQueue.
func (m *MockRepairQueue) Delete(ctx context.Context, s *InjuredSegment) error {
panic("implement me")
}
// Clean implements RepairQueue.
func (m *MockRepairQueue) Clean(ctx context.Context, before time.Time) (deleted int64, err error) {
panic("implement me")
}
// SelectN implements RepairQueue.
func (m *MockRepairQueue) SelectN(ctx context.Context, limit int) ([]InjuredSegment, error) {
panic("implement me")
}
// Count implements RepairQueue.
func (m *MockRepairQueue) Count(ctx context.Context) (count int, err error) {
panic("implement me")
}
// TestingSetAttemptedTime implements RepairQueue.
func (m *MockRepairQueue) TestingSetAttemptedTime(ctx context.Context, streamID uuid.UUID, position metabase.SegmentPosition, t time.Time) (rowsAffected int64, err error) {
panic("implement me")
}
var _ RepairQueue = &MockRepairQueue{}