Add more tests for ha
All checks were successful
Run short tests / run-tests (push) Successful in 1m1s

This commit is contained in:
2024-10-23 13:53:29 +02:00
parent 2a1a40f75f
commit 234323c766
6 changed files with 128 additions and 22 deletions

View File

@@ -37,16 +37,22 @@ func HandleHaMessage(w http.ResponseWriter, r *http.Request) {
decoder.DisallowUnknownFields()
err := decoder.Decode(&message)
if err != nil {
slog.Warn(fmt.Sprintln("HandleHaMessage Error decoding request body", err))
slog.Warn(fmt.Sprintln("HandleHaMessage: Error decoding request body", err))
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
switch message.Target {
case "poo_recorder":
handlePooRecorderMsg(message)
res := handlePooRecorderMsg(message)
if !res {
http.Error(w, "Error handling poo recorder message", http.StatusInternalServerError)
}
case "location_recorder":
handleLocationRecorderMsg(message)
res := handleLocationRecorderMsg(message)
if !res {
http.Error(w, "Error handling location recorder message", http.StatusInternalServerError)
}
case "ticktick":
handleTicktickMsg(message)
default:
@@ -55,32 +61,32 @@ func HandleHaMessage(w http.ResponseWriter, r *http.Request) {
}
}
func handlePooRecorderMsg(message haMessage) {
func handlePooRecorderMsg(message haMessage) bool {
switch message.Action {
case "get_latest":
handleGetLatestPoo()
return handleGetLatestPoo()
default:
slog.Warn(fmt.Sprintln("handlePooRecorderMsg: Unknown action", message.Action))
return false
}
}
func handleLocationRecorderMsg(message haMessage) {
func handleLocationRecorderMsg(message haMessage) bool {
if message.Action == "record" {
port := viper.GetString("port")
req, err := http.NewRequest("POST", "http://localhost:"+port+"/location/record", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\"")))
if err != nil {
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error creating request to location recorder", err))
return
}
req.Header.Set("Content-Type", "application/json")
client := &http.Client{
Timeout: time.Second * 1,
}
_, err = client.Do(req)
_, err := client.Post("http://localhost:"+port+"/location/record", "application/json", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\"")))
if err != nil {
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error sending request to location recorder", err))
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg: Error sending request to location recorder", err))
return false
}
} else {
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Unknown action", message.Action))
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg: Unknown action", message.Action))
return false
}
return true
}
func handleTicktickMsg(message haMessage) {
@@ -97,7 +103,7 @@ func handleGetLatestPoo() bool {
port := viper.GetString("port")
_, err := client.Get("http://localhost:" + port + "/poo/latest")
if err != nil {
slog.Warn(fmt.Sprintln("handleGetLatestPoo Error sending request to poo recorder", err))
slog.Warn(fmt.Sprintln("handleGetLatestPoo: Error sending request to poo recorder", err))
return false
}