X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;ds=sidebyside;f=ns_test.go;fp=ns_test.go;h=084cd6d3b5d96bcc8c11272b7396a7ea3f662935;hb=b8868fafef88a7b8320d6e68a3c4c4697950d1e1;hp=0000000000000000000000000000000000000000;hpb=468e7a678129bddbbbda4b6b2c2b8a555fd47468;p=netstring.git

diff --git a/ns_test.go b/ns_test.go
new file mode 100644
index 0000000..084cd6d
--- /dev/null
+++ b/ns_test.go
@@ -0,0 +1,164 @@
+/*
+netstring -- netstring format serialization library
+Copyright (C) 2015-2020 Sergey Matveev <stargrave@stargrave.org>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, version 3 of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package netstring
+
+import (
+	"bytes"
+	"io/ioutil"
+	"testing"
+	"testing/quick"
+)
+
+func TestTrivial(t *testing.T) {
+	var buf bytes.Buffer
+	w := NewWriter(&buf)
+	if n, err := w.WriteChunk([]byte{}); err != nil || n != 3 || string(buf.Next(3)) != "0:," {
+		t.FailNow()
+	}
+	if n, err := w.WriteChunk([]byte("foo")); err != nil || n != 6 {
+		t.FailNow()
+	}
+	if n, err := w.WriteChunk([]byte("barz")); err != nil || n != 7 {
+		t.FailNow()
+	}
+	if string(buf.Bytes()) != "3:foo,4:barz," {
+		t.FailNow()
+	}
+	r := NewReader(&buf)
+	if n, err := r.Next(); err != nil || n != 3 {
+		t.FailNow()
+	}
+	if r.Discard() != nil {
+		t.FailNow()
+	}
+	if n, err := r.Next(); err != nil || n != 4 {
+		t.FailNow()
+	}
+	m := make([]byte, 4)
+	if n, err := r.Read(m); err != nil || n != 4 {
+		t.FailNow()
+	}
+	if bytes.Compare(m, []byte("barz")) != 0 {
+		t.FailNow()
+	}
+}
+
+func TestSymmetric(t *testing.T) {
+	var buf bytes.Buffer
+	w := NewWriter(&buf)
+	r := NewReader(&buf)
+	f := func(datum [][]byte) bool {
+		for _, data := range datum {
+			if n, err := w.WriteChunk(data); err != nil || n <= len(data) {
+				return false
+			}
+		}
+		for _, data := range datum {
+			if n, err := r.Next(); err != nil || n != uint64(len(data)) {
+				return false
+			}
+			got, err := ioutil.ReadAll(r)
+			if err != nil || bytes.Compare(got, data) != 0 {
+				return false
+			}
+		}
+		return true
+	}
+	if err := quick.Check(f, nil); err != nil {
+		t.Error(err)
+	}
+}
+
+func TestErrors(t *testing.T) {
+	b := bytes.NewBufferString("junk")
+	r := NewReader(b)
+	if _, err := r.Next(); err == nil {
+		t.FailNow()
+	}
+
+	b = bytes.NewBufferString("1111111111111111111111:")
+	r = NewReader(b)
+	if _, err := r.Next(); err == nil {
+		t.FailNow()
+	}
+
+	b = bytes.NewBufferString("6foobar")
+	r = NewReader(b)
+	if _, err := r.Next(); err == nil {
+		t.FailNow()
+	}
+
+	data := make([]byte, 1<<10)
+
+	b = bytes.NewBufferString("0:foobar,")
+	r = NewReader(b)
+	if _, err := r.Next(); err != nil {
+		t.FailNow()
+	}
+	if _, err := r.Read(data); err == nil {
+		t.FailNow()
+	}
+
+	b = bytes.NewBufferString("0:foobar")
+	r = NewReader(b)
+	if _, err := r.Next(); err != nil {
+		t.FailNow()
+	}
+	if _, err := r.Read(data); err == nil {
+		t.FailNow()
+	}
+
+	b = bytes.NewBufferString("6:foobar")
+	r = NewReader(b)
+	if _, err := r.Next(); err != nil {
+		t.FailNow()
+	}
+	if _, err := r.Read(data); err == nil {
+		t.FailNow()
+	}
+}
+
+func TestExample(t *testing.T) {
+	var b bytes.Buffer
+	w := NewWriter(&b)
+	if n, err := w.WriteChunk([]byte("hello")); err != nil || n != 8 || b.String() != "5:hello," {
+		t.FailNow()
+	}
+	if n, err := w.WriteSize(6); err != nil || n != 2 {
+		t.FailNow()
+	}
+	if n, err := w.Write([]byte("wor")); err != nil || n != 3 {
+		t.FailNow()
+	}
+	if n, err := w.Write([]byte("ld!")); err != nil || n != 3 || b.String() != "5:hello,6:world!," {
+		t.FailNow()
+	}
+	r := NewReader(&b)
+	if size, err := r.Next(); err != nil || size != 5 {
+		t.FailNow()
+	}
+	if err := r.Discard(); err != nil {
+		t.FailNow()
+	}
+	if size, err := r.Next(); err != nil || size != 6 {
+		t.FailNow()
+	}
+	if data, err := ioutil.ReadAll(r); err != nil || string(data) != "world!" {
+		t.FailNow()
+	}
+}