diff --git a/router_test.go b/router_test.go index f9725ba..b3d253e 100644 --- a/router_test.go +++ b/router_test.go @@ -274,6 +274,15 @@ func (s SenderMock) Send(packet stanza.Packet) error { return nil } +func (s SenderMock) SendIQ(ctx context.Context, iq stanza.IQ) (chan stanza.IQ, error) { + out, err := xml.Marshal(iq) + if err != nil { + return nil, err + } + s.buffer.Write(out) + return nil, nil +} + func (s SenderMock) SendRaw(str string) error { s.buffer.WriteString(str) return nil diff --git a/stanza/iq_test.go b/stanza/iq_test.go index 93f7ebb..54a8fc5 100644 --- a/stanza/iq_test.go +++ b/stanza/iq_test.go @@ -42,7 +42,7 @@ func TestGenerateIqId(t *testing.T) { } iq = stanza.NewIQ(stanza.Attrs{}) - if iq.Id != "1" { + if iq.Id == "" { t.Error("NewIQ did not generate an Id") }