package xmpp

import (
	"encoding/xml"
	"reflect"
	"testing"
)

// This is mostly just tests of the roster data structures.

func TestRosterIqMarshal(t *testing.T) {
	iq := &Iq{Header: Header{From: "from", Lang: "en",
		Nested: []interface{}{RosterQuery{}}}}
	exp := `<iq from="from" xml:lang="en"><query xmlns="` +
		NsRoster + `"></query></iq>`
	assertMarshal(t, exp, iq)
}

func TestRosterIqUnmarshal(t *testing.T) {
	str := `<iq from="from" xml:lang="en"><query xmlns="` +
		NsRoster + `"><item jid="a@b.c"/></query></iq>`
	iq := Iq{}
	xml.Unmarshal([]byte(str), &iq)
	m := make(map[xml.Name]reflect.Type)
	name := xml.Name{Space: NsRoster, Local: "query"}
	m[name] = reflect.TypeOf(RosterQuery{})
	err := parseExtended(&iq.Header, m)
	if err != nil {
		t.Fatalf("parseExtended: %v", err)
	}
	assertEquals(t, "iq", iq.XMLName.Local)
	assertEquals(t, "from", string(iq.From))
	assertEquals(t, "en", iq.Lang)
	nested := iq.Nested
	if nested == nil {
		t.Fatalf("nested nil")
	}
	if len(nested) != 1 {
		t.Fatalf("wrong size nested(%d): %v", len(nested),
			nested)
	}
	var rq *RosterQuery
	rq, ok := nested[0].(*RosterQuery)
	if !ok {
		t.Fatalf("nested not RosterQuery: %v",
			reflect.TypeOf(nested))
	}
	if len(rq.Item) != 1 {
		t.Fatalf("Wrong # items: %v", rq.Item)
	}
	item := rq.Item[0]
	assertEquals(t, "a@b.c", string(item.Jid))
}