Merge pull request #206 from matthisholleville/feature/add-tests-to-ingress-analyzer

pkg/analyzer/ingressAnalyzer_test.go

@@ -0,0 +1,110 @@
+package analyzer
+
+import (
+	"context"
+	"strings"
+	"testing"
+
+	"github.com/k8sgpt-ai/k8sgpt/pkg/kubernetes"
+	"github.com/magiconair/properties/assert"
+	networkingv1 "k8s.io/api/networking/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes/fake"
+)
+
+func TestIngressAnalyzer(t *testing.T) {
+	clientset := fake.NewSimpleClientset(
+		&networkingv1.Ingress{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:        "example",
+				Namespace:   "default",
+				Annotations: map[string]string{},
+			},
+		})
+	ingressAnalyzer := IngressAnalyzer{}
+	var analysisResults []Analysis
+	err := ingressAnalyzer.RunAnalysis(context.Background(),
+		&AnalysisConfiguration{
+			Namespace: "default",
+		},
+		&kubernetes.Client{
+			Client: clientset,
+		}, nil, &analysisResults)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	assert.Equal(t, len(analysisResults), 1)
+}
+
+func TestIngressAnalyzerWithMultipleIngresses(t *testing.T) {
+	clientset := fake.NewSimpleClientset(
+		&networkingv1.Ingress{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:        "example",
+				Namespace:   "default",
+				Annotations: map[string]string{},
+			},
+		},
+		&networkingv1.Ingress{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:        "example-2",
+				Namespace:   "default",
+				Annotations: map[string]string{},
+			},
+		},
+	)
+	ingressAnalyzer := IngressAnalyzer{}
+	var analysisResults []Analysis
+	err := ingressAnalyzer.RunAnalysis(context.Background(),
+		&AnalysisConfiguration{
+			Namespace: "default",
+		},
+		&kubernetes.Client{
+			Client: clientset,
+		}, nil, &analysisResults)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	assert.Equal(t, len(analysisResults), 2)
+}
+
+func TestIngressAnalyzerWithoutIngressClassAnnotation(t *testing.T) {
+
+	clientset := fake.NewSimpleClientset(
+		&networkingv1.Ingress{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:        "example",
+				Namespace:   "default",
+				Annotations: map[string]string{},
+			},
+		})
+	ingressAnalyzer := IngressAnalyzer{}
+
+	var analysisResults []Analysis
+	err := ingressAnalyzer.RunAnalysis(context.Background(),
+		&AnalysisConfiguration{
+			Namespace: "default",
+		},
+		&kubernetes.Client{
+			Client: clientset,
+		}, nil, &analysisResults)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	var errorFound bool
+	for _, analysis := range analysisResults {
+		for _, err := range analysis.Error {
+			if strings.Contains(err, "does not specify an Ingress class") {
+				errorFound = true
+				break
+			}
+		}
+		if errorFound {
+			break
+		}
+	}
+	if !errorFound {
+		t.Error("expected error 'does not specify an Ingress class' not found in analysis results")
+	}
+}

pkg/analyzer/podAnalyzer_test.go

@@ -33,13 +33,15 @@ func TestPodAnalzyer(t *testing.T) {
 
 	podAnalyzer := PodAnalyzer{}
 	var analysisResults []Analysis
-	podAnalyzer.RunAnalysis(context.Background(),
+	err := podAnalyzer.RunAnalysis(context.Background(),
 		&AnalysisConfiguration{
 			Namespace: "default",
 		},
 		&kubernetes.Client{
 			Client: clientset,
 		}, nil, &analysisResults)
-
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
 	assert.Equal(t, len(analysisResults), 1)
 }

pkg/analyzer/serviceAnalyzer_test.go

@@ -34,13 +34,15 @@ func TestServiceAnalzyer(t *testing.T) {
 
 	serviceAnalyzer := ServiceAnalyzer{}
 	var analysisResults []Analysis
-	serviceAnalyzer.RunAnalysis(context.Background(),
+	err := serviceAnalyzer.RunAnalysis(context.Background(),
 		&AnalysisConfiguration{
 			Namespace: "default",
 		},
 		&kubernetes.Client{
 			Client: clientset,
 		}, nil, &analysisResults)
-
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
 	assert.Equal(t, len(analysisResults), 1)
 }