From 2ef530daa47b301a40c1ee93cd43b8f36fc68c0b Mon Sep 17 00:00:00 2001
From: tjpcc <tjp@ctrl-c.club>
Date: Tue, 17 Jan 2023 15:59:29 -0700
Subject: pull request, response, handlers out of the gemini package

---
 gemini/response_test.go | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

(limited to 'gemini/response_test.go')

diff --git a/gemini/response_test.go b/gemini/response_test.go
index 616fac4..9287d71 100644
--- a/gemini/response_test.go
+++ b/gemini/response_test.go
@@ -6,14 +6,15 @@ import (
 	"io"
 	"testing"
 
+	"tildegit.org/tjp/gus"
 	"tildegit.org/tjp/gus/gemini"
 )
 
 func TestBuildResponses(t *testing.T) {
 	table := []struct {
 		name     string
-		response *gemini.Response
-		status   gemini.Status
+		response *gus.Response
+		status   gus.Status
 		meta     string
 		body     string
 	}{
@@ -137,7 +138,7 @@ func TestBuildResponses(t *testing.T) {
 				t.Errorf("expected meta %q, got %q", test.meta, test.response.Meta)
 			}
 
-			responseBytes, err := io.ReadAll(test.response)
+			responseBytes, err := io.ReadAll(gemini.NewResponseReader(test.response))
 			if err != nil {
 				t.Fatalf("error reading response body: %q", err.Error())
 			}
@@ -153,7 +154,7 @@ func TestBuildResponses(t *testing.T) {
 func TestParseResponses(t *testing.T) {
 	table := []struct {
 		input  string
-		status gemini.Status
+		status gus.Status
 		meta   string
 		body   string
 		err    error
@@ -232,7 +233,7 @@ func TestParseResponses(t *testing.T) {
 
 func TestResponseClose(t *testing.T) {
 	body := &rdCloser{Buffer: bytes.NewBufferString("the body here")}
-	resp := &gemini.Response{
+	resp := &gus.Response{
 		Status: gemini.StatusSuccess,
 		Meta:   "text/gemini",
 		Body:   body,
@@ -246,7 +247,7 @@ func TestResponseClose(t *testing.T) {
 		t.Error("response body was not closed by response.Close()")
 	}
 
-	resp = &gemini.Response{
+	resp = &gus.Response{
 		Status: gemini.StatusInput,
 		Meta:   "give me more",
 	}
@@ -269,8 +270,8 @@ func (rc *rdCloser) Close() error {
 func TestResponseWriteTo(t *testing.T) {
 	// invariant under test: WriteTo() sends the same bytes as Read()
 
-	clone := func(resp *gemini.Response) *gemini.Response {
-		other := &gemini.Response{
+	clone := func(resp *gus.Response) *gus.Response {
+		other := &gus.Response{
 			Status: resp.Status,
 			Meta:   resp.Meta,
 		}
@@ -296,7 +297,7 @@ func TestResponseWriteTo(t *testing.T) {
 
 	table := []struct {
 		name     string
-		response *gemini.Response
+		response *gus.Response
 	}{
 		{
 			name: "simple success",
@@ -316,13 +317,13 @@ func TestResponseWriteTo(t *testing.T) {
 			r1 := test.response
 			r2 := clone(test.response)
 
-			rdbuf, err := io.ReadAll(r1)
+			rdbuf, err := io.ReadAll(gemini.NewResponseReader(r1))
 			if err != nil {
 				t.Fatalf("response.Read(): %s", err.Error())
 			}
 
 			wtbuf := &bytes.Buffer{}
-			if _, err := r2.WriteTo(wtbuf); err != nil {
+			if _, err := gemini.NewResponseReader(r2).WriteTo(wtbuf); err != nil {
 				t.Fatalf("response.WriteTo(): %s", err.Error())
 			}
 
-- 
cgit v1.2.3