iris/_examples/request-body/read-headers/main_test.go

package main

import (
	"testing"

	"github.com/kataras/iris/v12/httptest"
)

func TestReadHeaders(t *testing.T) {
	app := newApp()

	e := httptest.New(t, app)

	expectedOKBody := `myHeaders: main.myHeaders{RequestID:"373713f0-6b4b-42ea-ab9f-e2e04bc38e73", Authentication:"Bearer my-token"}`

	tests := []struct {
		headers map[string]string
		code    int
		body    string
		regex   bool
	}{
		{headers: map[string]string{
			"X-Request-Id":   "373713f0-6b4b-42ea-ab9f-e2e04bc38e73",
			"Authentication": "Bearer my-token",
		}, code: 200, body: expectedOKBody, regex: false},
		{headers: map[string]string{
			"x-request-id":   "373713f0-6b4b-42ea-ab9f-e2e04bc38e73",
			"authentication": "Bearer my-token",
		}, code: 200, body: expectedOKBody, regex: false},
		{headers: map[string]string{
			"X-Request-Id":   "373713f0-6b4b-42ea-ab9f-e2e04bc38e73",
			"Authentication": "Bearer my-token",
		}, code: 200, body: expectedOKBody, regex: false},
		{headers: map[string]string{
			"Authentication": "Bearer my-token",
		}, code: 500, body: "X-Request-Id is empty", regex: false},
		{headers: map[string]string{
			"X-Request-Id": "373713f0-6b4b-42ea-ab9f-e2e04bc38e73",
		}, code: 500, body: "Authentication is empty", regex: false},
		{headers: map[string]string{}, code: 500, body: ".*\\(and 1 other error\\)$", regex: true},
	}

	for _, tt := range tests {
		te := e.GET("/").WithHeaders(tt.headers).Expect().Status(tt.code).Body()
		if tt.regex {
			te.Match(tt.body)
		} else {
			te.Equal(tt.body)
		}
	}
}