Improve presence / message test

message.go

@@ -50,5 +50,3 @@ func (msg *Message) XMPPFormat() string {
 		msg.To,
 		xmlEscape(msg.Body))
 }
-
-// TODO: Func new message to create an empty message structure without the XML tag matching elements

message_test.go

@@ -22,7 +22,7 @@ func TestGenerateMessage(t *testing.T) {
 		t.Errorf("Unmarshal(%s) returned error", data)
 	}
 
-	if !xmlEqual(parsedMessage.Body, message.Body) {
+	if !xmlEqual(parsedMessage, message) {
-		t.Errorf("non matching items\n%s", cmp.Diff(parsedMessage.Body, message.Body))
+		t.Errorf("non matching items\n%s", cmp.Diff(parsedMessage, message))
 	}
 }

presence.go

@@ -18,6 +18,18 @@ func (Presence) Name() string {
 	return "presence"
 }
 
+func NewPresence(from, to, id, lang string) Presence {
+	return Presence{
+		XMLName: xml.Name{Local: "presence"},
+		PacketAttrs: PacketAttrs{
+			Id:   id,
+			From: from,
+			To:   to,
+			Lang: lang,
+		},
+	}
+}
+
 type presenceDecoder struct{}
 
 var presence presenceDecoder

presence_test.go

@@ -0,0 +1,27 @@
+package xmpp
+
+import (
+	"encoding/xml"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func TestGeneratePresence(t *testing.T) {
+	presence := NewPresence("admin@localhost", "test@localhost", "1", "en")
+	presence.Show = "chat"
+
+	data, err := xml.Marshal(presence)
+	if err != nil {
+		t.Errorf("cannot marshal xml structure")
+	}
+
+	parsedPresence := Presence{}
+	if err = xml.Unmarshal(data, &parsedPresence); err != nil {
+		t.Errorf("Unmarshal(%s) returned error", data)
+	}
+
+	if !xmlEqual(parsedPresence, presence) {
+		t.Errorf("non matching items\n%s", cmp.Diff(parsedPresence, presence))
+	}
+}