Fix tests after refactor

stanza/msg_html_test.go

@@ -3,16 +3,18 @@ package stanza_test
 import (
 	"encoding/xml"
 	"testing"
+
+	"gosrc.io/xmpp/stanza"
 )
 
 func TestHTMLGen(t *testing.T) {
 	htmlBody := "<p>Hello <b>World</b></p>"
-	msg := NewMessage(Attrs{To: "test@localhost"})
+	msg := stanza.NewMessage(stanza.Attrs{To: "test@localhost"})
 	msg.Body = "Hello World"
-	body := HTMLBody{
+	body := stanza.HTMLBody{
 		InnerXML: htmlBody,
 	}
-	html := HTML{Body: body}
+	html := stanza.HTML{Body: body}
 	msg.Extensions = append(msg.Extensions, html)
 
 	result := msg.XMPPFormat()
@@ -21,7 +23,7 @@ func TestHTMLGen(t *testing.T) {
 		t.Errorf("incorrect serialize message:\n%s", result)
 	}
 
-	parsedMessage := Message{}
+	parsedMessage := stanza.Message{}
 	if err := xml.Unmarshal([]byte(str), &parsedMessage); err != nil {
 		t.Errorf("message HTML unmarshall error: %v", err)
 		return
@@ -31,7 +33,7 @@ func TestHTMLGen(t *testing.T) {
 		t.Errorf("incorrect parsed body: '%s'", parsedMessage.Body)
 	}
 
-	var h HTML
+	var h stanza.HTML
 	if ok := parsedMessage.Get(&h); !ok {
 		t.Error("could not extract HTML body")
 	}