fix unit test

Signed-off-by: zwwhdls <zww@hdls.me>
basenana · Dec 9, 2023 · b4105f0 · b4105f0
1 parent ed83334
commit b4105f0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/cmd/apps/ingest.go b/cmd/apps/ingest.go
@@ -18,6 +18,7 @@ package apps
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/spf13/cobra"
 
@@ -40,9 +41,10 @@ var IngestCmd = &cobra.Command{
 }
 
 func ingest(ps string) error {
-	err := friday.Fri.IngestFromOriginFile(context.TODO(), ps)
+	usage, err := friday.Fri.IngestFromOriginFile(context.TODO(), ps)
 	if err != nil {
 		return err
 	}
+	fmt.Printf("Usage: %v", usage)
 	return nil
 }
diff --git a/flow/operator/ingest.go b/flow/operator/ingest.go
@@ -41,5 +41,6 @@ func (i *ingestOperator) Do(ctx context.Context, param *flow.Parameter) error {
 		Name:    source,
 		Content: knowledge,
 	}
-	return friday.Fri.IngestFromFile(context.TODO(), doc)
+	_, err := friday.Fri.IngestFromFile(context.TODO(), doc)
+	return err
 }
diff --git a/pkg/friday/ingest_test.go b/pkg/friday/ingest_test.go
@@ -50,7 +50,7 @@ var _ = Describe("TestIngest", func() {
 					Group:   0,
 				},
 			}
-			err := loFriday.Ingest(context.TODO(), elements)
+			_, err := loFriday.Ingest(context.TODO(), elements)
 			Expect(err).Should(BeNil())
 		})
 	})