Function TestWrite

context_test.go:18–55

normal write

(t *testing.T)

// normal write
func TestWrite(t *testing.T) {
	param := &InitContextParam{
		t,
		&Animal{},
		"",
		test.ToDefault,
	}

	// init param
	context := initResponseContext(param)

	exceptedObject := &Animal{
		"Black",
		true,
	}

	animalJson, err := json.Marshal(exceptedObject)
	test.Nil(t, err)

	// call function
	status := http.StatusNotFound
	_, contextErr := context.Write(status, animalJson)
	test.Nil(t, contextErr)

	// check result

	// header
	contentType := context.response.header.Get(HeaderContentType)

	//  check the default value
	test.Contains(t, CharsetUTF8, contentType)
	test.Equal(t, status, context.response.Status)

	// body
	body := string(context.response.body)

	test.Equal(t, string(animalJson), body)
}

// normal write string
func TestWriteString(t *testing.T) {