mirror of
https://github.com/gotenberg/gotenberg.git
synced 2026-08-08 00:22:14 +01:00
46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
package test
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
// AssertStatusCode checks if the given request
|
|
// returns the expected status code.
|
|
func AssertStatusCode(t *testing.T, expectedStatusCode int, srv http.Handler, req *http.Request) {
|
|
rec := httptest.NewRecorder()
|
|
srv.ServeHTTP(rec, req)
|
|
assert.Equal(t, expectedStatusCode, rec.Code)
|
|
}
|
|
|
|
// AssertDirectoryEmpty checks if given directory
|
|
// is empty.
|
|
func AssertDirectoryEmpty(t *testing.T, directory string) {
|
|
f, err := os.Open(directory)
|
|
assert.Nil(t, err)
|
|
defer f.Close() // nolint: errcheck
|
|
_, err = f.Readdir(1)
|
|
if err == nil {
|
|
return
|
|
}
|
|
assert.Equal(t, io.EOF, err)
|
|
}
|
|
|
|
// AssertConcurrent runs all functions simultaneously
|
|
// and wait until execution has completed
|
|
// or an error is encountered.
|
|
func AssertConcurrent(t *testing.T, fn func() error, amount int) {
|
|
eg := errgroup.Group{}
|
|
for i := 0; i < amount; i++ {
|
|
eg.Go(fn)
|
|
}
|
|
err := eg.Wait()
|
|
assert.NoError(t, err)
|
|
}
|