diff --git a/stanza/iq_disco.go b/stanza/iq_disco.go index 8e50f90..369a7ad 100644 --- a/stanza/iq_disco.go +++ b/stanza/iq_disco.go @@ -21,6 +21,7 @@ type DiscoInfo struct { Identity []Identity `xml:"identity"` Features []Feature `xml:"feature"` ResultSet *ResultSet `xml:"set,omitempty"` + Form *Form `xml:"x,omitempty"` } // Namespace lets DiscoInfo implement the IQPayload interface diff --git a/stanza/iq_disco_test.go b/stanza/iq_disco_test.go index 87b7001..e18f518 100644 --- a/stanza/iq_disco_test.go +++ b/stanza/iq_disco_test.go @@ -16,6 +16,7 @@ func TestDiscoInfo_Builder(t *testing.T) { disco := iq.DiscoInfo() disco.AddIdentity("Test Component", "gateway", "service") disco.AddFeatures(stanza.NSDiscoInfo, stanza.NSDiscoItems, "jabber:iq:version", "urn:xmpp:delegation:1") + disco.Form = stanza.NewForm([]*stanza.Field{}, "result") parsedIQ, err := checkMarshalling(t, iq) if err != nil { @@ -48,6 +49,15 @@ func TestDiscoInfo_Builder(t *testing.T) { t.Errorf("Incorrect identity name: %#v", pp.Identity[0].Name) } } + + // Check form + if pp.Form == nil { + t.Errorf("Form is nil") + } else { + if len(pp.Form.Fields) != 0 { + t.Errorf("Form fields length mismatch: %#v", pp.Form.Fields) + } + } } // Implements XEP-0030 example 17