response_test.go

package goyave

import (
	"bufio"
	"errors"
	"fmt"
	"io"
	"net"
	"net/http"
	"net/http/httptest"
	"os"
	"strconv"
	"strings"
	"testing"

	"gorm.io/gorm"
	"goyave.dev/goyave/v4/config"
	"goyave.dev/goyave/v4/database"
)

type ResponseTestSuite struct {
	TestSuite
}

func (suite *ResponseTestSuite) getFileSize(path string) string {
	file, err := os.Open(path)
	if err != nil {
		suite.FailNow(err.Error())
	}
	stats, err := file.Stat()
	if err != nil {
		suite.FailNow(err.Error())
	}
	return strconv.FormatInt(stats.Size(), 10)
}

func (suite *ResponseTestSuite) TestResponseStatus() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	response.Status(403)
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(200, resp.StatusCode) // Not written yet
	suite.True(response.empty)
	suite.Equal(403, response.GetStatus())
	resp.Body.Close()

	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.String(403, "test")
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(403, resp.StatusCode)
	suite.False(response.empty)
	suite.Equal(403, response.GetStatus())
	resp.Body.Close()

	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.Status(403)
	response.Status(200) // Should have no effect
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(200, resp.StatusCode) // Not written yet
	suite.True(response.empty)
	suite.Equal(403, response.GetStatus())
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestResponseHeader() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	response.Header().Set("Content-Type", "application/json")
	response.Status(200)
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(200, resp.StatusCode)
	suite.Equal("application/json", resp.Header.Get("Content-Type"))
	suite.True(response.empty)
	suite.Equal(200, response.status)
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestResponseError() {
	config.Set("app.debug", true)
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	response.Error(fmt.Errorf("random error"))
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(500, resp.StatusCode)

	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("{\"error\":\"random error\"}\n", string(body))
	suite.False(response.empty)
	suite.Equal(500, response.status)
	suite.NotNil(response.err)

	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.Error("random error")
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(500, resp.StatusCode)
	suite.NotNil(response.err)

	body, err = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("{\"error\":\"random error\"}\n", string(body))
	suite.False(response.empty)
	suite.Equal(500, response.status)
	suite.NotNil(response.err)

	config.Set("app.debug", false)
	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.Error("random error")
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(500, response.GetStatus())

	body, err = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Empty(string(body))
	suite.True(response.empty)
	suite.Equal("random error", response.GetError())
	suite.Equal(500, response.status)
	config.Set("app.debug", true)

	// Take user-defined status code in debug mode
	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.Status(503)
	response.Error("random error")
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(503, response.GetStatus())

	body, err = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("{\"error\":\"random error\"}\n", string(body))
	suite.False(response.empty)
	suite.Equal("random error", response.GetError())
	suite.Equal(503, response.status)
	suite.NotNil(response.err)
}

func (suite *ResponseTestSuite) TestIsEmpty() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	suite.True(response.IsEmpty())

	response.String(http.StatusOK, "test")
	suite.False(response.IsEmpty())
}

func (suite *ResponseTestSuite) TestIsHeaderWritten() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	suite.False(response.IsHeaderWritten())

	response.Status(http.StatusOK)
	suite.False(response.IsHeaderWritten())

	response.String(http.StatusOK, "test")
	suite.True(response.IsHeaderWritten())
}

func (suite *ResponseTestSuite) TestGetStacktrace() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	suite.Empty(response.GetStacktrace())

	response.stacktrace = "fake stacktrace"
	suite.Equal("fake stacktrace", response.GetStacktrace())
}

func (suite *ResponseTestSuite) TestResponseFile() {
	size := suite.getFileSize("config/config.test.json")
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	response.File("config/config.test.json")
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(200, resp.StatusCode)
	suite.Equal("inline", resp.Header.Get("Content-Disposition"))
	suite.Equal("application/json", resp.Header.Get("Content-Type"))
	suite.Equal(size, resp.Header.Get("Content-Length"))
	suite.False(response.empty)
	suite.Equal(200, response.status)
	resp.Body.Close()

	// Test no Content-Type override
	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	response.Header().Set("Content-Type", "text/plain")
	response.File("config/config.test.json")
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()
	suite.Equal("text/plain", resp.Header.Get("Content-Type"))
	resp.Body.Close()

	// File doesn't exist
	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)
	err := response.File("config/doesntexist")
	suite.Equal("open config/doesntexist: no such file or directory", err.Error())
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()
	suite.Equal(404, response.status)
	suite.True(response.empty)
	suite.False(response.wroteHeader)
	suite.Empty(resp.Header.Get("Content-Type"))
	suite.Empty(resp.Header.Get("Content-Disposition"))
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestResponseJSON() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	response.JSON(http.StatusOK, map[string]interface{}{
		"status": "ok",
		"code":   200,
	})

	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()
	suite.Equal(200, resp.StatusCode)
	suite.Equal("application/json; charset=utf-8", resp.Header.Get("Content-Type"))
	suite.False(response.empty)

	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	if err != nil {
		panic(err)
	}
	suite.Equal("{\"code\":200,\"status\":\"ok\"}\n", string(body))
}

