ziggs/vendor/github.com/amimof/huego/bridge_test.go

package huego

import (
	"context"
	"fmt"
	"net"
	"net/http"
	"strings"
	"testing"

	"github.com/jarcoal/httpmock"
	"github.com/stretchr/testify/assert"
	"golang.org/x/net/proxy"
)

func ExampleBridge_CreateUser() {
	bridge, _ := Discover()
	user, err := bridge.CreateUser("my awesome hue app") // Link button needs to be pressed
	if err != nil {
		fmt.Printf("Error creating user: %s", err.Error())
	}
	bridge = bridge.Login(user)
	light, _ := bridge.GetLight(1)
	light.Off()
}

func TestLogin(t *testing.T) {
	b := New(hostname, username)
	c, err := b.GetConfig()
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("Logged in and got config which means that we are authorized")
	t.Logf("Name: %s, SwVersion: %s", c.Name, c.SwVersion)
}

func TestLoginUnauthorized(t *testing.T) {
	b := New(hostname, "")
	b = b.Login("invalid_password")
	_, err := b.GetLights()
	if err != nil {
		if strings.Contains(err.Error(), "unauthorized user") == false {
			t.Fatal(err)
		}
	}
	t.Logf("Bridge: %s, Username: %s", b.Host, b.User)
	t.Log("User logged in and authenticated which isn't what we want")
}

func TestUpdateBridgeConfig(t *testing.T) {
	b := New(hostname, username)
	c, err := b.GetConfig()
	if err != nil {
		t.Fatal(err)
	}
	_, err = b.UpdateConfig(c)
	if err != nil {
		t.Fatal(err)
	}
}

func TestUpdateBridgeConfigError(t *testing.T) {
	b := New(badHostname, username)
	_, err := b.GetConfig()
	if err == nil {
		t.Fatal("Expected error not to be nil")
	}
}

func TestBridge_getAPIPathError(t *testing.T) {
	b := New("invalid hostname", "")
	_, err := b.getAPIPath("/")
	assert.NotNil(t, err)
}

func TestBridge_getError(t *testing.T) {
	httpmock.Deactivate()
	defer httpmock.Activate()
	_, err := get(context.Background(), "invalid hostname", http.DefaultClient)
	assert.NotNil(t, err)
}

func TestBridge_putError(t *testing.T) {
	httpmock.Deactivate()
	defer httpmock.Activate()
	_, err := put(context.Background(), "invalid hostname", []byte("huego"), http.DefaultClient)
	assert.NotNil(t, err)
}

func TestBridge_postError(t *testing.T) {
	httpmock.Deactivate()
	defer httpmock.Activate()
	_, err := post(context.Background(), "invalid hostname", []byte("huego"), http.DefaultClient)
	assert.NotNil(t, err)
}

func TestBridge_deleteError(t *testing.T) {
	httpmock.Deactivate()
	defer httpmock.Activate()
	_, err := del(context.Background(), "invalid hostname", http.DefaultClient)
	assert.NotNil(t, err)
}

func TestNewWithClient(t *testing.T) {
	newTransport := httpmock.InitialTransport.(*http.Transport).Clone()
	newTransport.DialContext = func(ctx context.Context, network, address string) (net.Conn, error) {
		return proxy.Direct.Dial(network, address)
	}
	newClient := http.DefaultClient
	newClient.Transport = newTransport

	httpmock.ActivateNonDefault(newClient)

	b := NewWithClient(hostname, username, newClient)

	c, err := b.GetConfig()
	if err != nil {
		t.Fatal(err)
	}
	if c == nil {
		t.Fatal("failed to get config")
	}
	t.Logf("Logged in and got config which means that we are authorized")
	t.Logf("Name: %s, SwVersion: %s", c.Name, c.SwVersion)
}

func TestNewCustom(t *testing.T) {
	newTransport := httpmock.InitialTransport.(*http.Transport).Clone()
	newTransport.DialContext = func(ctx context.Context, network, address string) (net.Conn, error) {
		return proxy.Direct.Dial(network, address)
	}
	newClient := http.DefaultClient
	newClient.Transport = newTransport

	httpmock.ActivateNonDefault(newClient)

	testData := `{"name":"Philips hue","datastoreversion":"125","swversion":"1952043030","apiversion":"1.52.0",
	"mac":"ec:b5:00:00:00:01","bridgeid":"EC000CF00V0000223","factorynew":false,"replacesbridgeid":null,
	"modelid":"BSB002","starterkitid":""}`

	b, err := NewCustom([]byte(testData), hostname, newClient)
	if err != nil {
		t.Fatal(err)
	}

	b.User = username

	c, err := b.GetConfig()
	if err != nil {
		t.Fatal(err)
	}
	if c == nil {
		t.Fatal("failed to get config")
	}
	t.Logf("Logged in and got config which means that we are authorized")
	t.Logf("Name: %s, SwVersion: %s", c.Name, c.SwVersion)
}