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
|
package logsource
import (
"context"
"os"
"testing"
"github.com/stretchr/testify/assert"
"gopkg.in/alecthomas/kingpin.v2"
)
func TestKubernetesLogSourceFactory_Init(t *testing.T) {
app := kingpin.New("test", "test")
factory := &kubernetesLogSourceFactory{enable: true}
factory.Init(app)
// Parse some test flags
args := []string{
"--kubernetes.enable",
"--kubernetes.namespace", "default",
"--kubernetes.service", "postfix-svc",
"--kubernetes.container", "postfix",
"--kubernetes.kubeconfig", "/path/to/kubeconfig",
}
_, err := app.Parse(args)
assert.NoError(t, err)
assert.True(t, factory.enable)
assert.Equal(t, "default", factory.namespace)
assert.Equal(t, "postfix-svc", factory.serviceName)
assert.Equal(t, "postfix", factory.containerName)
assert.Equal(t, "/path/to/kubeconfig", factory.kubeconfigPath)
}
func TestKubernetesLogSourceFactory_New_NoConfig(t *testing.T) {
ctx := context.Background()
factory := &kubernetesLogSourceFactory{enable: true}
// Should return nil when not configured
src, err := factory.New(ctx)
assert.NoError(t, err)
assert.Nil(t, src)
}
func TestKubernetesLogSource_Path(t *testing.T) {
source := &KubernetesLogSource{
namespace: "test-namespace",
podName: "test-pod",
containerName: "test-container",
}
expected := "kubernetes://test-namespace/test-pod/test-container"
assert.Equal(t, expected, source.Path())
}
func TestKubernetesLogSourceFactory_EnvironmentVariables(t *testing.T) {
// Set environment variables
os.Setenv("KUBERNETES_POD_NAME", "test-pod-from-env")
os.Setenv("KUBERNETES_NAMESPACE", "test-ns-from-env")
os.Setenv("KUBERNETES_SERVICE", "test-postfix-svc")
os.Setenv("KUBERNETES_CONTAINER", "test-container-from-env")
defer func() {
// Clean up environment variables
os.Unsetenv("KUBERNETES_POD_NAME")
os.Unsetenv("KUBERNETES_NAMESPACE")
os.Unsetenv("KUBERNETES_SERVICE")
os.Unsetenv("KUBERNETES_CONTAINER")
}()
// Create a new kingpin app and factory
app := kingpin.New("test", "test")
factory := &kubernetesLogSourceFactory{enable: true}
factory.Init(app)
// Parse empty args (should use environment variables)
_, err := app.Parse([]string{})
assert.NoError(t, err)
// Verify environment variables were used
assert.Equal(t, "test-pod-from-env", factory.podName)
assert.Equal(t, "test-ns-from-env", factory.namespace)
assert.Equal(t, "test-postfix-svc", factory.serviceName)
assert.Equal(t, "test-container-from-env", factory.containerName)
}
|
