rest: Add context
This commit is contained in:
parent
f2b83ece10
commit
db4fa48f66
1 changed files with 7 additions and 3 deletions
|
@ -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{
|
||||||
|
|
Loading…
Add table
Reference in a new issue