More Magic tests

diff --git a/go/magic_test.go b/go/magic_test.go
index f501c85808265d4426108c452a3757978a7d9a0c9eb4396e930e0c362614d982..de6ed6e0b01877e3529ab3ba5f52b26199592cbadbd2b8c08f7f8ed05357a3aa 100644 (file)
--- a/go/magic_test.go
+++ b/go/magic_test.go
@@ -72,3 +72,23 @@ func TestMagicBadMagic(t *testing.T) {
                t.Fatal(err)
        }
 }
+
+func TestMagicVector(t *testing.T) {
+       encoded, err := EncodeBuf(Magic("test-vector"), nil)
+       if err != nil {
+               t.Fatal(err)
+       }
+       if !bytes.Equal(encoded, []byte("KEKStest-vector\x00")) {
+               t.Fail()
+       }
+}
+
+func TestMagicEmpty(t *testing.T) {
+       encoded, err := EncodeBuf(Magic(""), nil)
+       if err != nil {
+               t.Fatal(err)
+       }
+       if !bytes.Equal(encoded, append([]byte("KEKS"), bytes.Repeat([]byte{0}, 12)...)) {
+               t.Fail()
+       }
+}