Add test structure
This commit is contained in:
@@ -23,19 +23,7 @@ type actionTask struct {
|
||||
DueHour int `json:"due_hour"`
|
||||
}
|
||||
|
||||
type HaMessageHandler struct {
|
||||
PooRecorderHandler func(message haMessage)
|
||||
LocationRecorderHandler func(message haMessage)
|
||||
TicktickHandler func(message haMessage)
|
||||
}
|
||||
|
||||
var DefaultHandler = HaMessageHandler{
|
||||
PooRecorderHandler: handlePooRecorderMsg,
|
||||
LocationRecorderHandler: handleLocationRecorderMsg,
|
||||
TicktickHandler: handleTicktickMsg,
|
||||
}
|
||||
|
||||
func (handler HaMessageHandler) HandleHaMessage(w http.ResponseWriter, r *http.Request) {
|
||||
func HandleHaMessage(w http.ResponseWriter, r *http.Request) {
|
||||
var message haMessage
|
||||
decoder := json.NewDecoder(r.Body)
|
||||
decoder.DisallowUnknownFields()
|
||||
@@ -48,11 +36,14 @@ func (handler HaMessageHandler) HandleHaMessage(w http.ResponseWriter, r *http.R
|
||||
|
||||
switch message.Target {
|
||||
case "poo_recorder":
|
||||
handler.PooRecorderHandler(message)
|
||||
handlePooRecorderMsg(message)
|
||||
case "location_recorder":
|
||||
handler.LocationRecorderHandler(message)
|
||||
handleLocationRecorderMsg(message)
|
||||
case "ticktick":
|
||||
handler.TicktickHandler(message)
|
||||
handleTicktickMsg(message)
|
||||
default:
|
||||
slog.Warn(fmt.Sprintln("HandleHaMessage Unknown target", message.Target))
|
||||
http.Error(w, "Unknown target", http.StatusBadRequest)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -91,7 +82,7 @@ func handleTicktickMsg(message haMessage) {
|
||||
}
|
||||
}
|
||||
|
||||
func handleGetLatestPoo() {
|
||||
func handleGetLatestPoo() bool {
|
||||
client := &http.Client{
|
||||
Timeout: time.Second * 1,
|
||||
}
|
||||
@@ -99,7 +90,10 @@ func handleGetLatestPoo() {
|
||||
_, err := client.Get("http://localhost:" + port + "/poo/latest")
|
||||
if err != nil {
|
||||
slog.Warn(fmt.Sprintln("handleGetLatestPoo Error sending request to poo recorder", err))
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func createActionTask(message haMessage) {
|
||||
|
||||
Reference in New Issue
Block a user