package stanza_test import ( "encoding/xml" "testing" "gosrc.io/xmpp/stanza" ) func TestDiscoInfo_Builder(t *testing.T) { iq := stanza.NewIQ(stanza.Attrs{Type: "get", To: "service.localhost", Id: "disco-get-1"}) disco := iq.DiscoInfo() disco.AddIdentity("Test Component", "gateway", "service") disco.AddFeatures(stanza.NSDiscoInfo, stanza.NSDiscoItems, "jabber:iq:version", "urn:xmpp:delegation:1") parsedIQ, err := marshallUnmarshall(t, iq) if err != nil { return } // Check result pp, ok := parsedIQ.Payload.(*stanza.DiscoInfo) if !ok { t.Errorf("Parsed stanza does not contain correct IQ payload") } // Check features features := []string{stanza.NSDiscoInfo, stanza.NSDiscoItems, "jabber:iq:version", "urn:xmpp:delegation:1"} if len(pp.Features) != len(features) { t.Errorf("Features length mismatch: %#v", pp.Features) } else { for i, f := range pp.Features { if f.Var != features[i] { t.Errorf("Missing feature: %s", features[i]) } } } // Check identity if len(pp.Identity) != 1 { t.Errorf("Identity length mismatch: %#v", pp.Identity) } else { if pp.Identity[0].Name != "Test Component" { t.Errorf("Incorrect identity name: %#v", pp.Identity[0].Name) } } } // Implements XEP-0030 example 17 // https://xmpp.org/extensions/xep-0030.html#example-17 func TestDiscoItems_Builder(t *testing.T) { iq := stanza.NewIQ(stanza.Attrs{Type: "result", From: "catalog.shakespeare.lit", To: "romeo@montague.net/orchard", Id: "items-2"}) iq.DiscoItems(). AddItem("catalog.shakespeare.lit", "books", "Books by and about Shakespeare"). AddItem("catalog.shakespeare.lit", "clothing", "Wear your literary taste with pride"). AddItem("catalog.shakespeare.lit", "music", "Music from the time of Shakespeare") parsedIQ, err := marshallUnmarshall(t, iq) if err != nil { return } // Check result pp, ok := parsedIQ.Payload.(*stanza.DiscoItems) if !ok { t.Errorf("Parsed stanza does not contain correct IQ payload") } // Check items items := []stanza.DiscoItem{{xml.Name{}, "catalog.shakespeare.lit", "books", "Books by and about Shakespeare"}, {xml.Name{}, "catalog.shakespeare.lit", "clothing", "Wear your literary taste with pride"}, {xml.Name{}, "catalog.shakespeare.lit", "music", "Music from the time of Shakespeare"}} if len(pp.Items) != len(items) { t.Errorf("Items length mismatch: %#v", pp.Items) } else { for i, item := range pp.Items { if item.JID != items[i].JID { t.Errorf("JID Mismatch (expected: %s): %s", items[i].JID, item.JID) } } } }