coredns/middleware/errors/setup_test.go

package errors

import (
	"testing"

	"github.com/mholt/caddy"
)

func TestErrorsParse(t *testing.T) {
	tests := []struct {
		inputErrorsRules     string
		shouldErr            bool
		expectedErrorHandler errorHandler
	}{
		{`errors`, false, errorHandler{
			LogFile: "stdout",
		}},
		{`errors stdout`, false, errorHandler{
			LogFile: "stdout",
		}},
		{`errors errors.txt`, true, errorHandler{
			LogFile: "",
		}},
		{`errors visible`, true, errorHandler{
			LogFile: "",
		}},
		{`errors { log visible }`, true, errorHandler{
			LogFile: "stdout",
		}},
	}
	for i, test := range tests {
		c := caddy.NewTestController("dns", test.inputErrorsRules)
		actualErrorsRule, err := errorsParse(c)

		if err == nil && test.shouldErr {
			t.Errorf("Test %d didn't error, but it should have", i)
		} else if err != nil && !test.shouldErr {
			t.Errorf("Test %d errored, but it shouldn't have; got '%v'", i, err)
		}
		if actualErrorsRule.LogFile != test.expectedErrorHandler.LogFile {
			t.Errorf("Test %d expected LogFile to be %s, but got %s",
				i, test.expectedErrorHandler.LogFile, actualErrorsRule.LogFile)
		}
	}
}