rest: Add context

This commit is contained in:
Alexander Neumann 2017-05-11 22:40:50 +02:00
parent f2b83ece10
commit db4fa48f66

View file

@ -2,6 +2,7 @@ package rest_test
import ( import (
"bufio" "bufio"
"context"
"io/ioutil" "io/ioutil"
"net/url" "net/url"
"os" "os"
@ -15,13 +16,13 @@ import (
. "restic/test" . "restic/test"
) )
func runRESTServer(t testing.TB, dir string) func() { func runRESTServer(ctx context.Context, t testing.TB, dir string) func() {
srv, err := exec.LookPath("rest-server") srv, err := exec.LookPath("rest-server")
if err != nil { if err != nil {
t.Skip(err) t.Skip(err)
} }
cmd := exec.Command(srv, "--path", dir) cmd := exec.CommandContext(ctx, srv, "--path", dir)
cmd.Stdout = os.Stdout cmd.Stdout = os.Stdout
stderr, err := cmd.StderrPipe() stderr, err := cmd.StderrPipe()
if err != nil { if err != nil {
@ -50,10 +51,13 @@ func runRESTServer(t testing.TB, dir string) func() {
} }
func TestBackend(t *testing.T) { func TestBackend(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
dir, cleanup := TempDir(t) dir, cleanup := TempDir(t)
defer cleanup() defer cleanup()
cleanup = runRESTServer(t, dir) cleanup = runRESTServer(ctx, t, dir)
defer cleanup() defer cleanup()
suite := test.Suite{ suite := test.Suite{