110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
// Copyright (c) 2026 Storm Dragon
|
|
// Authors: Storm Dragon, Angus Kola
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package atspi
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestNativePidginTranscriptCapture(t *testing.T) {
|
|
if os.Getenv("CTHULHU_NATIVE_MESSAGING") != "1" {
|
|
t.Skip("requires private Xorg/AT-SPI session")
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 25*time.Second)
|
|
defer cancel()
|
|
adapter := New()
|
|
if err := adapter.Start(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer adapter.Stop(context.Background())
|
|
for _, processName := range []string{"pidgin", "messaging-test"} {
|
|
t.Run(processName, func(t *testing.T) {
|
|
command := exec.CommandContext(ctx, "/usr/bin/python3", "../../tests/gtk_messaging.py", processName)
|
|
input, err := command.StdinPipe()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
output, err := command.StdoutPipe()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
command.Stderr = os.Stderr
|
|
if err := command.Start(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer func() { input.Close(); command.Process.Kill(); command.Wait() }()
|
|
line, err := bufio.NewReader(output).ReadString('\n')
|
|
fields := strings.Fields(line)
|
|
if err != nil || len(fields) != 2 || fields[0] != "ready" {
|
|
t.Fatalf("fixture readiness: %q, %v", line, err)
|
|
}
|
|
if _, err := adapter.CaptureDesktop(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if output, err := exec.CommandContext(ctx, "xdotool", "windowfocus", "--sync", fields[1]).CombinedOutput(); err != nil {
|
|
t.Fatalf("focus: %s, %v", output, err)
|
|
}
|
|
fmt.Fprintln(input, "message\ntyping\ntyping-weight\nedit\nbarrier")
|
|
messages, typing := 0, 0
|
|
var transcript ObjectID
|
|
retired, barrier := false, false
|
|
for {
|
|
select {
|
|
case event := <-adapter.Events():
|
|
if event.Source.ApplicationProcess != processName {
|
|
continue
|
|
}
|
|
if event.ChatMessage != nil {
|
|
if processName != "pidgin" {
|
|
t.Fatal("unrelated process received chat metadata")
|
|
}
|
|
message := event.ChatMessage
|
|
transcript = event.Source.ID()
|
|
// GTK can publish an empty insertion during initial enumeration.
|
|
if message.Text == "" {
|
|
continue
|
|
}
|
|
if message.Provider != event.Source.BusName || message.Conversation != event.Source.Path || message.Window == "" || message.RoomName == "" {
|
|
t.Fatalf("incomplete identities: %+v", message)
|
|
}
|
|
if message.Typing {
|
|
typing++
|
|
} else {
|
|
messages++
|
|
if message.Text != "Username: Hello\n" {
|
|
t.Fatalf("message content = %q", message.Text)
|
|
}
|
|
}
|
|
}
|
|
if event.Type == DefunctStateEvent && event.Detail1 != 0 && transcript.Valid() && event.Source.ID() == transcript {
|
|
retired = true
|
|
}
|
|
if event.Type == TextInsertedEvent && event.Payload == "Messaging barrier" {
|
|
barrier = true
|
|
if processName == "pidgin" && (messages != 1 || typing != 2) {
|
|
t.Fatalf("captured %d messages, %d typing statuses", messages, typing)
|
|
}
|
|
if processName == "pidgin" {
|
|
fmt.Fprintln(input, "destroy")
|
|
}
|
|
}
|
|
if barrier && (processName != "pidgin" || retired) {
|
|
return
|
|
}
|
|
case <-ctx.Done():
|
|
t.Fatalf("transcript/barrier missing: %v (messages=%d, typing=%d)", ctx.Err(), messages, typing)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|