package dilog

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestContextLogger(t *testing.T) {
	logger := NewTestLogger(t)
	ctx := context.Background()

	l := FromContext(ctx)
	require.Equal(t, Discard, l)

	c := NewContext(ctx, logger)
	ll := FromContext(c)
	require.NotEqual(t, Discard, ll)
	require.Equal(t, logger, ll)
}