Stricter header validation

[netstring.git] / ns_test.go

diff --git a/ns_test.go b/ns_test.go
index dbdaa8d4c81505eca1059b8a04a5d16582fa9519..6ac07e42f662ce075532867fd06854c98d81c7bf 100644 (file)
--- a/ns_test.go
+++ b/ns_test.go
@@ -131,6 +131,18 @@ func TestErrors(t *testing.T) {
        if _, err := r.Read(data); err == nil {
                t.FailNow()
        }
+
+       b = bytes.NewBufferString(":foobar,")
+       r = NewReader(b)
+       if _, err := r.Next(); err == nil {
+               t.FailNow()
+       }
+
+       b = bytes.NewBufferString("06:foobar,")
+       r = NewReader(b)
+       if _, err := r.Next(); err == nil {
+               t.FailNow()
+       }
 }
 
 func TestExample(t *testing.T) {