65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package command
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/nomad/api"
|
|
"github.com/hashicorp/nomad/testutil"
|
|
)
|
|
|
|
// seen is used to track which tests we have already
|
|
// marked as parallel. Marking twice causes panic.
|
|
var seen map[*testing.T]struct{}
|
|
|
|
func init() {
|
|
seen = make(map[*testing.T]struct{})
|
|
}
|
|
|
|
func testServer(
|
|
t *testing.T,
|
|
cb testutil.ServerConfigCallback) (*testutil.TestServer, *api.Client, string) {
|
|
|
|
// Always run these tests in parallel.
|
|
if _, ok := seen[t]; !ok {
|
|
seen[t] = struct{}{}
|
|
t.Parallel()
|
|
}
|
|
|
|
// Make a new test server
|
|
srv := testutil.NewTestServer(t, cb)
|
|
|
|
// Make a client
|
|
clientConf := api.DefaultConfig()
|
|
clientConf.Address = "http://" + srv.HTTPAddr
|
|
client, err := api.NewClient(clientConf)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
return srv, client, clientConf.Address
|
|
}
|
|
|
|
func testJob(jobID string) *api.Job {
|
|
task := api.NewTask("task1", "mock_driver").
|
|
SetConfig("kill_after", "1s").
|
|
SetConfig("run_for", "5s").
|
|
SetConfig("exit_code", 0).
|
|
Require(&api.Resources{
|
|
MemoryMB: 256,
|
|
DiskMB: 20,
|
|
CPU: 100,
|
|
}).
|
|
SetLogConfig(&api.LogConfig{
|
|
MaxFiles: 1,
|
|
MaxFileSizeMB: 2,
|
|
})
|
|
|
|
group := api.NewTaskGroup("group1", 1).
|
|
AddTask(task)
|
|
|
|
job := api.NewBatchJob(jobID, jobID, "region1", 1).
|
|
AddDatacenter("dc1").
|
|
AddTaskGroup(group)
|
|
|
|
return job
|
|
}
|