newcontext_test.go

package httpctx

import (
	"context"
	"net/http"
	"sync"
	"testing"

	"github.com/stretchr/testify/assert"
)

func doSomethingWith(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	// ... perform processing here ...
}

type fakeWriter struct {
	CloseNotifyChan chan bool
}

func newFakeWriter() *fakeWriter {
	return &fakeWriter{
		CloseNotifyChan: make(chan bool),
	}
}

func (fw *fakeWriter) CloseNotify() <-chan bool {
	return fw.CloseNotifyChan
}

func (fw *fakeWriter) Header() http.Header {
	panic("not implemented")
}

func (fw *fakeWriter) Write([]byte) (int, error) {
	panic("not implemented")
}

func (fw *fakeWriter) WriteHeader(status int) {
	panic("not implemented")
}

func TestNewContext_CloseNotifier(t *testing.T) {
	assert := assert.New(t)
	fw := newFakeWriter()
	fr := &http.Request{}
	ctx := context.Background()

	ctx, cancelFunc := newContext(ctx, fw, fr)
	defer cancelFunc()

	var wg sync.WaitGroup
	finished := false
	wg.Add(1)
	go func(ctx context.Context) {
		<-ctx.Done()
		finished = true
		wg.Done()
	}(ctx)

	fw.CloseNotifyChan <- true
	wg.Wait()
	assert.True(finished)
}

func TestNewContext_CancelFunc(t *testing.T) {
	assert := assert.New(t)
	fw := newFakeWriter()
	fr := &http.Request{}
	ctx := context.Background()

	ctx, cancelFunc := newContext(ctx, fw, fr)

	var wg sync.WaitGroup
	finished := false
	wg.Add(1)
	go func(ctx context.Context) {
		<-ctx.Done()
		finished = true
		wg.Done()
	}(ctx)

	cancelFunc()
	wg.Wait()
	assert.True(finished)
}

func TestNewContext_WithNils(t *testing.T) {
	assert := assert.New(t)
	ctx, cancelFunc := newContext(nil, nil, nil)
	assert.NotNil(ctx)
	assert.NotNil(cancelFunc)
}