numscript/internal/analysis/document_symbols_test.go

package analysis_test

import (
	"slices"
	"testing"

	"github.com/hanzo-fi/numscript/internal/analysis"
	"github.com/hanzo-fi/numscript/internal/parser"

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

func TestDocumentSymbolsWhenEmpty(t *testing.T) {
	input := `set_tx_meta(@a, "k", 42)`

	program := parser.Parse(input).Value
	checkResult := analysis.CheckProgram(program)

	symbols := checkResult.GetSymbols()
	assert.Empty(t, symbols)
}

func TestDocumentSymbols(t *testing.T) {
	input := `vars {
		monetary $mon
		account $acc
	}`

	program := parser.Parse(input).Value
	checkResult := analysis.CheckProgram(program)

	symbols := checkResult.GetSymbols()

	assert.Len(t, symbols, 2)

	indexMon := slices.IndexFunc(symbols, func(s analysis.DocumentSymbol) bool {
		return s.Name == "mon"
	})
	assert.Equal(t, symbols[indexMon], analysis.DocumentSymbol{
		Name:           "mon",
		Detail:         "monetary",
		Range:          parser.RangeOfIndexed(input, "$mon", 0),
		SelectionRange: parser.RangeOfIndexed(input, "$mon", 0),
		Kind:           analysis.DocumentSymbolVariable,
	})

	indexSrc := slices.IndexFunc(symbols, func(s analysis.DocumentSymbol) bool {
		return s.Name == "acc"
	})
	assert.Equal(t, symbols[indexSrc], analysis.DocumentSymbol{
		Name:           "acc",
		Detail:         "account",
		Range:          parser.RangeOfIndexed(input, "$acc", 0),
		SelectionRange: parser.RangeOfIndexed(input, "$acc", 0),
		Kind:           analysis.DocumentSymbolVariable,
	})

}