新增机器人功能

2026-06-12 18:07:53 +08:00
parent 91267eb99c
commit b645907c52
13 changed files with 1494 additions and 10 deletions
@@ -0,0 +1,94 @@
+package mqtpp
+
+import "testing"
+
+func TestBuildTextMessageServiceEnvelopeRoundTrip(t *testing.T) {
+	key, err := ExpandPSK("AQ==")
+	if err != nil {
+		t.Fatalf("ExpandPSK() error = %v", err)
+	}
+
+	raw, err := BuildTextMessageServiceEnvelope(TextMessageBuildOptions{
+		FromNodeNum: 0x12345678,
+		ToNodeNum:   NodeNumBroadcast,
+		PacketID:    0x87654321,
+		ChannelID:   "LongFast",
+		GatewayID:   "!12345678",
+		Text:        "hello from bot",
+		PSK:         key,
+		Encrypt:     true,
+		ViaMQTT:     true,
+	})
+	if err != nil {
+		t.Fatalf("BuildTextMessageServiceEnvelope() error = %v", err)
+	}
+
+	valid, _, record := MQTTPP("msh/2/e/LongFast/!12345678", raw, key, Options{})
+	if !valid {
+		t.Fatalf("MQTTPP() valid = false, record = %#v", record)
+	}
+	if record["type"] != "text_message" {
+		t.Fatalf("record type = %v", record["type"])
+	}
+	if record["text"] != "hello from bot" {
+		t.Fatalf("text = %v", record["text"])
+	}
+	if record["from_num"] != uint32(0x12345678) {
+		t.Fatalf("from_num = %v", record["from_num"])
+	}
+	if record["packet_to_num"] != uint32(NodeNumBroadcast) {
+		t.Fatalf("packet_to_num = %v", record["packet_to_num"])
+	}
+	if record["decrypt_success"] != true {
+		t.Fatalf("decrypt_success = %v", record["decrypt_success"])
+	}
+}
+
+func TestBuildTextMessageServiceEnvelopeDirectRoundTrip(t *testing.T) {
+	key, err := ExpandPSK("AQ==")
+	if err != nil {
+		t.Fatalf("ExpandPSK() error = %v", err)
+	}
+
+	raw, err := BuildTextMessageServiceEnvelope(TextMessageBuildOptions{
+		FromNodeNum: 0x12345678,
+		ToNodeNum:   0x10203040,
+		PacketID:    0x11111111,
+		ChannelID:   "LongFast",
+		GatewayID:   "!12345678",
+		Text:        "direct hello",
+		PSK:         key,
+		Encrypt:     true,
+		ViaMQTT:     true,
+	})
+	if err != nil {
+		t.Fatalf("BuildTextMessageServiceEnvelope() error = %v", err)
+	}
+
+	valid, _, record := MQTTPP("msh/2/e/LongFast/!12345678", raw, key, Options{})
+	if !valid {
+		t.Fatalf("MQTTPP() valid = false, record = %#v", record)
+	}
+	if record["text"] != "direct hello" {
+		t.Fatalf("text = %v", record["text"])
+	}
+	if record["packet_to"] != "!10203040" {
+		t.Fatalf("packet_to = %v", record["packet_to"])
+	}
+	if record["packet_to_num"] != uint32(0x10203040) {
+		t.Fatalf("packet_to_num = %v", record["packet_to_num"])
+	}
+}
+
+func TestParseNodeID(t *testing.T) {
+	num, err := ParseNodeID("!1234abcd")
+	if err != nil {
+		t.Fatalf("ParseNodeID() error = %v", err)
+	}
+	if num != 0x1234abcd {
+		t.Fatalf("num = %#x", num)
+	}
+	if NodeNumToID(num) != "!1234abcd" {
+		t.Fatalf("NodeNumToID() = %s", NodeNumToID(num))
+	}
+}