Function TestMiddleware

riverlog/river_log_test.go:254–562

(t *testing.T)

}

func TestMiddleware(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	type testBundle struct {
		clientConfig *river.Config
		driver       *riverpgxv5.Driver
		middleware   *Middleware
		tx           pgx.Tx
	}

	setup := func(t *testing.T, config *MiddlewareConfig) (*rivertest.Worker[loggingArgs, pgx.Tx], *testBundle) {
		t.Helper()

		var (
			driver     = riverpgxv5.New(nil)
			middleware = NewMiddleware(func(w io.Writer) slog.Handler {
				return slog.NewTextHandler(w, &slog.HandlerOptions{ReplaceAttr: slogutil.NoLevelTime})
			}, config)
			clientConfig = &river.Config{
				Middleware: []rivertype.Middleware{middleware},
			}
			tx     = riverdbtest.TestTxPgx(ctx, t)
			worker = &loggingWorker{}
		)

		return rivertest.NewWorker(t, driver, clientConfig, worker), &testBundle{
			clientConfig: clientConfig,
			driver:       driver,
			middleware:   middleware,
			tx:           tx,
		}
	}

	t.Run("Success", func(t *testing.T) {
		t.Parallel()

		testWorker, bundle := setup(t, nil)

		workRes, err := testWorker.Work(ctx, t, bundle.tx, loggingArgs{Message: "Logged from worker"}, nil)
		require.NoError(t, err)

		var metadataWithLog metadataWithLog
		require.NoError(t, json.Unmarshal(workRes.Job.Metadata, &metadataWithLog))

		require.Equal(t, []logAttempt{
			{
				Attempt: 1,
				Log:     `msg="Logged from worker"` + "\n",
			},
		},
			metadataWithLog.RiverLog,
		)
	})

	t.Run("MultipleAttempts", func(t *testing.T) {
		t.Parallel()