Skip to content

Commit a28bde4

Browse files
authored
fix(cdn): random position (#5842)
1 parent df9af61 commit a28bde4

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

engine/cdn/cdn_sync_test.go

+11-2
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,17 @@ func TestSyncLog(t *testing.T) {
139139
cdnUnits.Start(ctx, sdk.NewGoRoutines(ctx))
140140
s.Units = cdnUnits
141141

142-
cdsStorage, ok := s.Units.Storages[0].(*cds.CDS)
143-
require.True(t, ok)
142+
var cdsStorage *cds.CDS
143+
for _, sto := range s.Units.Storages {
144+
cdsStorage = sto.(*cds.CDS)
145+
if cdsStorage != nil {
146+
break
147+
}
148+
}
149+
150+
if cdsStorage == nil {
151+
t.Fail()
152+
}
144153

145154
// Mock Http route
146155
gock.InterceptClient(cdsStorage.GetClient().HTTPClient())

engine/cdn/storage/storageunit_test.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ func TestDeduplicationCrossType(t *testing.T) {
163163
Type: sdk.CDNTypeItemRunResult,
164164
Status: sdk.CDNStatusItemCompleted,
165165
}
166-
iuArtifact, err := cdnUnits.NewItemUnit(ctx, cdnUnits.Buffers[1], itemArtifact)
166+
iuArtifact, err := cdnUnits.NewItemUnit(ctx, cdnUnits.FileBuffer(), itemArtifact)
167167
require.NoError(t, err)
168168

169169
// Create Destination Writer
@@ -188,7 +188,7 @@ func TestDeduplicationCrossType(t *testing.T) {
188188
itemArtifact.MD5 = md5S
189189
itemArtifact.Size = 0
190190
itemArtifact.Status = sdk.CDNStatusItemCompleted
191-
iuArtifact, err = cdnUnits.NewItemUnit(ctx, cdnUnits.Buffers[1], itemArtifact)
191+
iuArtifact, err = cdnUnits.NewItemUnit(ctx, cdnUnits.FileBuffer(), itemArtifact)
192192
require.NoError(t, err)
193193
require.NoError(t, item.Insert(ctx, m, db, itemArtifact))
194194
defer func() {

0 commit comments

Comments
 (0)