97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package nats
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/micro/go-micro/broker"
|
|
"github.com/nats-io/nats"
|
|
)
|
|
|
|
var addrTestCases = []struct {
|
|
name string
|
|
description string
|
|
addrs map[string]string // expected address : set address
|
|
}{
|
|
{
|
|
"brokerOptionConstructor",
|
|
"set broker addresses through a broker.Option in constructor",
|
|
map[string]string{
|
|
"nats://192.168.10.1:5222": "192.168.10.1:5222",
|
|
"nats://10.20.10.0:4222": "10.20.10.0:4222"},
|
|
},
|
|
{
|
|
"brokerOptionInit",
|
|
"set broker addresses through a broker.Option in broker.Init()",
|
|
map[string]string{
|
|
"nats://192.168.10.1:5222": "192.168.10.1:5222",
|
|
"nats://10.20.10.0:4222": "10.20.10.0:4222"},
|
|
},
|
|
{
|
|
"natsOptionConstructor",
|
|
"set broker addresses through the nats.Option in constructor",
|
|
map[string]string{
|
|
"nats://192.168.10.1:5222": "192.168.10.1:5222",
|
|
"nats://10.20.10.0:4222": "10.20.10.0:4222"},
|
|
},
|
|
{
|
|
"default",
|
|
"check if default Address is set correctly",
|
|
map[string]string{
|
|
"nats://localhost:4222": ""},
|
|
},
|
|
}
|
|
|
|
// TestInitAddrs tests issue #100. Ensures that if the addrs is set by an option in init it will be used.
|
|
func TestInitAddrs(t *testing.T) {
|
|
|
|
for _, tc := range addrTestCases {
|
|
t.Run(fmt.Sprintf("%s: %s", tc.name, tc.description), func(t *testing.T) {
|
|
|
|
var br broker.Broker
|
|
var addrs []string
|
|
|
|
for _, addr := range tc.addrs {
|
|
addrs = append(addrs, addr)
|
|
}
|
|
|
|
switch tc.name {
|
|
case "brokerOptionConstructor":
|
|
// we know that there are just two addrs in the dict
|
|
br = NewBroker(broker.Addrs(addrs[0], addrs[1]))
|
|
br.Init()
|
|
case "brokerOptionInit":
|
|
br = NewBroker()
|
|
// we know that there are just two addrs in the dict
|
|
br.Init(broker.Addrs(addrs[0], addrs[1]))
|
|
case "natsOptionConstructor":
|
|
nopts := nats.GetDefaultOptions()
|
|
nopts.Servers = addrs
|
|
br = NewBroker(NatsOptions(nopts))
|
|
br.Init()
|
|
case "default":
|
|
br = NewBroker()
|
|
br.Init()
|
|
}
|
|
|
|
natsBroker, ok := br.(*nbroker)
|
|
if !ok {
|
|
t.Fatal("Expected broker to be of types *nbroker")
|
|
}
|
|
// check if the same amount of addrs we set has actually been set
|
|
if len(natsBroker.addrs) != len(tc.addrs) {
|
|
t.Errorf("Expected Addr count = %d, Actual Addr count = %d",
|
|
len(natsBroker.addrs), len(tc.addrs))
|
|
}
|
|
|
|
for _, addr := range natsBroker.addrs {
|
|
_, ok := tc.addrs[addr]
|
|
if !ok {
|
|
t.Errorf("Expected '%s' has not been set", addr)
|
|
}
|
|
}
|
|
})
|
|
|
|
}
|
|
}
|