sallust/context_test.go at main · xmidt-org/sallust · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// SPDX-FileCopyrightText: 2026 Comcast Cable Communications Management, LLC
// SPDX-License-Identifier: Apache-2.0

package sallust

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap"
)

func testWithNil(t *testing.T) {
	var (
		assert    = assert.New(t)
		parentCtx = context.Background()
		ctx       = With(parentCtx, nil)
	)

	assert.Equal(parentCtx, ctx)
}

func testWithNonNil(t *testing.T) {
	var (
		assert    = assert.New(t)
		parentCtx = context.Background()
		ctx       = With(parentCtx, new(zap.Logger))
	)

	assert.NotEqual(parentCtx, ctx)
}

func TestWith(t *testing.T) {
	t.Run("Nil", testWithNil)
	t.Run("NonNil", testWithNonNil)
}

func testGetNoLogger(t *testing.T) {
	var (
		assert  = assert.New(t)
		require = require.New(t)
		actual  = Get(context.Background())
	)

	require.NotNil(actual)
	assert.Equal(Default(), actual)
}

func testGetWithLogger(t *testing.T) {
	var (
		assert   = assert.New(t)
		require  = require.New(t)
		expected = new(zap.Logger)
		actual   = Get(With(context.Background(), expected))
	)

	require.NotNil(actual)
	assert.Equal(expected, actual)
}

func TestGet(t *testing.T) {
	t.Run("NoLogger", testGetNoLogger)
	t.Run("WithLogger", testGetWithLogger)
}

func testGetDefaultNoLogger(t *testing.T) {
	t.Run("NoDefault", func(t *testing.T) {
		var (
			assert  = assert.New(t)
			require = require.New(t)
			actual  = GetDefault(context.Background(), nil)
		)

		require.NotNil(actual)
		assert.Equal(Default(), actual)
	})

	t.Run("WithDefault", func(t *testing.T) {
		var (
			assert   = assert.New(t)
			require  = require.New(t)
			expected = new(zap.Logger)
			actual   = GetDefault(context.Background(), expected)
		)

		require.NotNil(actual)
		assert.Equal(expected, actual)
	})
}

func testGetDefaultWithLogger(t *testing.T) {
	var (
		assert   = assert.New(t)
		require  = require.New(t)
		expected = new(zap.Logger)
		actual   = GetDefault(With(context.Background(), expected), new(zap.Logger))
	)

	require.NotNil(actual)
	assert.Equal(expected, actual)
}

func TestGetDefault(t *testing.T) {
	t.Run("NoLogger", testGetDefaultNoLogger)
	t.Run("WithLogger", testGetDefaultWithLogger)
}