package search

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"go.skia.org/infra/go/testutils"
	"go.skia.org/infra/golden/go/types"
)

var (
	TEST_1    = "test-1"
	DIGEST_01 = "abcefgh"
	PARAMS_01 = map[string][]string{
		"param-01": []string{"val-01"},
		"param-02": []string{"val-02"},
	}
)

func TestIntermediate(t *testing.T) {
	testutils.SmallTest(t)

	srMap := srInterMap{}
	srMap.add(TEST_1, DIGEST_01, "", nil, PARAMS_01)
	assert.Equal(t, srInterMap{TEST_1: map[string]*srIntermediate{
		DIGEST_01: &srIntermediate{
			test:   TEST_1,
			digest: DIGEST_01,
			params: PARAMS_01,
			traces: map[string]*types.GoldenTrace{},
		},
	}}, srMap)
}