61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
|
// Copyright (C) 2022 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package main_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"storj.io/common/testcontext"
|
||
|
|
"storj.io/common/uuid"
|
||
|
|
segmentverify "storj.io/storj/cmd/tools/segment-verify"
|
||
|
|
"storj.io/storj/satellite/metabase"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCSVWriter(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
|
||
|
|
var out strings.Builder
|
||
|
|
w := segmentverify.NewCustomCSVWriter(&out)
|
||
|
|
err := w.Write(ctx, []*segmentverify.Segment{
|
||
|
|
{
|
||
|
|
VerifySegment: metabase.VerifySegment{
|
||
|
|
StreamID: uuid.UUID{1, 2, 3, 4, 5, 6},
|
||
|
|
Position: metabase.SegmentPosition{Part: 10, Index: 56},
|
||
|
|
},
|
||
|
|
Status: segmentverify.Status{Retry: 1, Found: 3, NotFound: 5},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
VerifySegment: metabase.VerifySegment{
|
||
|
|
StreamID: uuid.UUID{10},
|
||
|
|
Position: metabase.SegmentPosition{Part: 1, Index: 1},
|
||
|
|
},
|
||
|
|
Status: segmentverify.Status{Retry: 1, Found: 3, NotFound: 0},
|
||
|
|
},
|
||
|
|
})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
err = w.Write(ctx, []*segmentverify.Segment{
|
||
|
|
{
|
||
|
|
VerifySegment: metabase.VerifySegment{
|
||
|
|
StreamID: uuid.UUID{11},
|
||
|
|
Position: metabase.SegmentPosition{Part: 5, Index: 2},
|
||
|
|
},
|
||
|
|
Status: segmentverify.Status{Retry: 2, Found: 5, NotFound: 9},
|
||
|
|
},
|
||
|
|
})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.NoError(t, w.Close())
|
||
|
|
|
||
|
|
require.Equal(t, ""+
|
||
|
|
"stream id,position,found,not found,retry\n"+
|
||
|
|
"01020304-0506-0000-0000-000000000000,42949673016,3,5,1\n"+
|
||
|
|
"0a000000-0000-0000-0000-000000000000,4294967297,3,0,1\n"+
|
||
|
|
"0b000000-0000-0000-0000-000000000000,21474836482,5,9,2\n",
|
||
|
|
out.String())
|
||
|
|
}
|