func (suite *ResponseTestSuite) TestResponseDownload() {
	size := suite.getFileSize("config/config.test.json")
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	response.Download("config/config.test.json", "config.json")
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(200, resp.StatusCode)
	suite.Equal("attachment; filename=\"config.json\"", resp.Header.Get("Content-Disposition"))
	suite.Equal("application/json", resp.Header.Get("Content-Type"))
	suite.Equal(size, resp.Header.Get("Content-Length"))
	suite.False(response.empty)
	suite.Equal(200, response.status)
	resp.Body.Close()

	rawRequest = httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response = newResponse(httptest.NewRecorder(), rawRequest)

	err := response.Download("config/doesntexist", "config.json")
	suite.Equal("open config/doesntexist: no such file or directory", err.Error())
	resp = response.responseWriter.(*httptest.ResponseRecorder).Result()
	suite.Equal(404, response.status)
	suite.True(response.empty)
	suite.False(response.wroteHeader)
	suite.Empty(resp.Header.Get("Content-Type"))
	suite.Empty(resp.Header.Get("Content-Disposition"))
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestResponseRedirect() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	response.Redirect("https://www.google.com")
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(308, resp.StatusCode)
	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("<a href=\"https://www.google.com\">Permanent Redirect</a>.\n\n", string(body))
	suite.False(response.empty)
	suite.Equal(308, response.status)
}

func (suite *ResponseTestSuite) TestResponseTemporaryRedirect() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)

	response.TemporaryRedirect("https://www.google.com")
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()

	suite.Equal(307, resp.StatusCode)
	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("<a href=\"https://www.google.com\">Temporary Redirect</a>.\n\n", string(body))
	suite.False(response.empty)
	suite.Equal(307, response.status)
}

func (suite *ResponseTestSuite) TestResponseCookie() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	response.Cookie(&http.Cookie{
		Name:  "cookie-name",
		Value: "test",
	})

	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()
	cookies := resp.Cookies()
	suite.Equal(1, len(cookies))
	suite.Equal("cookie-name", cookies[0].Name)
	suite.Equal("test", cookies[0].Value)
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestResponseWrite() {
	rawRequest := httptest.NewRequest("GET", "/test-route", strings.NewReader("body"))
	response := newResponse(httptest.NewRecorder(), rawRequest)
	response.Write([]byte("byte array"))
	resp := response.responseWriter.(*httptest.ResponseRecorder).Result()
	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("byte array", string(body))
	suite.False(response.empty)
}

func (suite *ResponseTestSuite) TestCreateTestResponse() {
	recorder := httptest.NewRecorder()
	response := suite.CreateTestResponse(recorder)
	suite.NotNil(response)
	if response != nil {
		suite.Equal(recorder, response.responseWriter)
	}
}

