Testing: add immutable proto test

proto_test.go

@@ -0,0 +1,36 @@
+package prox5
+
+import (
+	"testing"
+)
+
+func TestImmutableProto(t *testing.T) {
+	prt := newImmutableProto()
+	if prt.Get() != ProtoNull {
+		t.Fatal("expected protonull")
+	}
+	prt.set(ProtoSOCKS5)
+	if prt.Get() != ProtoSOCKS5 {
+		t.Fatal("expected socks5 proto")
+	}
+	prt.set(ProtoSOCKS4)
+	if prt.Get() != ProtoSOCKS5 {
+		t.Fatal("expected socks5 proto still after trying to set twice")
+	}
+	str := strs.Get()
+	defer strs.MustPut(str)
+	prt.Get().writeProtoString(str)
+	if str.String() != "socks5" {
+		t.Fatalf("expected socks5://, got %s", str.String())
+	}
+	str.MustReset()
+	prt.Get().writeProtoURI(str)
+	if str.String() != "socks5://" {
+		t.Fatalf("expected socks5://, got %s", str.String())
+	}
+	str.MustReset()
+	ptrstr := prt.Get().String()
+	if ptrstr != "socks5" {
+		t.Fatalf("expected socks5://, got %s", ptrstr)
+	}
+}