1 files changed, 55 insertions, 0 deletions
diff --git a/xmpp/element_buffer_test.go b/xmpp/element_buffer_test.go
new file mode 100644
index 0000000..113a2c4
--- /dev/null
+++ b/xmpp/element_buffer_test.go
@@ -0,0 +1,55 @@
+package xmpp
+
+import (
+	"encoding/xml"
+	"strings"
+	"testing"
+)
+
+// bufTest is a struct containing a test point for the
+// xengineering.eu/limox/xmpp.elementBuffer. It contains a test XML string
+// which has to be exactly one XML element and an array of indentation levels
+// which have to be checked after each token which is parsed.
+type bufTest struct {
+	xml     string
+	levels  []int
+}
+
+func TestElementBuffer(t *testing.T) {
+	tests := []bufTest{
+		bufTest{`<stream></stream>`, []int{1, 0}},
+		bufTest{`<stream/>`, []int{1, 0}},
+		bufTest{`<a><b>testing</b></a>`, []int{1, 2, 2, 1, 0}},
+		bufTest{`<a><!-- comment --><b>testing</b></a>`, []int{1, 1, 2, 2, 1, 0}},
+		bufTest{`<!-- comment --><a><b>testing</b></a>`, []int{0, 1, 2, 2, 1, 0}},
+	}
+
+	for _, v := range tests {
+		r := strings.NewReader(v.xml)
+		d := xml.NewDecoder(r)
+		b := newElementBuffer()
+
+		i := 0
+
+		for {
+			token, err := d.Token()
+			if err != nil {
+				if i != len(v.levels) {
+					t.Fatalf("Stopped parsing at unexpected index due to error `%v`!\n", err)
+				}
+				break
+			}
+
+			err = b.add(token)
+			if err != nil {
+				t.Fatalf("add(token) failed with error `%v`!\n", err)
+			}
+
+			if b.level != v.levels[i] {
+				t.Fatalf("Indent level of xmpp.elementBuffer %d does not match value given by test data %d!\n", b.level, v.levels[i])
+			}
+
+			i += 1
+		}
+	}
+}