71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
// +build !windows
|
|
|
|
package shim
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"syscall"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestShimUSR1SignalingWorks(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip()
|
|
return
|
|
}
|
|
stdinReader, stdinWriter := io.Pipe()
|
|
stdoutReader, stdoutWriter := io.Pipe()
|
|
|
|
stdin = stdinReader
|
|
stdout = stdoutWriter
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
metricProcessed, exited := runInputPlugin(t, 20*time.Minute)
|
|
|
|
// signal USR1 to yourself.
|
|
pid := os.Getpid()
|
|
process, err := os.FindProcess(pid)
|
|
require.NoError(t, err)
|
|
|
|
go func() {
|
|
// On slow machines this signal can fire before the service comes up.
|
|
// rather than depend on accurate sleep times, we'll just retry sending
|
|
// the signal every so often until it goes through.
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return // test is done
|
|
default:
|
|
// test isn't done, keep going.
|
|
process.Signal(syscall.SIGUSR1)
|
|
time.Sleep(200 * time.Millisecond)
|
|
}
|
|
}
|
|
}()
|
|
|
|
timeout := time.NewTimer(10 * time.Second)
|
|
|
|
select {
|
|
case <-metricProcessed:
|
|
case <-timeout.C:
|
|
require.Fail(t, "Timeout waiting for metric to arrive")
|
|
}
|
|
cancel()
|
|
|
|
r := bufio.NewReader(stdoutReader)
|
|
out, err := r.ReadString('\n')
|
|
require.NoError(t, err)
|
|
require.Equal(t, "measurement,tag=tag field=1i 1234000005678\n", out)
|
|
|
|
stdinWriter.Close()
|
|
<-exited
|
|
}
|