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)
			}
		}
	}
}