Single byte read/write tests.

1 files changed, 67 insertions, 0 deletions

diff --git a/regress/lib/libtls/gotls/tls_test.go b/regress/lib/libtls/gotls/tls_test.go
index 2331ec0be62..3a430924a72 100644
--- a/regress/lib/libtls/gotls/tls_test.go
+++ b/regress/lib/libtls/gotls/tls_test.go
@@ -118,6 +118,73 @@ func TestTLSBasic(t *testing.T) {
 	}
 }
 
+func TestTLSSingleByteReadWrite(t *testing.T) {
+	ts, u, caFile, err := newTestServer()
+	if err != nil {
+		t.Fatalf("Failed to start test server: %v", err)
+	}
+	defer os.Remove(caFile)
+	defer ts.Close()
+
+	if err := Init(); err != nil {
+		t.Fatal(err)
+	}
+
+	cfg, err := NewConfig()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cfg.Free()
+	cfg.SetCAFile(caFile)
+
+	tls, err := NewClient(cfg)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer tls.Free()
+
+	t.Logf("Connecting to %s", u.Host)
+
+	if err := tls.Connect(u.Host, ""); err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		if err := tls.Close(); err != nil {
+			t.Fatalf("Close failed: %v", err)
+		}
+	}()
+
+	for _, b := range []byte("GET / HTTP/1.0\n\n") {
+		n, err := tls.Write([]byte{b})
+		if err != nil {
+			t.Fatal(err)
+		}
+		if n != 1 {
+			t.Fatalf("Wrote byte %v, got length %d, want 1", b, n)
+		}
+	}
+
+	var body []byte
+	for {
+		buf := make([]byte, 1)
+		n, err := tls.Read(buf)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if n == 0 {
+			break
+		}
+		if n != 1 {
+			t.Fatalf("Read single byte, got length %d, want 1", n)
+		}
+		body = append(body, buf...)
+	}
+
+	if !strings.Contains(string(body), httpContent) {
+		t.Errorf("Response does not contain %q", httpContent)
+	}
+}
+
 func TestTLSInfo(t *testing.T) {
 	ts, u, caFile, err := newTestServer()
 	if err != nil {