go-calc-local-handlers-tests.go

func TestNotFoundHandler(t *testing.T) {
	req := httptest.NewRequest("GET", "/not-found", nil)
	rr := httptest.NewRecorder()

	notFoundHandler(rr, req)
	res := rr.Result()
	resBody, _ := ioutil.ReadAll(res.Body)

	assert.Equal(t, http.StatusNotFound, res.StatusCode)
	assert.JSONEq(t, `{"message": "not found"}`, string(resBody))
}

func TestDoHandler(t *testing.T) {
	req := httptest.NewRequest(
		"POST",
		"/do",
		strings.NewReader(`{"operation": "add", "arguments": [1,2]}`))
	rr := httptest.NewRecorder()

	doHandler()(rr, req)
	res := rr.Result()
	resBody, _ := ioutil.ReadAll(res.Body)

	assert.Equal(t, http.StatusOK, res.StatusCode)
	assert.JSONEq(
		t,
		`{"operation": "add", "arguments": [1,2], "result": 3}`,
		string(resBody))
}