http_log_part_sink_test.go

package worker

import (
	"net/http"
	"net/http/httptest"
	"testing"

	gocontext "context"

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

func TestNewHTTPLogPartSink(t *testing.T) {
	ctx, cancel := gocontext.WithCancel(gocontext.TODO())
	cancel()
	lps := newHTTPLogPartSink(
		ctx,
		"http://example.org/log-parts/multi",
		uint64(1000))

	assert.NotNil(t, lps)
}

func TestHTTPLogPartSink_flush(t *testing.T) {
	lss := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == "POST" {
			w.WriteHeader(http.StatusNoContent)
			return
		}
		w.WriteHeader(http.StatusNotImplemented)
	}))
	defer lss.Close()

	httpLogPartSinksByURLMutex.Lock()
	httpLogPartSinksByURL[lss.URL] = newHTTPLogPartSink(gocontext.TODO(), lss.URL, uint64(1000))
	httpLogPartSinksByURLMutex.Unlock()

	ctx := gocontext.TODO()
	lps := newHTTPLogPartSink(ctx, lss.URL, uint64(10))
	lps.flush(gocontext.TODO())
	_ = lps.Add(ctx, &httpLogPart{
		JobID:   uint64(4),
		Content: "wat",
		Number:  3,
		Final:   false,
	})

	lps.partsBufferMutex.Lock()
	assert.Len(t, lps.partsBuffer, 1)
	lps.partsBufferMutex.Unlock()

	lps.flush(ctx)

	lps.partsBufferMutex.Lock()
	assert.Len(t, lps.partsBuffer, 0)
	lps.partsBufferMutex.Unlock()
}