summaryrefslogtreecommitdiff
path: root/internal/mapr/groupset_ordering_test.go
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2026-07-22 23:51:18 +0300
committerPaul Buetow <paul@buetow.org>2026-07-22 23:51:18 +0300
commit849951be1d1a7ee9f9302006ccb187bf5b4e36f3 (patch)
tree496c924a03a9ea6212e29bb4699e268066ebad81 /internal/mapr/groupset_ordering_test.go
parentbf78b3abffee6d49c08ca2980156afc455994969 (diff)
feat: DTail fork — server/client feature development
Squashed development of the snonux/dtail fork's product code (internal/, cmd/) since diverging from mimecast/dtail. Major areas: - Read/output path: the former "turbo" channel-less path is now the single, default server-side read/output path for cat/grep/tail and MapReduce; the old channel-based path and its config/env toggles were removed. - MapReduce: single aggregate implementation (server + serverless) fed directly by a processor pipeline, with input-exhausted finalization via the shutdown coordinator; high-concurrency and data-race fixes. - Journal source reads (journal:unit.service) via journalctl, Linux-gated behind a journal-v1 capability. - Auth-key fast reconnect: in-memory per-user public-key cache with TTL/max-keys, registered over an authenticated session (AUTHKEY), checked before authorized_keys. - Interactive query reload (--interactive-query) with SESSION START/UPDATE generation boundaries and capability negotiation. - Client-side deadlines: --timeout / --shutdownAfter as context deadlines; follow shutdown handling. - Client logging: diagnostics-only daily log by default, opt-in payload tee via --log-payload. - Numerous correctness fixes (buffer-pool double-recycle races, EOF-sentinel leaks, glob-expansion cap, TOCTOU in CSV parsing) with accompanying unit tests. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Diffstat (limited to 'internal/mapr/groupset_ordering_test.go')
-rw-r--r--internal/mapr/groupset_ordering_test.go125
1 files changed, 125 insertions, 0 deletions
diff --git a/internal/mapr/groupset_ordering_test.go b/internal/mapr/groupset_ordering_test.go
new file mode 100644
index 0000000..18b845f
--- /dev/null
+++ b/internal/mapr/groupset_ordering_test.go
@@ -0,0 +1,125 @@
+package mapr
+
+import (
+ "reflect"
+ "testing"
+)
+
+// TestGroupSetResultOrderIsDeterministicWithoutOrderBy is a negative test that
+// reproduces the non-determinism bug in result(): when OrderBy is unset, the
+// output row order depended on Go's map iteration order, which is intentionally
+// randomised per runtime invocation. Two consecutive calls to result() on the
+// same GroupSet could return rows in different orders.
+//
+// The fix collects group keys, sorts them lexicographically before building
+// rows, and only then applies SortStable for the OrderBy pass. Ties on OrderBy
+// (or no OrderBy) therefore resolve to lexicographic groupKey order rather than
+// to random map iteration order.
+func TestGroupSetResultOrderIsDeterministicWithoutOrderBy(t *testing.T) {
+ t.Parallel()
+
+ // Query with no ORDER BY clause — the bug case where map iteration order
+ // was the sole determinant of row order.
+ query, err := NewQuery("select count(line) from logs group by host")
+ if err != nil {
+ t.Fatalf("Unable to parse query: %v", err)
+ }
+
+ groupSet := NewGroupSet()
+
+ // Insert keys in reverse lexicographic order to ensure the expected sorted
+ // order cannot coincide with insertion order.
+ for _, host := range []string{"host-z", "host-m", "host-a", "host-b"} {
+ set := groupSet.GetSet(host)
+ if err := set.Aggregate("count(line)", Count, "1", false); err != nil {
+ t.Fatalf("Aggregate failed for %s: %v", host, err)
+ }
+ }
+
+ // Run result() many times. Before the fix a handful of iterations was
+ // enough to observe a different ordering; with the fix every call must
+ // return exactly the same lexicographically sorted sequence of groupKeys.
+ var firstKeys []string
+ const iterations = 50
+ for i := range iterations {
+ rows, _, err := groupSet.result(query, false)
+ if err != nil {
+ t.Fatalf("result() iteration %d returned error: %v", i, err)
+ }
+ if len(rows) != 4 {
+ t.Fatalf("Expected 4 rows, got %d on iteration %d", len(rows), i)
+ }
+
+ keys := make([]string, len(rows))
+ for j, r := range rows {
+ keys[j] = r.groupKey
+ }
+
+ if i == 0 {
+ firstKeys = keys
+ // Verify the order is lexicographic (the contract of the fix).
+ expected := []string{"host-a", "host-b", "host-m", "host-z"}
+ if !reflect.DeepEqual(keys, expected) {
+ t.Fatalf("First result not in lexicographic order: got %v, want %v", keys, expected)
+ }
+ continue
+ }
+
+ // Every subsequent call must return the identical key sequence.
+ if !reflect.DeepEqual(keys, firstKeys) {
+ t.Fatalf("Non-deterministic ordering detected on iteration %d: got %v, want %v", i, keys, firstKeys)
+ }
+ }
+}
+
+// TestGroupSetResultOrderIsDeterministicWithOrderByTies verifies that when
+// multiple rows share the same OrderBy value (a tie), the tie-break falls back
+// to lexicographic groupKey order rather than to random map iteration order.
+// SortStable preserves the relative order of equal elements, so the pre-sort of
+// keys guarantees a deterministic tie-break.
+func TestGroupSetResultOrderIsDeterministicWithOrderByTies(t *testing.T) {
+ t.Parallel()
+
+ // ORDER BY count(line) — all rows will have the same count (1), creating a
+ // full tie that must resolve to lexicographic groupKey order.
+ query, err := NewQuery("select count(line) from logs group by host order by count(line)")
+ if err != nil {
+ t.Fatalf("Unable to parse query: %v", err)
+ }
+
+ groupSet := NewGroupSet()
+
+ // All hosts receive the same count value to force a tie.
+ for _, host := range []string{"host-z", "host-m", "host-a", "host-b"} {
+ set := groupSet.GetSet(host)
+ if err := set.Aggregate("count(line)", Count, "1", false); err != nil {
+ t.Fatalf("Aggregate failed for %s: %v", host, err)
+ }
+ }
+
+ var firstKeys []string
+ const iterations = 50
+ for i := range iterations {
+ rows, _, err := groupSet.result(query, false)
+ if err != nil {
+ t.Fatalf("result() iteration %d returned error: %v", i, err)
+ }
+ if len(rows) != 4 {
+ t.Fatalf("Expected 4 rows, got %d on iteration %d", len(rows), i)
+ }
+
+ keys := make([]string, len(rows))
+ for j, r := range rows {
+ keys[j] = r.groupKey
+ }
+
+ if i == 0 {
+ firstKeys = keys
+ continue
+ }
+
+ if !reflect.DeepEqual(keys, firstKeys) {
+ t.Fatalf("Non-deterministic tie-break detected on iteration %d: got %v, want %v", i, keys, firstKeys)
+ }
+ }
+}