fix: unit test

ovh · Jun 11, 2021 · fd66edc · fd66edc
1 parent 978b795
commit fd66edc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/engine/cdn/storage/storageunit_test.go b/engine/cdn/storage/storageunit_test.go
@@ -163,7 +163,7 @@ func TestDeduplicationCrossType(t *testing.T) {
 		Type:       sdk.CDNTypeItemRunResult,
 		Status:     sdk.CDNStatusItemCompleted,
 	}
-	iuArtifact, err := cdnUnits.NewItemUnit(ctx, cdnUnits.Buffers[1], itemArtifact)
+	iuArtifact, err := cdnUnits.NewItemUnit(ctx, cdnUnits.FileBuffer(), itemArtifact)
 	require.NoError(t, err)
 
 	// Create Destination Writer
@@ -188,7 +188,7 @@ func TestDeduplicationCrossType(t *testing.T) {
 	itemArtifact.MD5 = md5S
 	itemArtifact.Size = 0
 	itemArtifact.Status = sdk.CDNStatusItemCompleted
-	iuArtifact, err = cdnUnits.NewItemUnit(ctx, cdnUnits.Buffers[1], itemArtifact)
+	iuArtifact, err = cdnUnits.NewItemUnit(ctx, cdnUnits.FileBuffer(), itemArtifact)
 	require.NoError(t, err)
 	require.NoError(t, item.Insert(ctx, m, db, itemArtifact))
 	defer func() {