// 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())
}