core/http/middleware/hlsrewrite/hlsrewrite_test.go

package hlsrewrite

import (
	"os"
	"testing"

	"github.com/datarhei/core/v16/mem"
	"github.com/stretchr/testify/require"
)

func TestRewrite(t *testing.T) {
	data, err := os.ReadFile("./fixtures/data.txt")
	require.NoError(t, err)

	rewrittendata, err := os.ReadFile("./fixtures/data_rewritten.txt")
	require.NoError(t, err)

	r := &hlsRewriter{
		buffer: &mem.Buffer{},
	}

	r.Write(data)

	buffer := &mem.Buffer{}
	prefix := []byte("/path/to/foobar/")
	r.rewrite(prefix, buffer)

	require.Equal(t, rewrittendata, buffer.Bytes())
}

func BenchmarkRewrite(b *testing.B) {
	data, err := os.ReadFile("./fixtures/data.txt")
	require.NoError(b, err)

	r := &hlsRewriter{
		buffer: &mem.Buffer{},
	}

	buffer := &mem.Buffer{}
	prefix := []byte("/path/to/foobar/")

	for i := 0; i < b.N; i++ {
		r.buffer.Reset()
		r.Write(data)

		buffer.Reset()
		r.rewrite(prefix, buffer)
	}
}