65 lines
3.0 KiB
Go
65 lines
3.0 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
rvboxv1 "github.com/rvbox/rvbox/gen/go/rvbox/v1"
|
|
"github.com/rvbox/rvbox/internal/domain"
|
|
)
|
|
|
|
func TestReadCommandEventsValidatesSegmentAndPreservesOrder_HP_EVENT_04(t *testing.T) {
|
|
ctx := context.Background()
|
|
opened, err := Open(ctx, Options{DataDir: filepath.Join(t.TempDir(), "state"), BusyTimeout: time.Second})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer opened.Close()
|
|
if _, err := opened.RegisterClientSession(ctx, ClientRegistration{ClientID: "win-client", Platform: 3, Architecture: "amd64", DaemonVersion: "test", DaemonCWD: `C:\`, SupportedShells: []byte{1}, ClientInstanceID: [16]byte{1}, SessionID: [16]byte{2}, ConnectedAt: time.Now()}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
issue, _ := domain.ParseUUIDv7("019c46f1-1d02-7000-8000-0000000000b1")
|
|
now := time.Now().UTC()
|
|
if _, err := opened.QueueCommand(ctx, QueueCommandInput{IssueUUID: issue, ClientID: "win-client", IssueTime: now, ReceiptTime: now, ImmutableSHA256: sha256.Sum256([]byte("request")), ExecutionSpec: []byte("spec")}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := opened.ClaimNextDispatch(ctx, "win-client", 1, now); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := opened.RecordCommandAcceptance(ctx, issue, "win-client", 1, 1, true, now); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for sequence, lifecycle := range []rvboxv1.CommandLifecycle{rvboxv1.CommandLifecycle_COMMAND_RUNNING, rvboxv1.CommandLifecycle_COMMAND_SUCCEEDED} {
|
|
event := []byte{byte(sequence + 1), 'e', 'v', 't'}
|
|
var identity *rvboxv1.WindowsExecutionIdentity
|
|
if sequence == 0 {
|
|
effective := rvboxv1.WindowsExecutionContext_WINDOWS_EXECUTION_CONTEXT_ACTIVE_USER
|
|
identity = &rvboxv1.WindowsExecutionIdentity{EffectiveContext: &effective, AttemptedContexts: []rvboxv1.WindowsExecutionContext{effective}, SelectionDetail: "selected"}
|
|
}
|
|
if _, err := opened.AppendCommandEvent(ctx, EventAppend{IssueUUID: [16]byte(issue), ClientID: "win-client", SessionGeneration: 1, EventSeq: uint64(sequence + 1), ObservedUnixNano: now.UnixNano(), ReceiptUnixNano: now.Add(time.Duration(sequence) * time.Millisecond).UnixNano(), EventType: 4, Compression: 1, RawLength: uint64(len(event)), Payload: event, ImmutableSHA256: sha256.Sum256(event), Lifecycle: &lifecycle, LifecycleRevision: 1, WindowsIdentity: identity}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
view, err := opened.GetCommandView(ctx, "win-client", issue)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(view.WindowsIdentity) == 0 {
|
|
t.Fatal("running lifecycle did not persist Windows identity")
|
|
}
|
|
events, err := opened.ReadCommandEvents(ctx, issue, 0, 10)
|
|
if err != nil || len(events) != 2 || events[0].EventSeq != 1 || events[1].EventSeq != 2 || string(events[1].Payload) != "\x02evt" {
|
|
t.Fatalf("read events = %#v, %v", events, err)
|
|
}
|
|
decoded, err := DecodeEventPayload(events[0])
|
|
if err != nil || string(decoded) != "\x01evt" {
|
|
t.Fatalf("decoded event payload = %q, %v", decoded, err)
|
|
}
|
|
if _, err := opened.ReadCommandEvents(ctx, issue, 0, 0); err == nil {
|
|
t.Fatal("zero event limit accepted")
|
|
}
|
|
}
|