logger_test.go

package logger

import (
	"context"
	"math/rand"
	"net/http"
	"testing"
)

func TestReqID(t *testing.T) {
	t.Run("check exist key", func(t *testing.T) {
		reqID := randString(16)
		r, err := http.NewRequest("GET", "http://localhost/test", nil)
		if err != nil {
			t.Fatal(err)
		}
		r.Header.Set("x-request-id", reqID)
		r = ReqWithLoggerContext(r)
		ctxVal := r.Context().Value(requestIDKey{})
		if ctxVal == nil {
			t.Fatal("key not found")
		}
		result, ok := ctxVal.(string)
		if !ok {
			t.Fatal("incorrect key type")
		}
		if result != reqID {
			t.Fatal("result is incorrect")
		}
		Logger()
	})
	t.Run("check correct key value", func(t *testing.T) {
		reqID := randString(16)
		r, err := http.NewRequest("GET", "http://localhost/test", nil)
		if err != nil {
			t.Fatal(err)
		}
		r.Header.Set("x-request-id", reqID)

		type testType struct{}
		r = r.WithContext(context.WithValue(context.Background(), testType{}, "test"))
		r = ReqWithLoggerContext(r)
		r = r.WithContext(context.WithValue(r.Context(), testType{}, "test2"))

		ctxVal := r.Context().Value(requestIDKey{})
		if ctxVal == nil {
			t.Fatal("key not found")
		}
		result, ok := ctxVal.(string)
		if !ok {
			t.Fatal("incorrect key type")
		}
		if result != reqID {
			t.Fatal("result is incorrect")
		}
	})
	t.Run("outcome request", func(t *testing.T) {
		reqID := randString(16)
		r, err := http.NewRequest("GET", "http://localhost/test", nil)
		if err != nil {
			t.Fatal(err)
		}
		r.Header.Set("x-request-id", reqID)
		r = ReqWithLoggerContext(r)

		outReq, err := http.NewRequest("GET", "http://localhost/test", nil)
		if err != nil {
			t.Fatal(err)
		}
		AddHeadersFromContext(outReq.Header, r.Context())
		result := outReq.Header.Get("x-request-id")
		if result != reqID {
			t.Fatal("no request ID in result")
		}
	})
}

func randString(n int) string {
	b := make([]byte, n)
	const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
	for i := range b {
		b[i] = letterBytes[rand.Intn(len(letterBytes))]
	}
	return string(b)
}