setup_test.go

package redis

import (
	"testing"
	"time"

	"github.com/coredns/caddy"
)

func TestSetup(t *testing.T) {
	const defEndpoint = "127.0.0.1:6379"

	tests := []struct {
		input            string
		shouldErr        bool
		expectedNttl     time.Duration
		expectedPttl     time.Duration
		expectedEndpoint string
	}{
		{`redis`, false, maxNTTL, maxTTL, defEndpoint},
		{`redis example.nl {
					success 10
				}`, false, maxNTTL, 10 * time.Second, defEndpoint},
		{`redis example.nl {
					success 10
					denial 15
				}`, false, 15 * time.Second, 10 * time.Second, defEndpoint},
		{`redis	{
				endpoint 127.0.0.2:6379
			}`, false, maxNTTL, maxTTL, "127.0.0.2:6379"},
		{`redis	{
				endpoint 127.0.0.3
			}`, false, maxNTTL, maxTTL, "127.0.0.3:6379"},

		// fails
		{`redis example.nl {
				success 15
				denial aaa
			}`, true, maxTTL, maxTTL, defEndpoint},
		{`redis example.nl {
				positive 15
				negative aaa
			}`, true, maxTTL, maxTTL, defEndpoint},
		{`redis {
				endpoint :1:1:6379
			}`, true, maxTTL, maxTTL, defEndpoint},
		{`redis {
				endpoint 127.0.0.a
			}`, true, maxTTL, maxTTL, defEndpoint},
	}
	for i, test := range tests {
		c := caddy.NewTestController("dns", test.input)
		re, err := parse(c)
		if test.shouldErr && err == nil {
			t.Errorf("Test %v: Expected error but found nil", i)
			continue
		} else if !test.shouldErr && err != nil {
			t.Errorf("Test %v: Expected no error but found error: %v", i, err)
			continue
		}
		if test.shouldErr && err != nil {
			continue
		}

		if re.nttl != test.expectedNttl {
			t.Errorf("Test %v: Expected nttl %v but found: %v", i, test.expectedNttl, re.nttl)
		}
		if re.pttl != test.expectedPttl {
			t.Errorf("Test %v: Expected pttl %v but found: %v", i, test.expectedPttl, re.pttl)
		}
		if re.addr != test.expectedEndpoint {
			t.Errorf("Test %v: Expected endpoint %v but found: %v", i, test.expectedEndpoint, re.addr)
		}
	}
}