mirror of
https://github.com/replicatedhq/troubleshoot.git
synced 2026-08-27 00:37:20 +00:00
Gerard Nguyen
7484b10914
* store DNS collector in JSON output for analyze later * fix incorrect path * configurable dns image * make non resolvable domain configurable * nit update address field * * update dns util image * add unit test
83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
package collect
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
func TestGetKubernetesClusterIP(t *testing.T) {
|
|
k8sSvcIp := "10.0.0.1"
|
|
client := fake.NewSimpleClientset()
|
|
service := &corev1.Service{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "kubernetes",
|
|
Namespace: "default",
|
|
},
|
|
Spec: corev1.ServiceSpec{
|
|
ClusterIP: k8sSvcIp,
|
|
},
|
|
}
|
|
|
|
// Add the service to the fake clientset
|
|
_, err := client.CoreV1().Services("default").Create(context.TODO(), service, metav1.CreateOptions{})
|
|
if err != nil {
|
|
t.Fatalf("error injecting service into fake clientset: %v", err)
|
|
}
|
|
|
|
// Call the function
|
|
clusterIP, err := getKubernetesClusterIP(client, context.TODO())
|
|
if err != nil {
|
|
t.Fatalf("error getting cluster IP: %v", err)
|
|
}
|
|
|
|
// Check the result
|
|
if clusterIP != k8sSvcIp {
|
|
t.Errorf("expected %s, got %s", k8sSvcIp, clusterIP)
|
|
}
|
|
}
|
|
|
|
func TestExtractDNSQueriesFromPodLog(t *testing.T) {
|
|
podLog := `
|
|
=== /etc/resolv.conf ===
|
|
search default.svc.cluster.local svc.cluster.local cluster.local
|
|
nameserver 10.43.0.10
|
|
options ndots:5
|
|
=== dig kubernetes ===
|
|
10.43.0.1
|
|
=== dig non-existent-domain ===`
|
|
|
|
expectedResolvConf := `search default.svc.cluster.local svc.cluster.local cluster.local
|
|
nameserver 10.43.0.10
|
|
options ndots:5
|
|
`
|
|
|
|
expectedKubernetesQuery := struct {
|
|
Name string `json:"name"`
|
|
Address string `json:"address"`
|
|
}{
|
|
Name: "kubernetes",
|
|
Address: "10.43.0.1",
|
|
}
|
|
|
|
expectedNonResolvableDomainQuery := struct {
|
|
Name string `json:"name"`
|
|
Address string `json:"address"`
|
|
}{
|
|
Name: "",
|
|
Address: "",
|
|
}
|
|
|
|
dnsDebug := &DNSTroubleshootResult{}
|
|
err := extractDNSQueriesFromPodLog(podLog, dnsDebug)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, expectedResolvConf, dnsDebug.PodResolvConf)
|
|
assert.Equal(t, expectedKubernetesQuery, dnsDebug.Query.Kubernetes)
|
|
assert.Equal(t, expectedNonResolvableDomainQuery, dnsDebug.Query.NonResolvableDomain)
|
|
}
|