go-xmpp/stanza/registry_test.go

package stanza

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

func TestRegistry_RegisterMsgExt(t *testing.T) {
	// Setup registry
	typeRegistry := newRegistry()

	// Register an element
	name := xml.Name{Space: "urn:xmpp:receipts", Local: "request"}
	typeRegistry.MapExtension(PKTMessage, name, ReceiptRequest{})

	// Match that element
	receipt := typeRegistry.GetMsgExtension(name)
	if receipt == nil {
		t.Error("cannot read element type from registry")
		return
	}

	switch r := receipt.(type) {
	case *ReceiptRequest:
	default:
		t.Errorf("Registry did not return expected type ReceiptRequest: %v", reflect.TypeOf(r))
	}
}

func BenchmarkRegistryGet(b *testing.B) {
	// Setup registry
	typeRegistry := newRegistry()

	// Register an element
	name := xml.Name{Space: "urn:xmpp:receipts", Local: "request"}
	typeRegistry.MapExtension(PKTMessage, name, ReceiptRequest{})

	for i := 0; i < b.N; i++ {
		// Match that element
		receipt := typeRegistry.GetExtensionType(PKTMessage, name)
		if receipt == nil {
			b.Error("cannot read element type from registry")
			return
		}
	}
}