evcc/util/homeassistant/supervisor_test.go at 8c4e92260ed29c03f06acc7275a9240c809ab094 · wlcrs/evcc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package homeassistant

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/evcc-io/evcc/util"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestSupervisorToken(t *testing.T) {
	t.Setenv(SupervisorToken, "test_supervisor_token")

	ts, ok := supervisorTokenSource(SupervisorURI)
	require.True(t, ok)

	tok, err := ts.Token()
	require.NoError(t, err)
	assert.Equal(t, "test_supervisor_token", tok.AccessToken)

	// empty uri does not match supervisor
	_, ok = supervisorTokenSource("")
	assert.False(t, ok)

	// other uri does not match supervisor
	_, ok = supervisorTokenSource("http://homeassistant.local:8123")
	assert.False(t, ok)

	// from config with SupervisorURI
	ts3, err := NewHomeAssistantFromConfig(map[string]any{"uri": SupervisorURI})
	require.NoError(t, err)

	tok3, err := ts3.Token()
	require.NoError(t, err)
	assert.Equal(t, "test_supervisor_token", tok3.AccessToken)

	// connection requires uri
	_, err = NewConnection(util.NewLogger("test"), "", "", false)
	assert.Error(t, err)

	conn, err := NewConnection(util.NewLogger("test"), SupervisorURI, "", false)
	require.NoError(t, err)
	assert.Equal(t, SupervisorURI, conn.URI())

	// test authenticated request using connection
	var authHeader string
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		authHeader = r.Header.Get("Authorization")
		w.Header().Set("Content-Type", "application/json")
		w.Write([]byte(`[{"entity_id":"sensor.test","state":"10"}]`))
	}))
	defer srv.Close()

	testConn, err := NewConnection(util.NewLogger("test"), srv.URL, "", false)
	require.NoError(t, err)
	// override instance token source directly or test via proxyInstance
	testConn.instance.TokenSource = ts

	states, err := testConn.GetStates()
	require.NoError(t, err)
	assert.Len(t, states, 1)
	assert.Equal(t, "Bearer test_supervisor_token", authHeader)
}

func TestSupervisorDiscovery(t *testing.T) {
	t.Setenv(SupervisorToken, "test_supervisor_token")

	addInstance(SupervisorInstance, SupervisorURI)
	assert.Equal(t, SupervisorURI, instanceUriByName(SupervisorInstance))
	assert.Equal(t, SupervisorInstance, instanceNameByUri(SupervisorURI))
}