Skip to content

Commit 67c8b3f

Browse files
apostasieapostasie
committed
Adapt to new test API
Signed-off-by: apostasie <[email protected]>
1 parent 2e02c9e commit 67c8b3f

13 files changed

+64
-64
lines changed

‎cmd/nerdctl/builder/builder_build_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ func TestBuildFromStdin(t *testing.T) {
234234
dockerfile := fmt.Sprintf(`FROM %s
235235
CMD ["echo", "nerdctl-build-test-stdin"]`, testutil.CommonImage)
236236
cmd := helpers.Command("build", "-t", data.Identifier(), "-f", "-", ".")
237-
cmd.WithStdin(strings.NewReader(dockerfile))
237+
cmd.Feed(strings.NewReader(dockerfile))
238238
return cmd
239239
},
240240
Expected: func(data test.Data, helpers test.Helpers) *test.Expected {

‎cmd/nerdctl/builder/builder_builder_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ CMD ["echo", "nerdctl-builder-debug-test-string"]`, testutil.CommonImage)
8383
err := os.WriteFile(filepath.Join(buildCtx, "Dockerfile"), []byte(dockerfile), 0o600)
8484
assert.NilError(helpers.T(), err)
8585
cmd := helpers.Command("builder", "debug", buildCtx)
86-
cmd.WithStdin(bytes.NewReader([]byte("c\n")))
86+
cmd.Feed(bytes.NewReader([]byte("c\n")))
8787
return cmd
8888
},
8989
Expected: test.Expects(0, nil, nil),

‎cmd/nerdctl/container/container_attach_linux_test.go‎

Lines changed: 20 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,9 @@
1717
package container
1818

1919
import (
20+
"bytes"
2021
"errors"
21-
"os"
22+
"io"
2223
"strings"
2324
"testing"
2425
"time"
@@ -56,11 +57,9 @@ func TestAttach(t *testing.T) {
5657

5758
testCase.Setup = func(data test.Data, helpers test.Helpers) {
5859
cmd := helpers.Command("run", "--rm", "-it", "--name", data.Identifier(), testutil.CommonImage)
59-
cmd.WithPseudoTTY(func(f *os.File) error {
60-
// ctrl+p and ctrl+q (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
61-
_, err := f.Write([]byte{16, 17})
62-
return err
63-
})
60+
cmd.WithPseudoTTY()
61+
// ctrl+p and ctrl+q (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
62+
cmd.Feed(bytes.NewReader([]byte{16, 17}))
6463

6564
cmd.Run(&test.Expected{
6665
ExitCode: 0,
@@ -74,15 +73,13 @@ func TestAttach(t *testing.T) {
7473
testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
7574
// Run interactively and detach
7675
cmd := helpers.Command("attach", data.Identifier())
77-
cmd.WithPseudoTTY(func(f *os.File) error {
78-
_, _ = f.WriteString("echo mark${NON}mark\n")
76+
cmd.Feed(strings.NewReader("echo mark${NON}mark\n"))
77+
cmd.WithFeeder(func() io.Reader {
7978
// Interestingly, and unlike with run, on attach, docker (like nerdctl) ALSO needs a pause so that the
8079
// container can read stdin before we detach
8180
time.Sleep(time.Second)
8281
// ctrl+p and ctrl+q (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
83-
_, err := f.Write([]byte{16, 17})
84-
85-
return err
82+
return bytes.NewReader([]byte{16, 17})
8683
})
8784

8885
return cmd
@@ -120,10 +117,8 @@ func TestAttachDetachKeys(t *testing.T) {
120117

121118
testCase.Setup = func(data test.Data, helpers test.Helpers) {
122119
cmd := helpers.Command("run", "--rm", "-it", "--detach-keys=ctrl-q", "--name", data.Identifier(), testutil.CommonImage)
123-
cmd.WithPseudoTTY(func(f *os.File) error {
124-
_, err := f.Write([]byte{17})
125-
return err
126-
})
120+
cmd.WithPseudoTTY()
121+
cmd.Feed(bytes.NewReader([]byte{17}))
127122

128123
cmd.Run(&test.Expected{
129124
ExitCode: 0,
@@ -137,15 +132,14 @@ func TestAttachDetachKeys(t *testing.T) {
137132
testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
138133
// Run interactively and detach
139134
cmd := helpers.Command("attach", "--detach-keys=ctrl-a,ctrl-b", data.Identifier())
140-
cmd.WithPseudoTTY(func(f *os.File) error {
141-
_, _ = f.WriteString("echo mark${NON}mark\n")
135+
cmd.WithPseudoTTY()
136+
cmd.Feed(strings.NewReader("echo mark${NON}mark\n"))
137+
cmd.WithFeeder(func() io.Reader {
142138
// Interestingly, and unlike with run, on attach, docker (like nerdctl) ALSO needs a pause so that the
143139
// container can read stdin before we detach
144140
time.Sleep(time.Second)
145-
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
146-
_, err := f.Write([]byte{1, 2})
147-
148-
return err
141+
// ctrl+p and ctrl+q (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
142+
return bytes.NewReader([]byte{1, 2})
149143
})
150144

151145
return cmd
@@ -179,11 +173,9 @@ func TestAttachForAutoRemovedContainer(t *testing.T) {
179173

180174
testCase.Setup = func(data test.Data, helpers test.Helpers) {
181175
cmd := helpers.Command("run", "--rm", "-it", "--detach-keys=ctrl-a,ctrl-b", "--name", data.Identifier(), testutil.CommonImage)
182-
cmd.WithPseudoTTY(func(f *os.File) error {
183-
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
184-
_, err := f.Write([]byte{1, 2})
185-
return err
186-
})
176+
cmd.WithPseudoTTY()
177+
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
178+
cmd.Feed(bytes.NewReader([]byte{1, 2}))
187179

188180
cmd.Run(&test.Expected{
189181
ExitCode: 0,
@@ -197,10 +189,8 @@ func TestAttachForAutoRemovedContainer(t *testing.T) {
197189
testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
198190
// Run interactively and detach
199191
cmd := helpers.Command("attach", data.Identifier())
200-
cmd.WithPseudoTTY(func(f *os.File) error {
201-
_, err := f.WriteString("echo mark${NON}mark\nexit 42\n")
202-
return err
203-
})
192+
cmd.WithPseudoTTY()
193+
cmd.Feed(strings.NewReader("echo mark${NON}mark\nexit 42\n"))
204194

205195
return cmd
206196
}

‎cmd/nerdctl/container/container_run_linux_test.go‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -503,8 +503,9 @@ func TestRunWithDetachKeys(t *testing.T) {
503503
testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
504504
// Run interactively and detach
505505
cmd := helpers.Command("run", "-it", "--detach-keys=ctrl-a,ctrl-b", "--name", data.Identifier(), testutil.CommonImage)
506-
cmd.WithPseudoTTY(func(f *os.File) error {
507-
_, _ = f.WriteString("echo mark${NON}mark\n")
506+
cmd.WithPseudoTTY()
507+
cmd.Feed(strings.NewReader("echo mark${NON}mark\n"))
508+
cmd.WithFeeder(func() io.Reader {
508509
// Because of the way we proxy stdin, we have to wait here, otherwise we detach before
509510
// the rest of the input ever reaches the container
510511
// Note that this only concerns nerdctl, as docker seems to behave ok LOCALLY.
@@ -514,8 +515,7 @@ func TestRunWithDetachKeys(t *testing.T) {
514515
nerdtest.EnsureContainerStarted(helpers, data.Identifier())
515516
// }
516517
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
517-
_, err := f.Write([]byte{1, 2})
518-
return err
518+
return bytes.NewReader([]byte{1, 2})
519519
})
520520

521521
return cmd
@@ -571,8 +571,9 @@ func TestIssue3568(t *testing.T) {
571571
testCase.Command = func(data test.Data, helpers test.Helpers) test.TestableCommand {
572572
// Run interactively and detach
573573
cmd := helpers.Command("run", "--rm", "-it", "--detach-keys=ctrl-a,ctrl-b", "--name", data.Identifier(), testutil.CommonImage)
574-
cmd.WithPseudoTTY(func(f *os.File) error {
575-
_, _ = f.WriteString("echo mark${NON}mark\n")
574+
cmd.WithPseudoTTY()
575+
cmd.Feed(strings.NewReader("echo mark${NON}mark\n"))
576+
cmd.WithFeeder(func() io.Reader {
576577
// Because of the way we proxy stdin, we have to wait here, otherwise we detach before
577578
// the rest of the input ever reaches the container
578579
// Note that this only concerns nerdctl, as docker seems to behave ok LOCALLY.
@@ -582,8 +583,7 @@ func TestIssue3568(t *testing.T) {
582583
nerdtest.EnsureContainerStarted(helpers, data.Identifier())
583584
// }
584585
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
585-
_, err := f.Write([]byte{1, 2})
586-
return err
586+
return bytes.NewReader([]byte{1, 2})
587587
})
588588

589589
return cmd

‎cmd/nerdctl/container/container_start_linux_test.go‎

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,9 @@
1717
package container
1818

1919
import (
20+
"bytes"
2021
"errors"
21-
"os"
22+
"io"
2223
"strings"
2324
"testing"
2425

@@ -40,10 +41,8 @@ func TestStartDetachKeys(t *testing.T) {
4041

4142
testCase.Setup = func(data test.Data, helpers test.Helpers) {
4243
cmd := helpers.Command("run", "-it", "--name", data.Identifier(), testutil.CommonImage)
43-
cmd.WithPseudoTTY(func(f *os.File) error {
44-
_, err := f.WriteString("exit\n")
45-
return err
46-
})
44+
cmd.WithPseudoTTY()
45+
cmd.Feed(strings.NewReader("exit\n"))
4746
cmd.Run(&test.Expected{
4847
ExitCode: 0,
4948
})
@@ -60,10 +59,10 @@ func TestStartDetachKeys(t *testing.T) {
6059
flags += "i"
6160
}
6261
cmd := helpers.Command("start", flags, "--detach-keys=ctrl-a,ctrl-b", data.Identifier())
63-
cmd.WithPseudoTTY(func(f *os.File) error {
62+
cmd.WithPseudoTTY()
63+
cmd.WithFeeder(func() io.Reader {
6464
// ctrl+a and ctrl+b (see https://en.wikipedia.org/wiki/C0_and_C1_control_codes)
65-
_, err := f.Write([]byte{1, 2})
66-
return err
65+
return bytes.NewReader([]byte{1, 2})
6766
})
6867

6968
return cmd

‎cmd/nerdctl/image/image_load_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ func TestLoadStdinFromPipe(t *testing.T) {
5353
cmd := helpers.Command("load")
5454
reader, err := os.Open(filepath.Join(data.TempDir(), "common.tar"))
5555
assert.NilError(t, err, "failed to open common.tar")
56-
cmd.WithStdin(reader)
56+
cmd.Feed(reader)
5757
return cmd
5858
},
5959
Expected: func(data test.Data, helpers test.Helpers) *test.Expected {

‎cmd/nerdctl/ipfs/ipfs_compose_linux_test.go‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ package ipfs
1919
import (
2020
"fmt"
2121
"io"
22+
"os"
2223
"strconv"
2324
"strings"
2425
"testing"
@@ -211,8 +212,9 @@ func TestIPFSCompBuild(t *testing.T) {
211212
// Start a local ipfs backed registry
212213
// FIXME: this is bad and likely to collide with other tests
213214
ipfsServer = helpers.Command("ipfs", "registry", "serve", "--listen-registry", listenAddr)
214-
// Once foregrounded, do not wait for it more than a second
215-
ipfsServer.Background(1 * time.Second)
215+
// This should not take longer than that
216+
ipfsServer.WithTimeout(30 * time.Second)
217+
ipfsServer.Background()
216218
// Apparently necessary to let it start...
217219
time.Sleep(time.Second)
218220

@@ -237,9 +239,8 @@ COPY index.html /usr/share/nginx/html/index.html
237239

238240
testCase.Cleanup = func(data test.Data, helpers test.Helpers) {
239241
if ipfsServer != nil {
240-
// Close the server once done
241242
helpers.Anyhow("rmi", "-f", data.Get(mainImageCIDKey))
242-
ipfsServer.Run(nil)
243+
ipfsServer.Signal(os.Kill)
243244
}
244245
if comp != nil {
245246
helpers.Anyhow("compose", "-f", comp.YAMLFullPath(), "down", "-v")

‎cmd/nerdctl/ipfs/ipfs_registry_linux_test.go‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,16 +70,17 @@ func TestIPFSNerdctlRegistry(t *testing.T) {
7070

7171
// Start a local ipfs backed registry
7272
ipfsServer = helpers.Command("ipfs", "registry", "serve", "--listen-registry", listenAddr)
73-
// Once foregrounded, do not wait for it more than a second
74-
ipfsServer.Background(1 * time.Second)
73+
// This should not take longer than that
74+
ipfsServer.WithTimeout(30 * time.Second)
75+
ipfsServer.Background()
7576
// Apparently necessary to let it start...
7677
time.Sleep(time.Second)
7778
}
7879

7980
testCase.Cleanup = func(data test.Data, helpers test.Helpers) {
8081
if ipfsServer != nil {
8182
// Close the server once done
82-
ipfsServer.Run(nil)
83+
ipfsServer.Signal(os.Kill)
8384
}
8485
}
8586

‎cmd/nerdctl/system/system_events_linux_test.go‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@ import (
3030

3131
func testEventFilterExecutor(data test.Data, helpers test.Helpers) test.TestableCommand {
3232
cmd := helpers.Command("events", "--filter", data.Get("filter"), "--format", "json")
33-
cmd.Background(1 * time.Second)
33+
cmd.WithTimeout(3 * time.Second)
34+
cmd.Background()
3435
helpers.Ensure("run", "--rm", testutil.CommonImage)
3536
return cmd
3637
}

‎go.mod‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@ require (
117117
github.com/petermattis/goid v0.0.0-20240813172612-4fcff4a6cae7 // indirect
118118
github.com/philhofer/fwd v1.1.3-0.20240612014219-fbbf4953d986 // indirect
119119
github.com/pkg/errors v0.9.1 // indirect
120+
github.com/rs/zerolog v1.34.0 // indirect
120121
github.com/sasha-s/go-deadlock v0.3.5 // indirect
121122
github.com/sirupsen/logrus v1.9.3 // indirect
122123
github.com/smallstep/pkcs7 v0.1.1 // indirect

0 commit comments

Comments
 (0)