mux_test.go

// Copyright (c) 2022, R.I. Pienaar and the Project contributors
//
// SPDX-License-Identifier: Apache-2.0

package asyncjobs

import (
	"context"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("Router", func() {
	Describe("ExternalProcess", func() {
		var (
			task   *Task
			err    error
			router *Mux
		)
		BeforeEach(func() {
			task, err = NewTask("email:new", nil)
			Expect(err).ToNot(HaveOccurred())
			router = NewTaskRouter()
		})

		It("Should handle missing commands", func() {
			Expect(router.ExternalProcess("email:new", "testdata/missing.sh")).ToNot(HaveOccurred())
			handler := router.Handler(task)
			_, err = handler(context.Background(), &defaultLogger{}, task)
			Expect(err).To(MatchError(ErrExternalCommandNotFound))
		})

		It("Should handle command failures", func() {
			Expect(router.ExternalProcess("email:new", "testdata/failing-handler.sh")).ToNot(HaveOccurred())
			handler := router.Handler(task)
			_, err = handler(context.Background(), &defaultLogger{}, task)
			Expect(err).To(MatchError(ErrExternalCommandFailed))
		})

		It("Should handle success", func() {
			Expect(router.ExternalProcess("email:new", "testdata/passing-handler.sh")).ToNot(HaveOccurred())
			handler := router.Handler(task)
			payload, err := handler(context.Background(), &defaultLogger{}, task)
			Expect(err).ToNot(HaveOccurred())
			Expect(payload).To(Equal("success\n"))
		})
	})

	Describe("Handler", func() {
		It("Should support default handler", func() {
			router := NewTaskRouter()
			router.HandleFunc("x", func(_ context.Context, _ Logger, _ *Task) (any, error) {
				return "x", nil
			})

			task, err := NewTask("y", nil)
			Expect(err).ToNot(HaveOccurred())

			handler := router.Handler(task)
			_, err = handler(nil, &defaultLogger{}, task)
			Expect(err).To(MatchError(ErrNoHandlerForTaskType))
		})

		It("Should find the correct handler", func() {
			router := NewTaskRouter()
			router.HandleFunc("", func(_ context.Context, _ Logger, _ *Task) (any, error) {
				return "custom default", nil
			})
			router.HandleFunc("things:", func(_ context.Context, _ Logger, _ *Task) (any, error) {
				return "things:", nil
			})
			router.HandleFunc("things:very:specific", func(_ context.Context, _ Logger, _ *Task) (any, error) {
				return "things:very:specific", nil
			})
			router.HandleFunc("things:specific", func(_ context.Context, _ Logger, _ *Task) (any, error) {
				return "things:specific", nil
			})

			check := func(ttype string, expected string) {
				task := &Task{Type: ttype}
				res, err := router.Handler(task)(context.Background(), &defaultLogger{}, task)
				Expect(err).ToNot(HaveOccurred())
				Expect(res).To(Equal(expected))
			}

			check("things:other", "things:")
			check("things:very:other", "things:")
			check("things:very:specific", "things:very:specific")
			check("things:specific", "things:specific")
			check("things:specific:other", "things:specific")
			check("x", "custom default")
		})
	})
})