update

scanner/scanner_internal_test.go

@@ -0,0 +1,98 @@
+// nolint unused
+package scanner
+
+import (
+	"context"
+	"errors"
+	"sync/atomic"
+
+	ppl "github.com/google/go-pipeline/pkg/pipeline"
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+type mockPhase struct {
+	num           int
+	produceFunc   func() ppl.Producer[int]
+	stagesFunc    func() []ppl.Stage[int]
+	finalizeFunc  func(error) error
+	descriptionFn func() string
+}
+
+func (m *mockPhase) producer() ppl.Producer[int] {
+	return m.produceFunc()
+}
+
+func (m *mockPhase) stages() []ppl.Stage[int] {
+	return m.stagesFunc()
+}
+
+func (m *mockPhase) finalize(err error) error {
+	return m.finalizeFunc(err)
+}
+
+func (m *mockPhase) description() string {
+	return m.descriptionFn()
+}
+
+var _ = Describe("runPhase", func() {
+	var (
+		ctx      context.Context
+		phaseNum int
+		phase    *mockPhase
+		sum      atomic.Int32
+	)
+
+	BeforeEach(func() {
+		ctx = context.Background()
+		phaseNum = 1
+		phase = &mockPhase{
+			num: 3,
+			produceFunc: func() ppl.Producer[int] {
+				return ppl.NewProducer(func(put func(int)) error {
+					for i := 1; i <= phase.num; i++ {
+						put(i)
+					}
+					return nil
+				})
+			},
+			stagesFunc: func() []ppl.Stage[int] {
+				return []ppl.Stage[int]{ppl.NewStage(func(i int) (int, error) {
+					sum.Add(int32(i))
+					return i, nil
+				})}
+			},
+			finalizeFunc: func(err error) error {
+				return err
+			},
+			descriptionFn: func() string {
+				return "Mock Phase"
+			},
+		}
+	})
+
+	It("should run the phase successfully", func() {
+		err := runPhase(ctx, phaseNum, phase)()
+		Expect(err).ToNot(HaveOccurred())
+		Expect(sum.Load()).To(Equal(int32(1 * 2 * 3)))
+	})
+
+	It("should log an error if the phase fails", func() {
+		phase.finalizeFunc = func(err error) error {
+			return errors.New("finalize error")
+		}
+		err := runPhase(ctx, phaseNum, phase)()
+		Expect(err).To(HaveOccurred())
+		Expect(err.Error()).To(ContainSubstring("finalize error"))
+	})
+
+	It("should count the tasks", func() {
+		counter, countStageFn := countTasks[int]()
+		phase.stagesFunc = func() []ppl.Stage[int] {
+			return []ppl.Stage[int]{ppl.NewStage(countStageFn, ppl.Name("count tasks"))}
+		}
+		err := runPhase(ctx, phaseNum, phase)()
+		Expect(err).ToNot(HaveOccurred())
+		Expect(counter.Load()).To(Equal(int64(3)))
+	})
+})