fix nil access

Author: qmuntal

go/internal/e2e/mcp_and_agents_test.go

@@ -96,6 +96,10 @@ func TestMCPServers(t *testing.T) {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
+		if message == nil {
+			t.Fatalf("Expected a message, got nil")
+		}
+
 		if message.Data.Content == nil || !strings.Contains(*message.Data.Content, "6") {
 			t.Errorf("Expected message to contain '6', got: %v", message.Data.Content)
 		}
@@ -226,6 +230,10 @@ func TestCustomAgents(t *testing.T) {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
+		if message == nil {
+			t.Fatalf("Expected a message, got nil")
+		}
+
 		if message.Data.Content == nil || !strings.Contains(*message.Data.Content, "12") {
 			t.Errorf("Expected message to contain '12', got: %v", message.Data.Content)
 		}

go/internal/e2e/permissions_test.go

@@ -165,6 +165,10 @@ func TestPermissions(t *testing.T) {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
+		if message == nil {
+			t.Fatal("Expected a message, got nil")
+		}
+
 		if message.Data.Content == nil || !strings.Contains(*message.Data.Content, "4") {
 			t.Errorf("Expected message to contain '4', got: %v", message.Data.Content)
 		}

go/internal/e2e/skills_test.go

@@ -133,6 +133,10 @@ func TestSkills(t *testing.T) {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
+		if message1 == nil {
+			t.Fatalf("Expected a message, got nil")
+		}
+
 		if message1.Data.Content != nil && strings.Contains(*message1.Data.Content, skillMarker) {
 			t.Errorf("Expected message to NOT contain skill marker before skill was added, got: %v", *message1.Data.Content)
 		}
@@ -155,6 +159,10 @@ func TestSkills(t *testing.T) {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
+		if message2 == nil {
+			t.Fatalf("Expected a message, got nil")
+		}
+
 		if message2.Data.Content == nil || !strings.Contains(*message2.Data.Content, skillMarker) {
 			t.Errorf("Expected message to contain skill marker '%s' after resume, got: %v", skillMarker, message2.Data.Content)
 		}