Michael McGuinness
752fb2e2c0
Text Extraction Clean Up + Parallelization * merged_Call * directchildren * bitofimprovements * go * parallel * fixfullsuite * pondforconcurrency * comments * muchdone * bitsofclean * stabilisedtests * snappy * threadpooltests * childelements * testspassed
37 lines
889 B
Go
37 lines
889 B
Go
package test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/textract"
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func GetTextractFileResponse(t testing.TB, filename string) []map[string]*textract.AnalyzeDocumentOutput {
|
|
file, err := os.Open(filename)
|
|
require.NoError(t, err)
|
|
defer file.Close()
|
|
|
|
decoder := json.NewDecoder(file)
|
|
var out []map[string]*textract.AnalyzeDocumentOutput
|
|
err = decoder.Decode(&out)
|
|
require.NoError(t, err)
|
|
|
|
return out
|
|
}
|
|
|
|
func AssertStringToReader(t testing.TB, expected string, actual io.Reader) {
|
|
actualBytes, err := io.ReadAll(actual)
|
|
require.NoError(t, err)
|
|
actualStr := string(actualBytes)
|
|
if !assert.Equal(t, expected, actualStr) {
|
|
diff := cmp.Diff(expected, actualStr)
|
|
t.Logf("Detailed diff (-expected +actual):\n%s", diff)
|
|
}
|
|
}
|