update

pkg/mqtt/discovery_test.go

@@ -0,0 +1,65 @@
+package mqtt
+
+import (
+	"encoding/json"
+	"strings"
+	"testing"
+)
+
+func TestSensorDiscoveryJSON(t *testing.T) {
+	discovery := SensorDiscovery{
+		Platform:          "mqtt",
+		Name:              "Test Sensor Temperature",
+		StateTopic:        "telldus/sensor/arctech/selflearning/1/temperature",
+		UniqueID:          "telldus_sensor_arctech_selflearning_1_temperature",
+		UnitOfMeasurement: "°C",
+		DeviceClass:       "temperature",
+		StateClass:        "measurement",
+		Device: HADevice{
+			Identifiers:  []string{"telldus_sensor_arctech_selflearning_1"},
+			Name:         "Test Sensor",
+			Manufacturer: "Telldus",
+			Model:        "arctech selflearning",
+		},
+	}
+
+	got, err := json.Marshal(discovery)
+	if err != nil {
+		t.Fatalf("json.Marshal failed: %v", err)
+	}
+
+	want := `{"platform":"mqtt","name":"Test Sensor Temperature","state_topic":"telldus/sensor/arctech/selflearning/1/temperature","unique_id":"telldus_sensor_arctech_selflearning_1_temperature","device":{"identifiers":["telldus_sensor_arctech_selflearning_1"],"name":"Test Sensor","manufacturer":"Telldus","model":"arctech selflearning"},"unit_of_measurement":"°C","device_class":"temperature","state_class":"measurement"}`
+	if strings.TrimSpace(string(got)) != want {
+		t.Fatalf("unexpected payload\nwant: %s\n got: %s", want, got)
+	}
+}
+
+func TestSwitchDiscoveryJSON(t *testing.T) {
+	discovery := SwitchDiscovery{
+		Platform:     "mqtt",
+		Name:         "Test Switch",
+		CommandTopic: "telldus/device/1/set",
+		StateTopic:   "telldus/device/1/state",
+		UniqueID:     "telldus_device_1",
+		PayloadOn:    "ON",
+		PayloadOff:   "OFF",
+		StateOn:      "ON",
+		StateOff:     "OFF",
+		Device: HADevice{
+			Identifiers:  []string{"telldus_1"},
+			Name:         "Test Switch",
+			Manufacturer: "Telldus",
+			Model:        "arctech codeswitch",
+		},
+	}
+
+	got, err := json.Marshal(discovery)
+	if err != nil {
+		t.Fatalf("json.Marshal failed: %v", err)
+	}
+
+	want := `{"platform":"mqtt","name":"Test Switch","command_topic":"telldus/device/1/set","state_topic":"telldus/device/1/state","unique_id":"telldus_device_1","device":{"identifiers":["telldus_1"],"name":"Test Switch","manufacturer":"Telldus","model":"arctech codeswitch"},"payload_on":"ON","payload_off":"OFF","state_on":"ON","state_off":"OFF"}`
+	if strings.TrimSpace(string(got)) != want {
+		t.Fatalf("unexpected payload\nwant: %s\n got: %s", want, got)
+	}
+}