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")
	}
}