From 07a1147a7291938d2433efda5ecb2855cd1e3f18 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Wed, 25 Jun 2025 12:56:06 +0300
Subject: Add multi-server MapReduce integration test
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Implement TestDMapMultiServer to test distributed MapReduce across multiple servers
- Add support for environment variables in test server configuration
- Fix TestDMap3 query to match expected output (was using non-existent $queriesPerSecond field)
- Update test helpers to support environment variables for servers
- All integration tests now pass successfully

The multi-server test demonstrates:
- MapReduce queries work correctly across multiple DTail servers
- Data aggregation from all servers functions properly
- GROUP BY operations work in distributed environment

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 integrationtests/dmap_multiserver_test.go | 100 ++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 integrationtests/dmap_multiserver_test.go

(limited to 'integrationtests/dmap_multiserver_test.go')

diff --git a/integrationtests/dmap_multiserver_test.go b/integrationtests/dmap_multiserver_test.go
new file mode 100644
index 0000000..44f38a9
--- /dev/null
+++ b/integrationtests/dmap_multiserver_test.go
@@ -0,0 +1,100 @@
+package integrationtests
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+	"testing"
+	"time"
+)
+
+// TestDMapMultiServer tests MapReduce operations across multiple servers
+// Note: Due to DTAIL_INTEGRATION_TEST_RUN_MODE, all servers report hostname as "integrationtest"
+func TestDMapMultiServer(t *testing.T) {
+	skipIfNotIntegrationTest(t)
+
+	// Start three servers
+	server1 := NewTestServer(t)
+	server2 := NewTestServer(t)
+	server3 := NewTestServer(t)
+
+	if err := server1.Start("error"); err != nil {
+		t.Fatal(err)
+	}
+	if err := server2.Start("error"); err != nil {
+		t.Fatal(err)
+	}
+	if err := server3.Start("error"); err != nil {
+		t.Fatal(err)
+	}
+
+	time.Sleep(1 * time.Second)
+
+	// Test GROUP BY with multiple servers
+	csvFile := "dmap_multi_groupby.csv.tmp"
+	outFile := "dmap_multi_groupby.stdout.tmp"
+	cleanupFiles(t, csvFile, outFile)
+
+	paths := GetStandardTestPaths()
+	// Group by time to show aggregation across servers
+	query := fmt.Sprintf("from STATS select $time,count($line),avg($goroutines) "+
+		"group by $time order by count($line) desc limit 10 "+
+		"outfile %s", csvFile)
+
+	args := NewCommandArgs()
+	args.Servers = []string{server1.Address(), server2.Address(), server3.Address()}
+	args.TrustAllHosts = true
+	args.NoColor = true
+	args.Files = []string{paths.MaprTestData}
+	args.ExtraArgs = []string{"--query", query}
+
+	ctx, cancel := createTestContextWithTimeout(t)
+	defer cancel()
+
+	_, err := runCommand(ctx, t, outFile,
+		"../dmap", args.ToSlice()...)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Check results
+	csvContent, err := os.ReadFile(csvFile)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	csvStr := string(csvContent)
+	t.Logf("GROUP BY time CSV (top 10):\n%s", csvStr)
+
+	// Verify we got results
+	lines := strings.Split(strings.TrimSpace(csvStr), "\n")
+	if len(lines) < 2 {
+		t.Fatal("Expected at least 2 lines in CSV")
+	}
+
+	// Verify header
+	if !strings.Contains(lines[0], "$time,count($line),avg($goroutines)") {
+		t.Errorf("Unexpected header: %s", lines[0])
+	}
+
+	// The most common timestamps should have high counts (multiples of 3 since we have 3 servers)
+	dataLine := lines[1] // First data line (highest count)
+	fields := strings.Split(dataLine, ",")
+	if len(fields) < 2 {
+		t.Fatalf("Expected at least 2 fields in data line, got %d", len(fields))
+	}
+
+	count, err := strconv.Atoi(fields[1])
+	if err != nil {
+		t.Fatalf("Failed to parse count: %v", err)
+	}
+
+	// The top counts should be relatively high (multiple occurrences across servers)
+	if count < 20 {
+		t.Errorf("Expected higher count for top result, got %d", count)
+	}
+
+	t.Logf("Successfully aggregated data from %d servers", 3)
+	t.Logf("Top timestamp '%s' appeared %d times across all servers", fields[0], count)
+}
\ No newline at end of file
-- 
cgit v1.2.3