gobuster/gobusterdir/gobusterdir_test.go

package gobusterdir

import (
	"testing"

	"github.com/OJ/gobuster/v3/libgobuster"
)

func TestAdditionalWordsLen(t *testing.T) {
	t.Parallel()

	tt := []struct {
		testName   string
		extensions map[string]bool
	}{
		{"No extensions", map[string]bool{}},
		{"Some extensions", map[string]bool{"htm": true, "html": true, "php": true}},
	}

	globalOpts := libgobuster.Options{}
	for _, x := range tt {
		t.Run(x.testName, func(t *testing.T) {
			t.Parallel()

			opts := OptionsDir{}
			opts.ExtensionsParsed.Set = x.extensions

			d, err := New(&globalOpts, &opts, nil)
			if err != nil {
				t.Fatalf("got error creating gobusterdir: %v", err)
			}

			calculatedLen := d.AdditionalWordsLen()
			wordsLen := len(d.AdditionalWords("dummy"))

			if calculatedLen != wordsLen {
				t.Fatalf("Mismatched additional words length: %d got %d generated words %v", calculatedLen, wordsLen, d.AdditionalWords("dummy"))
			}
		})
	}
}