38 lines
817 B
Go
38 lines
817 B
Go
|
package raw_exec
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
||
|
|
"github.com/hashicorp/nomad/helper/uuid"
|
||
|
|
"github.com/hashicorp/nomad/plugins/drivers/base"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDriverStartTask(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
require := require.New(t)
|
||
|
|
|
||
|
|
d := NewRawExecDriver(testlog.HCLogger(t))
|
||
|
|
harness := base.NewDriverHarness(t, d)
|
||
|
|
task := &base.TaskConfig{
|
||
|
|
ID: uuid.Generate(),
|
||
|
|
Name: "test",
|
||
|
|
}
|
||
|
|
task.EncodeDriverConfig(&TaskConfig{
|
||
|
|
Command: "go",
|
||
|
|
Args: []string{"version"},
|
||
|
|
})
|
||
|
|
cleanup := harness.MkAllocDir(task)
|
||
|
|
defer cleanup()
|
||
|
|
|
||
|
|
handle, err := harness.StartTask(task)
|
||
|
|
require.NoError(err)
|
||
|
|
|
||
|
|
ch, err := harness.WaitTask(context.Background(), handle.Config.ID)
|
||
|
|
require.NoError(err)
|
||
|
|
result := <-ch
|
||
|
|
require.Zero(result.ExitCode)
|
||
|
|
}
|