context_test.go

package dotweb

import (
	"encoding/json"
	"fmt"
	"net/http"
	"testing"

	"github.com/devfeel/dotweb/test"
)

type Animal struct {
	Hair     string
	HasMouth bool
}

// 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) {
	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
	// 这里是一个interface数组,用例需要小心.
	contextErr := context.WriteString(string(animalJson))
	test.Nil(t, contextErr)

	// header
	contentType := context.response.header.Get(HeaderContentType)
	// 因writer中的header方法调用过http.Header默认设置
	test.Contains(t, CharsetUTF8, contentType)
	test.Equal(t, defaultHttpCode, context.response.Status)

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

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

func TestWriteJson(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
	contextErr := context.WriteJson(exceptedObject)
	test.Nil(t, contextErr)

	// header
	contentType := context.response.header.Get(HeaderContentType)
	// 因writer中的header方法调用过http.Header默认设置
	test.Equal(t, MIMEApplicationJSONCharsetUTF8, contentType)
	test.Equal(t, defaultHttpCode, context.response.Status)

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

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

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

	// init param
	context := initResponseContext(param)

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

	callback := "jsonCallBack"

	// call function
	err := context.WriteJsonp(callback, exceptedObject)
	test.Nil(t, err)

	// check result

	// header
	contentType := context.response.header.Get(HeaderContentType)
	test.Equal(t, MIMEApplicationJavaScriptCharsetUTF8, contentType)
	test.Equal(t, defaultHttpCode, context.response.Status)

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

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

	excepted := fmt.Sprint(callback, "(", string(animalJson), ");")

	test.Equal(t, excepted, body)
}