func (suite *ResponseTestSuite) TestRender() {
	// With map data
	recorder := httptest.NewRecorder()
	response := suite.CreateTestResponse(recorder)

	mapData := map[string]interface{}{
		"Status":  http.StatusNotFound,
		"Message": "Not Found.",
	}
	suite.Nil(response.Render(http.StatusNotFound, "error.txt", mapData))
	resp := recorder.Result()
	suite.Equal(404, resp.StatusCode)
	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("Error 404: Not Found.", string(body))

	// With struct data
	recorder = httptest.NewRecorder()
	response = suite.CreateTestResponse(recorder)

	structData := struct {
		Message string
		Status  int
	}{
		Status:  http.StatusNotFound,
		Message: "Not Found.",
	}
	suite.Nil(response.Render(http.StatusNotFound, "error.txt", structData))
	resp = recorder.Result()
	suite.Equal(404, resp.StatusCode)
	body, err = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("Error 404: Not Found.", string(body))
	resp.Body.Close()

	// Non-existing template and exec error
	recorder = httptest.NewRecorder()
	response = suite.CreateTestResponse(recorder)

	suite.NotNil(response.Render(http.StatusNotFound, "non-existing-template", nil))

	suite.NotNil(response.Render(http.StatusNotFound, "invalid.txt", nil))
	resp = recorder.Result()
	suite.Equal(0, response.status)
	suite.Equal(200, resp.StatusCode) // Status not written in case of error
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestRenderHTML() {
	// With map data
	recorder := httptest.NewRecorder()
	response := suite.CreateTestResponse(recorder)

	mapData := map[string]interface{}{
		"Status":  http.StatusNotFound,
		"Message": "Not Found.",
	}
	suite.Nil(response.RenderHTML(http.StatusNotFound, "error.html", mapData))
	resp := recorder.Result()
	suite.Equal(404, resp.StatusCode)
	body, err := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("<html>\n    <head></head>\n    <body>\n        <p>Error 404: Not Found.</p>\n    </body>\n</html>", string(body))

	// With struct data
	recorder = httptest.NewRecorder()
	response = suite.CreateTestResponse(recorder)

	structData := struct {
		Message string
		Status  int
	}{
		Status:  http.StatusNotFound,
		Message: "Not Found.",
	}
	suite.Nil(response.RenderHTML(http.StatusNotFound, "error.html", structData))
	resp = recorder.Result()
	suite.Equal(404, resp.StatusCode)
	body, err = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Nil(err)
	suite.Equal("<html>\n    <head></head>\n    <body>\n        <p>Error 404: Not Found.</p>\n    </body>\n</html>", string(body))

	// Non-existing template and exec error
	recorder = httptest.NewRecorder()
	response = suite.CreateTestResponse(recorder)

	suite.NotNil(response.RenderHTML(http.StatusNotFound, "non-existing-template", nil))

	suite.NotNil(response.RenderHTML(http.StatusNotFound, "invalid.txt", nil))
	resp = recorder.Result()
	suite.Equal(0, response.status)
	suite.Equal(200, resp.StatusCode) // Status not written in case of error
	resp.Body.Close()
}

func (suite *ResponseTestSuite) TestHandleDatabaseError() {
	type TestRecord struct {
		gorm.Model
	}
	config.Set("database.connection", "mysql")
	defer config.Set("database.connection", "none")
	db := database.GetConnection()

	response := newResponse(httptest.NewRecorder(), nil)
	suite.False(response.HandleDatabaseError(db.Find(&TestRecord{})))

	suite.Equal(http.StatusInternalServerError, response.status)

	db.AutoMigrate(&TestRecord{})
	defer db.Migrator().DropTable(&TestRecord{})
	response = newResponse(httptest.NewRecorder(), nil)

	suite.False(response.HandleDatabaseError(db.First(&TestRecord{}, -1)))

	suite.Equal(http.StatusNotFound, response.status)

	response = newResponse(httptest.NewRecorder(), nil)
	suite.True(response.HandleDatabaseError(db.Exec("SHOW TABLES;")))

	suite.Equal(0, response.status)

	response = newResponse(httptest.NewRecorder(), nil)
	results := []TestRecord{}
	suite.True(response.HandleDatabaseError(db.Find(&results))) // Get all but empty result should not be an error
	suite.Equal(0, response.status)
}

func (suite *ResponseTestSuite) TestHead() {
	suite.RunServer(func(router *Router) {
		router.Route("GET|HEAD", "/test", func(response *Response, r *Request) {
			response.String(http.StatusOK, "hello world")
		})
	}, func() {
		resp, err := suite.Request("HEAD", "/test", nil, nil)
		if err != nil {
			suite.Fail(err.Error())
		}
		defer resp.Body.Close()

		body := suite.GetBody(resp)
		suite.Equal(http.StatusOK, resp.StatusCode)
		suite.Equal("text/plain; charset=utf-8", resp.Header.Get("Content-Type"))
		suite.Empty(body)
	})

	suite.RunServer(func(router *Router) {
		router.Route("GET", "/test", func(response *Response, r *Request) {
			response.String(http.StatusOK, "hello world")
		})
	}, func() {
		resp, err := suite.Request("HEAD", "/test", nil, nil)
		if err != nil {
			suite.Fail(err.Error())
		}
		defer resp.Body.Close()

		body := suite.GetBody(resp)
		suite.Equal(http.StatusOK, resp.StatusCode)
		suite.Equal("text/plain; charset=utf-8", resp.Header.Get("Content-Type"))
		suite.Empty(body)
	})

	suite.RunServer(func(router *Router) {
		router.Route("POST", "/test", func(response *Response, r *Request) {
			response.String(http.StatusOK, "hello world")
		})
	}, func() {
		resp, err := suite.Request("HEAD", "/test", nil, nil)
		if err != nil {
			suite.Fail(err.Error())
		}
		defer resp.Body.Close()

		body := suite.GetBody(resp)
		suite.Equal(http.StatusMethodNotAllowed, resp.StatusCode)
		suite.Equal("application/json; charset=utf-8", resp.Header.Get("Content-Type"))
		suite.Empty(body)
	})

	suite.RunServer(func(router *Router) {
		router.Route("HEAD", "/test", func(response *Response, r *Request) {
			response.Header().Set("Content-Type", "text/plain; charset=utf-8")
			response.Status(http.StatusOK)
		})
		router.Route("GET", "/test", func(response *Response, r *Request) {
			response.String(http.StatusOK, "hello world")
		})
	}, func() {
		resp, err := suite.Request("HEAD", "/test", nil, nil)
		if err != nil {
			suite.Fail(err.Error())
		}
		defer resp.Body.Close()

		body := suite.GetBody(resp)
		suite.Equal(http.StatusOK, resp.StatusCode)
		suite.Equal("text/plain; charset=utf-8", resp.Header.Get("Content-Type"))
		suite.Empty(body)
	})
}

type hijackableRecorder struct {
	*httptest.ResponseRecorder
}

func (r *hijackableRecorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
	conn := &net.TCPConn{}
	return conn, bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn)), nil
}

