Function UnsetVariables

testing/generation/utils/statement_generator.go:692–726

UnsetVariables returns the name of all variables that do not have a definition. Sorted in ascending order.

(gen StatementGenerator)

// UnsetVariables returns the name of all variables that do not have a definition. Sorted in ascending order.
func UnsetVariables(gen StatementGenerator) ([]string, error) {
	varNames := make(map[string]struct{})
	switch gen := gen.(type) {
	case *CollectionGen:
		for _, child := range gen.children {
			children, err := UnsetVariables(child)
			if err != nil {
				return nil, err
			}
			for _, childName := range children {
				varNames[childName] = struct{}{}
			}
		}
	case *OptionalGen:
		return UnsetVariables(gen.children)
	case *OrGen:
		return UnsetVariables(Collection(gen.children...))
	case *TextGen:
		// Nothing to do here
	case *VariableGen:
		if gen.options == nil {
			return []string{gen.name}, nil
		} else {
			return UnsetVariables(gen.options)
		}
	default:
		return nil, fmt.Errorf("unknown generator encountered: %T", gen)
	}
	var varNamesSlice []string
	for varName := range varNames {
		varNamesSlice = append(varNamesSlice, varName)
	}
	sort.Strings(varNamesSlice)
	return varNamesSlice, nil
}

// copyGenerators returns a full copy of the given slice of generators. Each generator will be in its original state.
func copyGenerators(gens []StatementGenerator) []StatementGenerator {