mirror of
https://github.com/Xevion/go-ha.git
synced 2025-12-05 23:15:07 -06:00
114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package example
|
|
|
|
import (
|
|
"log/slog"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
"gopkg.in/yaml.v3"
|
|
ga "saml.dev/gome-assistant"
|
|
)
|
|
|
|
type (
|
|
MySuite struct {
|
|
suite.Suite
|
|
app *ga.App
|
|
config *Config
|
|
suiteCtx map[string]any
|
|
}
|
|
|
|
Config struct {
|
|
Hass struct {
|
|
HAAuthToken string `yaml:"token"`
|
|
IpAddress string `yaml:"address"`
|
|
Port string `yaml:"port"`
|
|
HomeZoneEntityId string `yaml:"zone"`
|
|
}
|
|
Entities struct {
|
|
LightEntityId string `yaml:"light_entity_id"`
|
|
}
|
|
}
|
|
)
|
|
|
|
func (s *MySuite) SetupSuite() {
|
|
s.suiteCtx = make(map[string]any)
|
|
|
|
configFile, err := os.ReadFile("./config.yaml")
|
|
if err != nil {
|
|
slog.Error("Error reading config file", err)
|
|
}
|
|
s.config = &Config{}
|
|
// either env var or config file can be used to set HA auth. token
|
|
s.config.Hass.HAAuthToken = os.Getenv("HA_AUTH_TOKEN")
|
|
if err := yaml.Unmarshal(configFile, s.config); err != nil {
|
|
slog.Error("Error unmarshalling config file:", err)
|
|
}
|
|
|
|
s.app, err = ga.NewApp(ga.NewAppRequest{
|
|
HAAuthToken: s.config.Hass.HAAuthToken,
|
|
IpAddress: s.config.Hass.IpAddress,
|
|
HomeZoneEntityId: s.config.Hass.HomeZoneEntityId,
|
|
})
|
|
if err != nil {
|
|
slog.Error("Failed to createw new app:", err)
|
|
s.T().FailNow()
|
|
}
|
|
|
|
entityId := s.config.Entities.LightEntityId
|
|
if entityId != "" {
|
|
s.suiteCtx["entityCallbackInvoked"] = false
|
|
etl := ga.NewEntityListener().EntityIds(entityId).Call(s.entityCallback).Build()
|
|
s.app.RegisterEntityListeners(etl)
|
|
go s.app.Start()
|
|
}
|
|
}
|
|
|
|
func (s *MySuite) TearDownSuite() {
|
|
if s.app != nil {
|
|
s.app.Cleanup()
|
|
s.app = nil
|
|
}
|
|
}
|
|
|
|
// Basic test of light toggle service and entity listener
|
|
func (s *MySuite) TestLightService() {
|
|
entityId := s.config.Entities.LightEntityId
|
|
|
|
if entityId != "" {
|
|
initState := getEntityState(s, entityId)
|
|
s.app.GetService().Light.Toggle(entityId)
|
|
|
|
assert.EventuallyWithT(s.T(), func(c *assert.CollectT) {
|
|
newState := getEntityState(s, entityId)
|
|
assert.NotEqual(c, initState, newState)
|
|
assert.True(c, s.suiteCtx["entityCallbackInvoked"].(bool))
|
|
}, 10*time.Second, 1*time.Second, "State of light entity did not change or callback was invoked")
|
|
} else {
|
|
s.T().Skip("No light entity id provided")
|
|
}
|
|
}
|
|
|
|
// Test if event has been captured after light entity state changed
|
|
func (s *MySuite) entityCallback(se *ga.Service, st ga.State, e ga.EntityData) {
|
|
slog.Info("Entity callback called.", "entity id:", e.TriggerEntityId, "from state:", e.FromState, "to state:", e.ToState)
|
|
s.suiteCtx["entityCallbackInvoked"] = true
|
|
}
|
|
|
|
func getEntityState(s *MySuite, entityId string) string {
|
|
state, err := s.app.GetState().Get(entityId)
|
|
if err != nil {
|
|
slog.Error("Error getting entity state:", err)
|
|
s.T().FailNow()
|
|
}
|
|
slog.Info("State of entity:", "state", state.State)
|
|
return state.State
|
|
}
|
|
|
|
// Run the test suite
|
|
func TestMySuite(t *testing.T) {
|
|
suite.Run(t, new(MySuite))
|
|
}
|