mirror of https://github.com/docker/cli.git
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package opts
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types/container"
|
|
"gotest.tools/v3/assert"
|
|
is "gotest.tools/v3/assert/cmp"
|
|
)
|
|
|
|
func TestGpusOptAll(t *testing.T) {
|
|
for _, testcase := range []string{
|
|
"all",
|
|
"-1",
|
|
"count=all",
|
|
"count=-1",
|
|
} {
|
|
var gpus GpuOpts
|
|
assert.Check(t, gpus.Set(testcase))
|
|
gpuReqs := gpus.Value()
|
|
assert.Assert(t, is.Len(gpuReqs, 1))
|
|
assert.Check(t, is.DeepEqual(gpuReqs[0], container.DeviceRequest{
|
|
Count: -1,
|
|
Capabilities: [][]string{{"gpu"}},
|
|
Options: map[string]string{},
|
|
}))
|
|
}
|
|
}
|
|
|
|
func TestGpusOptInvalidCount(t *testing.T) {
|
|
var gpus GpuOpts
|
|
err := gpus.Set(`count=invalid-integer`)
|
|
assert.Error(t, err, `invalid count (invalid-integer): value must be either "all" or an integer: invalid syntax`)
|
|
}
|
|
|
|
func TestGpusOpts(t *testing.T) {
|
|
for _, testcase := range []string{
|
|
`driver=nvidia,"capabilities=compute,utility","options=foo=bar,baz=qux"`,
|
|
`1,driver=nvidia,"capabilities=compute,utility","options=foo=bar,baz=qux"`,
|
|
`count=1,driver=nvidia,"capabilities=compute,utility","options=foo=bar,baz=qux"`,
|
|
`driver=nvidia,"capabilities=compute,utility","options=foo=bar,baz=qux",count=1`,
|
|
} {
|
|
var gpus GpuOpts
|
|
assert.Check(t, gpus.Set(testcase))
|
|
gpuReqs := gpus.Value()
|
|
assert.Assert(t, is.Len(gpuReqs, 1))
|
|
assert.Check(t, is.DeepEqual(gpuReqs[0], container.DeviceRequest{
|
|
Driver: "nvidia",
|
|
Count: 1,
|
|
Capabilities: [][]string{{"compute", "utility", "gpu"}},
|
|
Options: map[string]string{"foo": "bar", "baz": "qux"},
|
|
}))
|
|
}
|
|
}
|