Function TestDialerResolver

dialer_test.go:306–395

(t *testing.T)

}

func TestDialerResolver(t *testing.T) {
	ctx := context.TODO()

	tests := []struct {
		scenario string
		address  string
		resolver map[string][]string
	}{
		{
			scenario: "resolve domain to ip",
			address:  "example.com",
			resolver: map[string][]string{
				"example.com": {"127.0.0.1"},
			},
		},
		{
			scenario: "resolve domain to ip and port",
			address:  "example.com",
			resolver: map[string][]string{
				"example.com": {"127.0.0.1:9092"},
			},
		},
		{
			scenario: "resolve domain with port to ip",
			address:  "example.com:9092",
			resolver: map[string][]string{
				"example.com": {"127.0.0.1:9092"},
			},
		},
		{
			scenario: "resolve domain with port to ip with different port",
			address:  "example.com:9092",
			resolver: map[string][]string{
				"example.com": {"127.0.0.1:80"},
			},
		},
		{
			scenario: "resolve domain with port to ip",
			address:  "example.com:9092",
			resolver: map[string][]string{
				"example.com": {"127.0.0.1"},
			},
		},
	}

	for _, test := range tests {
		t.Run(test.scenario, func(t *testing.T) {
			topic := makeTopic()
			createTopic(t, topic, 1)
			defer deleteTopic(t, topic)

			d := Dialer{
				Resolver: &mockResolver{addrs: test.resolver},
			}

			// Write a message to ensure the partition gets created.
			w := NewWriter(WriterConfig{
				Brokers: []string{"localhost:9092"},