107 lines
2.0 KiB
Go
107 lines
2.0 KiB
Go
package agent
|
|
|
|
import (
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/scada-client"
|
|
)
|
|
|
|
func TestProviderService(t *testing.T) {
|
|
conf := DefaultConfig()
|
|
conf.Version = "0.5.0"
|
|
conf.VersionPrerelease = "rc1"
|
|
conf.AtlasJoin = true
|
|
conf.Server = true
|
|
ps := ProviderService(conf)
|
|
|
|
expect := &client.ProviderService{
|
|
Service: "consul",
|
|
ServiceVersion: "0.5.0rc1",
|
|
Capabilities: map[string]int{
|
|
"http": 1,
|
|
},
|
|
Meta: map[string]string{
|
|
"auto-join": "true",
|
|
"datacenter": "dc1",
|
|
"server": "true",
|
|
},
|
|
ResourceType: "infrastructures",
|
|
}
|
|
|
|
if !reflect.DeepEqual(ps, expect) {
|
|
t.Fatalf("bad: %v", ps)
|
|
}
|
|
}
|
|
|
|
func TestProviderConfig(t *testing.T) {
|
|
conf := DefaultConfig()
|
|
conf.Version = "0.5.0"
|
|
conf.VersionPrerelease = "rc1"
|
|
conf.AtlasJoin = true
|
|
conf.Server = true
|
|
conf.AtlasInfrastructure = "armon/test"
|
|
conf.AtlasToken = "foobarbaz"
|
|
conf.AtlasEndpoint = "foo.bar:1111"
|
|
pc := ProviderConfig(conf)
|
|
|
|
expect := &client.ProviderConfig{
|
|
Service: &client.ProviderService{
|
|
Service: "consul",
|
|
ServiceVersion: "0.5.0rc1",
|
|
Capabilities: map[string]int{
|
|
"http": 1,
|
|
},
|
|
Meta: map[string]string{
|
|
"auto-join": "true",
|
|
"datacenter": "dc1",
|
|
"server": "true",
|
|
},
|
|
ResourceType: "infrastructures",
|
|
},
|
|
Handlers: map[string]client.CapabilityProvider{
|
|
"http": nil,
|
|
},
|
|
Endpoint: "foo.bar:1111",
|
|
ResourceGroup: "armon/test",
|
|
Token: "foobarbaz",
|
|
}
|
|
|
|
if !reflect.DeepEqual(pc, expect) {
|
|
t.Fatalf("bad: %v", pc)
|
|
}
|
|
}
|
|
|
|
func TestSCADAListener(t *testing.T) {
|
|
list := newScadaListener("armon/test")
|
|
defer list.Close()
|
|
|
|
var raw interface{} = list
|
|
_, ok := raw.(net.Listener)
|
|
if !ok {
|
|
t.Fatalf("bad")
|
|
}
|
|
|
|
a, b := net.Pipe()
|
|
defer a.Close()
|
|
defer b.Close()
|
|
|
|
go list.Push(a)
|
|
out, err := list.Accept()
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if out != a {
|
|
t.Fatalf("bad")
|
|
}
|
|
}
|
|
|
|
func TestSCADAAddr(t *testing.T) {
|
|
var addr interface{} = &scadaAddr{"armon/test"}
|
|
_, ok := addr.(net.Addr)
|
|
if !ok {
|
|
t.Fatalf("bad")
|
|
}
|
|
}
|