make tagType witten early

nbt/snbt_decode_test.go

@@ -8,9 +8,42 @@ import (
 func TestEncoder_WriteSNBT(t *testing.T) {
 	var buf bytes.Buffer
 	e := NewEncoder(&buf)
-	if err := e.WriteSNBT(`[1,2,3]`); err != nil {
-		t.Fatal(err)
-	}
-	t.Log(buf.Bytes())
+	testCases := []struct {
+		snbt string
+		nbt  []byte
+	}{
+		{`10b`, []byte{1, 0, 0, 10}},
+		{`12S`, []byte{2, 0, 0, 0, 12}},
+		{`0`, []byte{3, 0, 0, 0, 0, 0, 0}},
+		{`12L`, []byte{4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 12}},
 
+		{`""`, []byte{8, 0, 0, 0, 0}},
+		{`'""' `, []byte{8, 0, 0, 0, 2, '"', '"'}},
+		{`"ab\"c\""`, []byte{8, 0, 0, 0, 5, 'a', 'b', '"', 'c', '"'}},
+		{` "1\\23"`, []byte{8, 0, 0, 0, 4, '1', '\\', '2', '3'}},
+
+		{`{}`, []byte{10, 0, 0, 0}},
+		{`{a:1b}`, []byte{10, 0, 0, 1, 0, 1, 'a', 1, 0}},
+		{`{ a : 1b }`, []byte{10, 0, 0, 1, 0, 1, 'a', 1, 0}},
+		{`{a:1,2:c}`, []byte{10, 0, 0, 3, 0, 1, 'a', 0, 0, 0, 1, 8, 0, 1, '2', 0, 1, 'c', 0}},
+		{`{a:{b:{}}}`, []byte{10, 0, 0, 10, 0, 1, 'a', 10, 0, 1, 'b', 0, 0, 0}},
+
+		{`[]`, []byte{9, 0, 0, 0, 0, 0, 0, 0}},
+		{`[1b,2b,3b]`, []byte{9, 0, 0, 1, 0, 0, 0, 3, 1, 2, 3}},
+		{`[a,"b",'c']`, []byte{9, 0, 0, 8, 0, 0, 0, 3, 0, 1, 'a', 0, 1, 'b', 0, 1, 'c'}},
+		{`[{},{a:1b},{}]`, []byte{9, 0, 0, 10, 0, 0, 0, 3, 0, 1, 0, 1, 'a', 1, 0, 0}},
+		{`[[],[]]`, []byte{9, 0, 0, 9, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0}},
+	}
+	for i := range testCases {
+		buf.Reset()
+		if err := e.WriteSNBT(testCases[i].snbt); err != nil {
+			t.Error(err)
+			continue
+		}
+		want := testCases[i].nbt
+		got := buf.Bytes()
+		if !bytes.Equal(want, got) {
+			t.Errorf("Convert SNBT %q wrong:\nwant: % 02X\ngot:  % 02X", testCases[i].snbt, want, got)
+		}
+	}
 }