github.com/outbrain/consul@v1.4.5/agent/structs/testing_connect_proxy_config.go (about)

     1  package structs
     2  
     3  import "github.com/mitchellh/go-testing-interface"
     4  
     5  // TestConnectProxyConfig returns a ConnectProxyConfig representing a valid
     6  // Connect proxy.
     7  func TestConnectProxyConfig(t testing.T) ConnectProxyConfig {
     8  	return ConnectProxyConfig{
     9  		DestinationServiceName: "web",
    10  		Upstreams:              TestUpstreams(t),
    11  	}
    12  }
    13  
    14  // TestUpstreams returns a set of upstreams to be used in tests exercising most
    15  // important configuration patterns.
    16  func TestUpstreams(t testing.T) Upstreams {
    17  	return Upstreams{
    18  		{
    19  			// We rely on this one having default type in a few tests...
    20  			DestinationName: "db",
    21  			LocalBindPort:   9191,
    22  			Config: map[string]interface{}{
    23  				// Float because this is how it is decoded by JSON decoder so this
    24  				// enables the value returned to be compared directly to a decoded JSON
    25  				// response without spurios type loss.
    26  				"connect_timeout_ms": float64(1000),
    27  			},
    28  		},
    29  		{
    30  			DestinationType:  UpstreamDestTypePreparedQuery,
    31  			DestinationName:  "geo-cache",
    32  			LocalBindPort:    8181,
    33  			LocalBindAddress: "127.10.10.10",
    34  		},
    35  	}
    36  }
    37  
    38  // TestAddDefaultsToUpstreams takes an array of upstreams (such as that from
    39  // TestUpstreams) and adds default values that are populated during
    40  // refigistration. Use this for generating the expected Upstreams value after
    41  // registration.
    42  func TestAddDefaultsToUpstreams(t testing.T, upstreams []Upstream) Upstreams {
    43  	ups := make([]Upstream, len(upstreams))
    44  	for i := range upstreams {
    45  		ups[i] = upstreams[i]
    46  		// Fill in default fields we expect to have back explicitly in a response
    47  		if ups[i].DestinationType == "" {
    48  			ups[i].DestinationType = UpstreamDestTypeService
    49  		}
    50  	}
    51  	return ups
    52  }