func (suite *ResponseTestSuite) TestHijack() {
	recorder := &hijackableRecorder{httptest.NewRecorder()}
	req := httptest.NewRequest(http.MethodGet, "/hijack", nil)
	resp := newResponse(recorder, req)

	suite.False(resp.hijacked)
	suite.False(resp.Hijacked())

	c, b, err := resp.Hijack()
	if err != nil {
		suite.Fail(err.Error())
	}
	defer c.Close()

	suite.Nil(err)
	suite.NotNil(c)
	suite.NotNil(b)
	suite.True(resp.hijacked)
	suite.True(resp.Hijacked())

}

func (suite *ResponseTestSuite) TestErrorOnHijacked() {
	recorder := &hijackableRecorder{httptest.NewRecorder()}
	req := httptest.NewRequest(http.MethodGet, "/hijack", nil)
	resp := newResponse(recorder, req)

	c, _, _ := resp.Hijack()
	defer c.Close()

	suite.Nil(resp.error("test error"))
	res := recorder.Result()
	defer res.Body.Close()
	suite.Equal(http.StatusInternalServerError, resp.status)

	body, err := io.ReadAll(res.Body)
	suite.Nil(err)
	suite.Empty(body)
}

func (suite *ResponseTestSuite) TestHijackNotHijackable() {
	recorder := httptest.NewRecorder()

	req := httptest.NewRequest(http.MethodGet, "/hijack", nil)
	resp := newResponse(recorder, req)

	c, b, err := resp.Hijack()
	suite.Nil(c)
	suite.Nil(b)
	suite.NotNil(err)
	suite.True(errors.Is(err, ErrNotHijackable))
}

// ------------------------

type testWriter struct {
	io.Writer
	result *string
	id     string
	closed bool
}

func (w *testWriter) Write(b []byte) (int, error) {
	*w.result += w.id + string(b)
	return w.Writer.Write(b)
}

func (w *testWriter) Close() error {
	w.closed = true
	return fmt.Errorf("Test close error")
}

func (suite *ResponseTestSuite) TestChainedWriter() {
	writer := httptest.NewRecorder()
	response := newResponse(writer, nil)
	result := ""
	testWr := &testWriter{response.Writer(), &result, "0", false}
	response.SetWriter(testWr)

	response.String(http.StatusOK, "hello world")

	suite.Equal("0hello world", result)
	suite.Equal(200, response.status)
	suite.True(response.wroteHeader)
	suite.False(response.empty)

	suite.Equal("Test close error", response.close().Error())
	suite.True(testWr.closed)

	resp := writer.Result()
	body, _ := io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Equal("hello world", string(body))

	// Test double chained writer
	writer = httptest.NewRecorder()
	response = newResponse(writer, nil)
	result = ""
	testWr = &testWriter{response.Writer(), &result, "0", false}
	testWr2 := &testWriter{testWr, &result, "1", false}
	response.SetWriter(testWr2)

	response.String(http.StatusOK, "hello world")
	suite.Equal("1hello world0hello world", result)
	suite.Equal(200, response.status)
	suite.True(response.wroteHeader)
	suite.False(response.empty)
	resp = writer.Result()
	body, _ = io.ReadAll(resp.Body)
	resp.Body.Close()
	suite.Equal("hello world", string(body))
}

func TestResponseTestSuite(t *testing.T) {
	RunTest(t, new(ResponseTestSuite))
}