package printer

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/thecodingmachine/gotenberg/internal/pkg/xerror"
	"github.com/thecodingmachine/gotenberg/internal/pkg/xlog"
	"github.com/thecodingmachine/gotenberg/test/internalpkg/printertest"
	"github.com/thecodingmachine/gotenberg/test/internalpkg/xerrortest"
	"github.com/thecodingmachine/gotenberg/test/internalpkg/xlogtest"
)

func TestMergePrinter(t *testing.T) {
	var (
		logger xlog.Logger = xlogtest.DebugLogger()
		fpaths []string    = printertest.MergeFpaths(t)
		opts   MergePrinterOptions
		dest   string
		p      Printer
		err    error
	)
	// default options.
	opts = MergePrinterOptions{
		WaitTimeout: 10.0,
	}
	p = NewMergePrinter(logger, fpaths, opts)
	dest = printertest.GenerateDestination()
	err = p.Print(dest)
	assert.Nil(t, err)
	err = os.RemoveAll(dest)
	assert.Nil(t, err)
	// should not be OK as context.Context
	// should timeout.
	opts = MergePrinterOptions{
		WaitTimeout: 0.0,
	}
	p = NewMergePrinter(logger, fpaths, opts)
	dest = printertest.GenerateDestination()
	err = p.Print(dest)
	xerrortest.AssertError(t, err)
	assert.Equal(t, xerror.TimeoutCode, xerror.Code(err))
	err = os.RemoveAll(dest)
	assert.Nil(t, err)
}