hs_monolith/lib/http/io_test.go

// Package that provides the logic of the webserver
package http

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

// Function to test write()
func TestWrite(t *testing.T) {
	assert := assert.New(t)

	w := httptest.NewRecorder()
	from := map[string]string{"a": "b"}
	Write(w, from)
	result := w.Result()

	assert.Equal([]string{"application/json"}, result.Header["Content-Type"], "no header information")
	buf := new(bytes.Buffer)
	buf.ReadFrom(result.Body)
	to := buf.String()
	assert.Equal("{\"a\":\"b\"}", to, "wrong content")

	w = httptest.NewRecorder()
	value := make(chan int)
	Write(w, value)
	result = w.Result()

	assert.Equal(http.StatusInternalServerError, result.StatusCode, "wrong statuscode")

}

// Function to test read()
func TestRead(t *testing.T) {
	assert := assert.New(t)

	to := make(map[string]string)
	r, _ := http.NewRequest("GET", "/a", strings.NewReader("{\"a\":\"b\"}"))

	r.Header["Content-Type"] = []string{"application/json"}
	err := Read(r, &to)
	assert.NoError(err, "no error")
	assert.Equal(map[string]string{"a": "b"}, to, "wrong content")

	r.Header["Content-Type"] = []string{""}
	err = Read(r, &to)
	assert.Error(err, "no error")
}