event_test.go RawHistoryFolder

1package domain
2
3import (
4	"context"
5	"testing"
6
7	"github.com/stretchr/testify/assert"
8)
9
10func TestIsInternalDeploy(t *testing.T) {
11	tests := []struct {
12		name     string
13		ctx      context.Context
14		expected bool
15	}{
16		{
17			name:     "returns false for plain context",
18			ctx:      context.Background(),
19			expected: false,
20		},
21		{
22			name:     "returns true for internal deploy context",
23			ctx:      WithInternalDeploy(context.Background()),
24			expected: true,
25		},
26		{
27			name:     "returns false for context with wrong type",
28			ctx:      context.WithValue(context.Background(), ContextKeyInternalDeploy, "true"),
29			expected: false,
30		},
31		{
32			name:     "returns false for context with false value",
33			ctx:      context.WithValue(context.Background(), ContextKeyInternalDeploy, false),
34			expected: false,
35		},
36	}
37
38	for _, tt := range tests {
39		t.Run(tt.name, func(t *testing.T) {
40			result := IsInternalDeploy(tt.ctx)
41			assert.Equal(t, tt.expected, result)
42		})
43	}
44}
45
46func TestWithInternalDeploy(t *testing.T) {
47	ctx := context.Background()
48
49	// Before marking
50	assert.False(t, IsInternalDeploy(ctx))
51
52	// After marking
53	internalCtx := WithInternalDeploy(ctx)
54	assert.True(t, IsInternalDeploy(internalCtx))
55
56	// Original context unchanged
57	assert.False(t, IsInternalDeploy(ctx))
58}