[TASK] add http io
This commit is contained in:
parent
44f9087010
commit
17890122cf
|
@ -0,0 +1,30 @@
|
|||
// Package that provides the logic of the webserver
|
||||
package http
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Function to read data from a http request via json format (input)
|
||||
func Read(r *http.Request, to interface{}) (err error) {
|
||||
if !strings.Contains(r.Header.Get("Content-Type"), "application/json") {
|
||||
err = errors.New("no json request recieved")
|
||||
return
|
||||
}
|
||||
err = json.NewDecoder(r.Body).Decode(to)
|
||||
return
|
||||
}
|
||||
|
||||
// Function to write data as json to a http response (output)
|
||||
func Write(w http.ResponseWriter, data interface{}) {
|
||||
js, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
http.Error(w, "failed to encode response: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.Write(js)
|
||||
}
|
|
@ -0,0 +1,53 @@
|
|||
// 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")
|
||||
}
|
Loading…
Reference in New Issue