pkg: replace the deprecated Expect with ExpectWithContext in pkg/expect/expect_test.go

Signed-off-by: demoManito <1430482733@qq.com>
This commit is contained in:
demoManito 2022-09-16 13:54:48 +08:00
parent b7ba0542f6
commit a281bb0494

View File

@ -75,7 +75,8 @@ func TestEcho(t *testing.T) {
if err != nil {
t.Fatal(err)
}
l, eerr := ep.Expect("world")
ctx := context.Background()
l, eerr := ep.ExpectWithContext(ctx, "world")
if eerr != nil {
t.Fatal(eerr)
}
@ -86,7 +87,7 @@ func TestEcho(t *testing.T) {
if cerr := ep.Close(); cerr != nil {
t.Fatal(cerr)
}
if _, eerr = ep.Expect("..."); eerr == nil {
if _, eerr = ep.ExpectWithContext(ctx, "..."); eerr == nil {
t.Fatalf("expected error on closed expect process")
}
}
@ -97,7 +98,7 @@ func TestLineCount(t *testing.T) {
t.Fatal(err)
}
wstr := "3"
l, eerr := ep.Expect(wstr)
l, eerr := ep.ExpectWithContext(context.Background(), wstr)
if eerr != nil {
t.Fatal(eerr)
}
@ -120,7 +121,7 @@ func TestSend(t *testing.T) {
if err := ep.Send("a\r"); err != nil {
t.Fatal(err)
}
if _, err := ep.Expect("b"); err != nil {
if _, err := ep.ExpectWithContext(context.Background(), "b"); err != nil {
t.Fatal(err)
}
if err := ep.Stop(); err != nil {