haproxy_test.go

package main

import (
	"fmt"
	"io/ioutil"
	"testing"
	"time"

	"net/http"

	"github.com/haproxytech/haproxy-consul-connect/lib"
	"github.com/hashicorp/consul/api"
	"github.com/stretchr/testify/require"
)

func TestService(t *testing.T) {
	sd := lib.NewShutdown()
	client := startAgent(t, sd)
	defer func() {
		sd.Shutdown("test end")
		sd.Wait()
	}()

	csd, _, upstreamPorts := startConnectService(t, sd, client, &api.AgentServiceRegistration{
		Name: "source",
		ID:   "source-1",

		Connect: &api.AgentServiceConnect{
			SidecarService: &api.AgentServiceRegistration{
				Proxy: &api.AgentServiceConnectProxyConfig{
					Upstreams: []api.Upstream{
						api.Upstream{
							DestinationName: "target",
						},
					},
				},
			},
		},
	})

	tsd, servicePort, _ := startConnectService(t, sd, client, &api.AgentServiceRegistration{
		Name: "target",
		ID:   "target-1",

		Connect: &api.AgentServiceConnect{
			SidecarService: &api.AgentServiceRegistration{
				Proxy: &api.AgentServiceConnectProxyConfig{},
			},
		},
	})

	startServer(t, sd, servicePort, "hello connect")
	wait(sd, csd, tsd)
	res, err := http.Get(fmt.Sprintf("http://127.0.0.1:%d", upstreamPorts["target"]))
	require.NoError(t, err)
	require.Equal(t, 200, res.StatusCode)

	body, err := ioutil.ReadAll(res.Body)
	require.NoError(t, err)
	err = res.Body.Close()
	require.NoError(t, err)
	require.Equal(t, "hello connect", string(body))
}

func TestPreparedQuery(t *testing.T) {
	sd := lib.NewShutdown()
	client := startAgent(t, sd)
	defer func() {
		sd.Shutdown("test end")
		sd.Wait()
	}()

	_, _, err := client.PreparedQuery().Create(&api.PreparedQueryDefinition{
		Name: "pq-",
		Service: api.ServiceQuery{
			Service:     "${match(1)}",
			OnlyPassing: true,
		},
		Template: api.QueryTemplate{
			Type:   "name_prefix_match",
			Regexp: "^pq-(.+)$",
		},
	}, &api.WriteOptions{})
	require.NoError(t, err)

	csd, _, upstreamPorts := startConnectService(t, sd, client, &api.AgentServiceRegistration{
		Name: "source",
		ID:   "source-1",

		Connect: &api.AgentServiceConnect{
			SidecarService: &api.AgentServiceRegistration{
				Proxy: &api.AgentServiceConnectProxyConfig{
					Upstreams: []api.Upstream{
						api.Upstream{
							DestinationType: api.UpstreamDestTypePreparedQuery,
							DestinationName: "pq-target",
							Config: map[string]interface{}{
								"poll_interval": (100 * time.Millisecond).String(),
							},
						},
					},
				},
			},
		},
	})

	tsd, servicePort, _ := startConnectService(t, sd, client, &api.AgentServiceRegistration{
		Name: "target",
		ID:   "target-1",

		Connect: &api.AgentServiceConnect{
			SidecarService: &api.AgentServiceRegistration{
				Proxy: &api.AgentServiceConnectProxyConfig{},
			},
		},
	})

	startServer(t, sd, servicePort, "hello connect prepared query")
	wait(sd, csd, tsd)
	res, err := http.Get(fmt.Sprintf("http://127.0.0.1:%d", upstreamPorts["pq-target"]))
	require.NoError(t, err)
	require.Equal(t, 200, res.StatusCode)

	body, err := ioutil.ReadAll(res.Body)
	require.NoError(t, err)
	err = res.Body.Close()
	require.NoError(t, err)
	require.Equal(t, "hello connect prepared query", string(body))
}