package stanza_test import ( "encoding/xml" "strings" "testing" "github.com/google/go-cmp/cmp" "gosrc.io/xmpp/stanza" ) func TestUnmarshalIqs(t *testing.T) { //var cs1 = new(iot.ControlSet) var tests = []struct { iqString string parsedIQ stanza.IQ }{ {"", stanza.IQ{XMLName: xml.Name{Local: "iq"}, Attrs: stanza.Attrs{Type: stanza.IQTypeSet, To: "test@localhost", Id: "1"}}}, //{"", IQ{XMLName: xml.Name{Space: "jabber:client", Local: "iq"}, PacketAttrs: PacketAttrs{To: "test@localhost", From: "server", Type: "set", Id: "2"}, Payload: cs1}}, } for _, test := range tests { parsedIQ := stanza.IQ{} err := xml.Unmarshal([]byte(test.iqString), &parsedIQ) if err != nil { t.Errorf("Unmarshal(%s) returned error", test.iqString) } if !xmlEqual(parsedIQ, test.parsedIQ) { t.Errorf("non matching items\n%s", cmp.Diff(parsedIQ, test.parsedIQ)) } } } func TestGenerateIqId(t *testing.T) { t.Parallel() iq := stanza.NewIQ(stanza.Attrs{Id: "1"}) if iq.Id != "1" { t.Errorf("NewIQ replaced id with %s", iq.Id) } iq = stanza.NewIQ(stanza.Attrs{}) if iq.Id == "" { t.Error("NewIQ did not generate an Id") } otherIq := stanza.NewIQ(stanza.Attrs{}) if iq.Id == otherIq.Id { t.Errorf("NewIQ generated two identical ids: %s", iq.Id) } } func TestGenerateIq(t *testing.T) { iq := stanza.NewIQ(stanza.Attrs{Type: stanza.IQTypeResult, From: "admin@localhost", To: "test@localhost", Id: "1"}) payload := stanza.DiscoInfo{ Identity: []stanza.Identity{ {Name: "Test Gateway", Category: "gateway", Type: "mqtt", }}, Features: []stanza.Feature{ {Var: stanza.NSDiscoInfo}, {Var: stanza.NSDiscoItems}, }, } iq.Payload = &payload data, err := xml.Marshal(iq) if err != nil { t.Errorf("cannot marshal xml structure") } if strings.Contains(string(data), " Not subscribed ` parsedIQ := stanza.IQ{} err := xml.Unmarshal([]byte(iq), &parsedIQ) if err != nil { t.Errorf("Unmarshal error: %s", iq) return } if parsedIQ.Error.Reason != "subscription-required" { t.Errorf("incorrect error value: '%s'", parsedIQ.Error.Reason) } } func TestUnknownPayload(t *testing.T) { iq := ` ` parsedIQ := stanza.IQ{} err := xml.Unmarshal([]byte(iq), &parsedIQ) if err != nil { t.Errorf("Unmarshal error: %#v (%s)", err, iq) return } if parsedIQ.Any.XMLName.Space != "unknown:ns" { t.Errorf("could not extract namespace: '%s'", parsedIQ.Any.XMLName.Space) } } func TestIsValid(t *testing.T) { type testCase struct { iq string shouldErr bool } testIQs := make(map[string]testCase) testIQs["Valid IQ"] = testCase{ ` `, false, } testIQs["Invalid IQ"] = testCase{ ` `, true, } for name, tcase := range testIQs { t.Run(name, func(st *testing.T) { parsedIQ := stanza.IQ{} err := xml.Unmarshal([]byte(tcase.iq), &parsedIQ) if err != nil { t.Errorf("Unmarshal error: %#v (%s)", err, tcase.iq) return } if !parsedIQ.IsValid() && !tcase.shouldErr { t.Errorf("failed iq validation for : %s", tcase.iq) } }) } }