upstream-gitea/modules/indexer/internal/bleve/util_test.go

// Copyright 2024 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package bleve

import (
	"fmt"
	"os"
	"path/filepath"
	"testing"

	"gitea.dev/modules/setting"
	"gitea.dev/modules/test"

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

func TestIndexMetadataVersion(t *testing.T) {
	indexDir := t.TempDir()
	version, err := readIndexMetadataVersion(indexDir)
	require.NoError(t, err)
	assert.Zero(t, version)

	require.NoError(t, writeIndexMetadataVersion(indexDir, 42))
	data, err := os.ReadFile(filepath.Join(indexDir, indexMetadataFilename))
	require.NoError(t, err)
	assert.Equal(t, `{"version":42}`, string(data))

	version, err = readIndexMetadataVersion(indexDir)
	require.NoError(t, err)
	assert.Equal(t, 42, version)
}

func TestBleveGuessFuzzinessByKeyword(t *testing.T) {
	defer test.MockVariableValue(&setting.Indexer.TypeBleveMaxFuzzniess, 2)()

	scenarios := []struct {
		Input     string
		Fuzziness int // See util.go for the definition of fuzziness in this particular context
	}{
		{
			Input:     "",
			Fuzziness: 0,
		},
		{
			Input:     "Avocado",
			Fuzziness: 1,
		},
		{
			Input:     "Geschwindigkeit",
			Fuzziness: 2,
		},
		{
			Input:     "non-exist",
			Fuzziness: 0,
		},
		{
			Input:     "갃갃갃",
			Fuzziness: 0,
		},
		{
			Input:     "repo1",
			Fuzziness: 0,
		},
		{
			Input:     "avocado.md",
			Fuzziness: 0,
		},
	}

	for _, scenario := range scenarios {
		t.Run(fmt.Sprintf("Fuziniess:%s=%d", scenario.Input, scenario.Fuzziness), func(t *testing.T) {
			assert.Equal(t, scenario.Fuzziness, GuessFuzzinessByKeyword(scenario.Input))
		})
	}
}