Michael McGuinness
81e7223560
Text Extraction * bases * go * splitting * structure * movetoasync * movetoasync * settinguptrigger * reorder * storevent * standardisepollingvalidation * unittests * fixlint * fixlint * awscfg * generatesample * followthrough * tests * clena * store * externalidcleanup * clientid * local * baseunittests * putobjecttests * tests
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package textract_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
textract "queryorchestration/internal/serviceconfig/textract"
|
|
|
|
awstextract "github.com/aws/aws-sdk-go-v2/service/textract"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetTextractClient(t *testing.T) {
|
|
c := textract.TextractConfig{}
|
|
assert.Nil(t, c.GetTextractClient())
|
|
c.TextractClient = &awstextract.Client{}
|
|
assert.Equal(t, &awstextract.Client{}, c.GetTextractClient())
|
|
}
|
|
|
|
func TestTextractClient(t *testing.T) {
|
|
os.Clearenv()
|
|
ctx := context.Background()
|
|
c := textract.TextractConfig{}
|
|
err := c.SetTextractClient(ctx)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, c.TextractClient)
|
|
}
|
|
|
|
func TestTextractClientWithProfile(t *testing.T) {
|
|
os.Clearenv()
|
|
ctx := context.Background()
|
|
c := textract.TextractConfig{}
|
|
err := c.SetTextractClientWithProfile(ctx, "")
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, c.TextractClient)
|
|
}
|
|
|
|
func TestGetTextractURL(t *testing.T) {
|
|
c := textract.TextractConfig{}
|
|
assert.Empty(t, c.GetTextractEndpoint())
|
|
c.AWSEndpointUrlTextract = "textract_endpoint"
|
|
assert.Equal(t, "textract_endpoint", c.GetTextractEndpoint())
|
|
}
|
|
|
|
func TestSetTextractURL(t *testing.T) {
|
|
os.Clearenv()
|
|
c := textract.TextractConfig{}
|
|
c.SetTextractEndpoint("example")
|
|
assert.Equal(t, "example", c.AWSEndpointUrlTextract)
|
|
}
|