Function TestFailover

plugin/forward/setup_test.go:601–673

(t *testing.T)

}

func TestFailover(t *testing.T) {
	server_fail_s := dnstest.NewMultipleServer(func(w dns.ResponseWriter, r *dns.Msg) {
		ret := new(dns.Msg)
		ret.SetRcode(r, dns.RcodeServerFailure)
		w.WriteMsg(ret)
	})
	defer server_fail_s.Close()

	server_refused_s := dnstest.NewMultipleServer(func(w dns.ResponseWriter, r *dns.Msg) {
		ret := new(dns.Msg)
		ret.SetRcode(r, dns.RcodeRefused)
		w.WriteMsg(ret)
	})
	defer server_refused_s.Close()

	s := dnstest.NewMultipleServer(func(w dns.ResponseWriter, r *dns.Msg) {
		ret := new(dns.Msg)
		ret.SetReply(r)
		ret.Answer = append(ret.Answer, test.A("example.org. IN A 127.0.0.1"))
		w.WriteMsg(ret)
	})
	defer s.Close()

	tests := []struct {
		input     string
		hasRecord bool
		failMsg   string
	}{
		{fmt.Sprintf(
			`forward . %s %s %s {
				policy sequential
				failover ServFail Refused
				}`, server_fail_s.Addr, server_refused_s.Addr, s.Addr), true, "If failover is set, records should be returned as long as one of the upstreams is work"},
		{fmt.Sprintf(
			`forward . %s %s %s {
				policy sequential
				}`, server_fail_s.Addr, server_refused_s.Addr, s.Addr), false, "If failover is not set and the first upstream is not work, no records should be returned"},
		{fmt.Sprintf(
			`forward . %s %s %s {
				policy sequential
				}`, s.Addr, server_fail_s.Addr, server_refused_s.Addr), true, "Although failover is not set, as long as the first upstream is work, there should be has a record return"},
	}

	for i, testCase := range tests {
		c := caddy.NewTestController("dns", testCase.input)
		fs, err := parseForward(c)

		f := fs[0]
		if err != nil {
			t.Errorf("Test #%d: Failed to create forwarder: %s", i, err)
		}
		f.OnStartup()
		defer f.OnShutdown()

		// Reduce per-upstream read timeout to make the test fit within the
		// per-query deadline defaultTimeout of 5 seconds.
		for _, p := range f.proxies {
			p.SetReadTimeout(500 * time.Millisecond)