// Copyright 2025 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.

package db

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestFindingRepo(t *testing.T) {
	client, ctx := NewTransientDB(t)
	seriesRepo := NewSeriesRepository(client)
	findingRepo := NewFindingRepository(client)
	dtd := &dummyTestData{t, ctx, client}

	series := &Series{ExtID: "some-series"}
	err := seriesRepo.Insert(ctx, series, nil)
	assert.NoError(t, err)

	session := dtd.dummySession(series)

	// Add test steps.
	dtd.addSessionTest(session, "first", "second")

	// Add findings.
	toInsert := []*Finding{
		{
			TestName:  "first",
			Title:     "A",
			SessionID: session.ID,
		},
		{
			TestName:  "first",
			Title:     "B",
			SessionID: session.ID,
		},
		{
			TestName:  "second",
			Title:     "A",
			SessionID: session.ID,
		},
	}
	// Insert them all.
	for _, finding := range toInsert {
		err := findingRepo.mustStore(ctx, finding)
		assert.NoError(t, err, "finding=%q", finding)
	}
	// Now it should report a duplicate each time.
	for _, finding := range toInsert {
		err := findingRepo.mustStore(ctx, finding)
		assert.ErrorIs(t, err, errFindingExists)
	}

	list, err := findingRepo.ListForSession(ctx, session.ID, NoLimit)
	assert.NoError(t, err)
	assert.Equal(t, toInsert, list)
}