Skip to content

Commit 8566dfd

Browse files
committed
rebasing
Signed-off-by: Valentin Daviot <valentin.daviot@scality.com>
1 parent c7225ee commit 8566dfd

3 files changed

Lines changed: 39 additions & 16 deletions

File tree

internal/controller/discovery_ticker_test.go

Lines changed: 24 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,14 @@ var _ = Describe("DiscoveryTicker", func() {
4646
NodeName: "test-node",
4747
Interval: 100 * time.Millisecond,
4848
EventChan: eventChan,
49-
UseCase: usecase.NewDiscoverPhysicalDrives(logr.Discard(), nil, nil, &noopCacheWriter{}, "test-node"),
49+
UseCase: usecase.NewDiscoverPhysicalDrives(
50+
logr.Discard(),
51+
[]service.PhysicalDriveDiscoverer{},
52+
[]service.LogicalVolumeDiscoverer{},
53+
nil,
54+
&noopCacheWriter{},
55+
"test-node",
56+
),
5057
}
5158

5259
done := make(chan error, 1)
@@ -70,7 +77,14 @@ var _ = Describe("DiscoveryTicker", func() {
7077
NodeName: "test-node",
7178
Interval: 50 * time.Millisecond,
7279
EventChan: eventChan,
73-
UseCase: usecase.NewDiscoverPhysicalDrives(logr.Discard(), nil, nil, &noopCacheWriter{}, "test-node"),
80+
UseCase: usecase.NewDiscoverPhysicalDrives(
81+
logr.Discard(),
82+
[]service.PhysicalDriveDiscoverer{},
83+
[]service.LogicalVolumeDiscoverer{},
84+
nil,
85+
&noopCacheWriter{},
86+
"test-node",
87+
),
7488
}
7589

7690
tickCount := 0
@@ -110,7 +124,14 @@ var _ = Describe("DiscoveryTicker", func() {
110124
NodeName: "test-node",
111125
Interval: time.Hour, // Long interval; should not matter.
112126
EventChan: eventChan,
113-
UseCase: usecase.NewDiscoverPhysicalDrives(logr.Discard(), nil, nil, &noopCacheWriter{}, "test-node"),
127+
UseCase: usecase.NewDiscoverPhysicalDrives(
128+
logr.Discard(),
129+
[]service.PhysicalDriveDiscoverer{},
130+
[]service.LogicalVolumeDiscoverer{},
131+
nil,
132+
&noopCacheWriter{},
133+
"test-node",
134+
),
114135
}
115136

116137
done := make(chan error, 1)

pkg/usecase/discover_physical_drives.go

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,12 @@ import (
3434
// It also populates the drive cache so that the reconciler can read the
3535
// latest discovered state.
3636
type DiscoverPhysicalDrives struct {
37-
logger logr.Logger
38-
discoverers []service.PhysicalDriveDiscoverer
39-
store service.DiscoveredPhysicalDiskStore
40-
cacheWriter service.DiscoveredDriveCacheWriter
41-
nodeName string
37+
logger logr.Logger
38+
pdDiscoverers []service.PhysicalDriveDiscoverer
39+
lvDiscoverers []service.LogicalVolumeDiscoverer
40+
store service.DiscoveredPhysicalDiskStore
41+
cacheWriter service.DiscoveredDriveCacheWriter
42+
nodeName string
4243
}
4344

4445
// NewDiscoverPhysicalDrives creates a new DiscoverPhysicalDrives use case.
@@ -51,11 +52,12 @@ func NewDiscoverPhysicalDrives(
5152
nodeName string,
5253
) *DiscoverPhysicalDrives {
5354
return &DiscoverPhysicalDrives{
54-
logger: logger.WithName("discover-physical-drives"),
55-
discoverers: discoverers,
56-
store: store,
57-
cacheWriter: cacheWriter,
58-
nodeName: nodeName,
55+
logger: logger.WithName("discover-physical-drives"),
56+
pdDiscoverers: pdDiscoverers,
57+
lvDiscoverers: lvDiscoverers,
58+
store: store,
59+
cacheWriter: cacheWriter,
60+
nodeName: nodeName,
5961
}
6062
}
6163

pkg/usecase/discover_physical_drives_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ func newTestSSD(ctrlType string, ctrlID int, slotID string) *domain.DiscoveredPh
132132
func TestExecute_NoDiscoverers(t *testing.T) {
133133
store := &mockStore{}
134134
cacheWriter := &mockCacheWriter{}
135-
uc := NewDiscoverPhysicalDrives(logr.Discard(), nil, store, cacheWriter, "node-1")
135+
uc := NewDiscoverPhysicalDrives(logr.Discard(), nil, nil, store, cacheWriter, "node-1")
136136

137137
existing, err := uc.Execute(context.Background())
138138

@@ -156,7 +156,7 @@ func TestExecute_SSDDrivesOnly(t *testing.T) {
156156
uc := NewDiscoverPhysicalDrives(
157157
logr.Discard(),
158158
[]service.PhysicalDriveDiscoverer{discoverer},
159-
nil,
159+
[]service.LogicalVolumeDiscoverer{},
160160
store,
161161
cache,
162162
"node-1",
@@ -184,7 +184,7 @@ func TestExecute_MixOfHDDAndSSD(t *testing.T) {
184184
uc := NewDiscoverPhysicalDrives(
185185
logr.Discard(),
186186
[]service.PhysicalDriveDiscoverer{discoverer},
187-
nil,
187+
[]service.LogicalVolumeDiscoverer{},
188188
store,
189189
cache,
190190
"node-1",

0 commit comments

Comments
 (0)