jacob-mathison/query-orchestration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Merged in feature/assertions (pull request #158)

Browse Source 
Assertions + Gen Text

* assert
This commit is contained in:
Michael McGuinness
2025-05-28 12:36:47 +00:00
parent cc2278086f
commit db469e7491
7 changed files with 290 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/document/text/documentText.go
+1 -1
View File
@@ -10,7 +10,7 @@ import (
documenttypes "queryorchestration/internal/document/types"
)
func (s *Service) getDocumentText(ctx context.Context, doc documenttypes.File) (io.Reader, error) {
func (s *Service) GetDocumentText(ctx context.Context, doc documenttypes.File) (io.Reader, error) {
pagesNum, err := doc.GetPageCount(ctx)
if err != nil {
return nil, err
internal/document/text/documentText_test.go
+12 -12
View File
@@ -37,7 +37,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
})
@@ -63,7 +63,7 @@ func TestGetDocumentText(t *testing.T) {
).
Return(nil, errors.New("no textract response"))
_, err := svc.getDocumentText(ctx, pdf)
_, err := svc.GetDocumentText(ctx, pdf)
require.EqualError(t, err, "no textract response")
})
t.Run("merged_cell_table", func(t *testing.T) {
@@ -80,7 +80,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -99,7 +99,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -118,7 +118,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -137,7 +137,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -156,7 +156,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -175,7 +175,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -195,7 +195,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -214,7 +214,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -233,7 +233,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
@@ -252,7 +252,7 @@ func TestGetDocumentText(t *testing.T) {
textExpectations(t, ctx, mockTextract, pdf, textractBaseOut)
text, err := svc.getDocumentText(ctx, pdf)
text, err := svc.GetDocumentText(ctx, pdf)
require.NoError(t, err)
assertReaders(t, txtFile, text)
internal/document/text/execute.go
+1 -1
View File
@@ -34,7 +34,7 @@ func (s *Service) executeExtraction(ctx context.Context, documentId uuid.UUID) e
return err
}
text, err := s.getDocumentText(ctx, doc)
text, err := s.GetDocumentText(ctx, doc)
if err != nil {
return err
}