Function TestClientDriverPlugin

plugin_test.go:19–57

(t *testing.T)

)

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

	ctx := context.Background()

	type testBundle struct {
		pluginDriver *TestDriverWithPlugin
	}

	setup := func(t *testing.T) (*Client[pgx.Tx], *testBundle) {
		t.Helper()

		var (
			dbPool = riversharedtest.DBPool(ctx, t)
			driver = riverpgxv5.New(dbPool)
			schema = riverdbtest.TestSchema(ctx, t, driver, nil)
			config = newTestConfig(t, schema)
		)

		pluginDriver := newDriverWithPlugin(t, dbPool)

		client, err := NewClient(pluginDriver, config)
		require.NoError(t, err)

		return client, &testBundle{
			pluginDriver: pluginDriver,
		}
	}

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

		client, bundle := setup(t)

		startClient(ctx, t, client)

		require.True(t, bundle.pluginDriver.initCalled)
	})
}

var _ driverPlugin[pgx.Tx] = &TestDriverWithPlugin{}