115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package i3
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
// TestRestartSubprocess runs in a process which has been started with
|
|
// DISPLAY= pointing to an Xvfb instance with i3 -c testdata/i3.config running.
|
|
func TestRestartSubprocess(t *testing.T) {
|
|
if os.Getenv("GO_WANT_XVFB") != "1" {
|
|
t.Skip("parent process")
|
|
}
|
|
|
|
// received is buffered so that we can blockingly read on tick.
|
|
received := make(chan *ShutdownEvent, 1)
|
|
tick := make(chan *TickEvent)
|
|
fatal := make(chan bool)
|
|
go func() {
|
|
defer close(tick)
|
|
defer close(received)
|
|
defer close(fatal)
|
|
recv := Subscribe(ShutdownEventType, TickEventType)
|
|
defer recv.Close()
|
|
log.Printf("reading events")
|
|
for recv.Next() {
|
|
log.Printf("received: %#v", recv.Event())
|
|
switch ev := recv.Event().(type) {
|
|
case *ShutdownEvent:
|
|
received <- ev
|
|
case *TickEvent:
|
|
tick <- ev
|
|
}
|
|
}
|
|
log.Printf("done reading events")
|
|
fatal <- true
|
|
}()
|
|
|
|
log.Printf("read initial tick")
|
|
<-tick // Wait until the subscription is ready
|
|
log.Printf("restart")
|
|
if err := Restart(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Restarting i3 triggered a close of the connection, i.e. also a new
|
|
// subscribe and initial tick event:
|
|
log.Printf("read next initial tick")
|
|
ev := <-tick
|
|
if !ev.First {
|
|
t.Fatalf("expected first tick after restart, got %#v instead", ev)
|
|
}
|
|
|
|
if _, err := SendTick(""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
log.Printf("read tick")
|
|
<-tick // Wait until tick was received
|
|
log.Printf("read received")
|
|
<-received // Verify shutdown event was received
|
|
log.Printf("getversion")
|
|
if _, err := GetVersion(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
select {
|
|
case _ = <-fatal:
|
|
t.Fatal("Subscribe has been canceled by restart")
|
|
default:
|
|
|
|
}
|
|
}
|
|
|
|
func TestRestart(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx, canc := context.WithCancel(context.Background())
|
|
defer canc()
|
|
|
|
_, DISPLAY, err := launchXvfb(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dir, err := ioutil.TempDir("", "i3restart")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
I3SOCK := filepath.Join(dir, "i3.sock")
|
|
|
|
cleanup, err := launchI3(ctx, DISPLAY, I3SOCK)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer cleanup()
|
|
|
|
cmd := exec.Command(os.Args[0], "-test.run=TestRestartSubprocess", "-test.v")
|
|
cmd.Env = []string{
|
|
"GO_WANT_XVFB=1",
|
|
"DISPLAY=" + DISPLAY,
|
|
"PATH=" + os.Getenv("PATH"),
|
|
}
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
}
|