fake_parser.go

package parser

import (
	"kool-dev/kool/core/builder"
	"sort"
	"strings"
)

// FakeParser implements all fake behaviors for using parser in tests.
type FakeParser struct {
	CalledAddLookupPath            bool
	TargetFiles                    []string
	CalledParse                    bool
	CalledParseAvailableScripts    bool
	CalledParseAvailableDetails    bool
	MockParsedCommands             map[string][]builder.Command
	MockParseError                 map[string]error
	MockScripts                    []string
	MockScriptDetails              []ScriptDetail
	MockParseAvailableScriptsError error
	MockParseAvailableDetailsError error
}

// AddLookupPath implements fake AddLookupPath behavior
func (f *FakeParser) AddLookupPath(rootPath string) (err error) {
	f.CalledAddLookupPath = true
	f.TargetFiles = append(f.TargetFiles, "kool.yml")
	return
}

// Parse implements fake Parse behavior
func (f *FakeParser) Parse(script string) (commands []builder.Command, err error) {
	f.CalledParse = true
	commands = f.MockParsedCommands[script]
	err = f.MockParseError[script]
	return
}

// ParseAvailableScripts implements fake ParseAvailableScripts behavior
func (f *FakeParser) ParseAvailableScripts(filter string) (scripts []string, err error) {
	f.CalledParseAvailableScripts = true

	if filter == "" {
		scripts = f.MockScripts
	} else {
		for _, script := range f.MockScripts {
			if strings.HasPrefix(script, filter) {
				scripts = append(scripts, script)
			}
		}
	}

	err = f.MockParseAvailableScriptsError
	return
}

// ParseAvailableScriptsDetails implements fake ParseAvailableScriptsDetails behavior
func (f *FakeParser) ParseAvailableScriptsDetails(filter string) (details []ScriptDetail, err error) {
	f.CalledParseAvailableDetails = true

	if filter == "" {
		details = append(details, f.MockScriptDetails...)
	} else {
		for _, detail := range f.MockScriptDetails {
			if strings.HasPrefix(detail.Name, filter) {
				details = append(details, detail)
			}
		}
	}

	if len(details) > 1 {
		sort.Slice(details, func(i, j int) bool {
			return details[i].Name < details[j].Name
		})
	}

	err = f.MockParseAvailableDetailsError
	return
}