84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package apitests
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/api"
|
|
)
|
|
|
|
// boolToPtr returns the pointer to a boolean
|
|
func boolToPtr(b bool) *bool {
|
|
return &b
|
|
}
|
|
|
|
// intToPtr returns the pointer to an int
|
|
func intToPtr(i int) *int {
|
|
return &i
|
|
}
|
|
|
|
// timeToPtr returns the pointer to a time stamp
|
|
func timeToPtr(t time.Duration) *time.Duration {
|
|
return &t
|
|
}
|
|
|
|
// stringToPtr returns the pointer to a string
|
|
func stringToPtr(str string) *string {
|
|
return &str
|
|
}
|
|
|
|
func assertQueryMeta(t *testing.T, qm *api.QueryMeta) {
|
|
t.Helper()
|
|
if qm.LastIndex == 0 {
|
|
t.Fatalf("bad index: %d", qm.LastIndex)
|
|
}
|
|
if !qm.KnownLeader {
|
|
t.Fatalf("expected known leader, got none")
|
|
}
|
|
}
|
|
|
|
func assertWriteMeta(t *testing.T, wm *api.WriteMeta) {
|
|
t.Helper()
|
|
if wm.LastIndex == 0 {
|
|
t.Fatalf("bad index: %d", wm.LastIndex)
|
|
}
|
|
}
|
|
|
|
func testJob() *api.Job {
|
|
task := api.NewTask("task1", "exec").
|
|
SetConfig("command", "/bin/sleep").
|
|
Require(&api.Resources{
|
|
CPU: intToPtr(100),
|
|
MemoryMB: intToPtr(256),
|
|
}).
|
|
SetLogConfig(&api.LogConfig{
|
|
MaxFiles: intToPtr(1),
|
|
MaxFileSizeMB: intToPtr(2),
|
|
})
|
|
|
|
group := api.NewTaskGroup("group1", 1).
|
|
AddTask(task).
|
|
RequireDisk(&api.EphemeralDisk{
|
|
SizeMB: intToPtr(25),
|
|
})
|
|
|
|
job := api.NewBatchJob("job1", "redis", "region1", 1).
|
|
AddDatacenter("dc1").
|
|
AddTaskGroup(group)
|
|
|
|
return job
|
|
}
|
|
|
|
// conversions utils only used for testing
|
|
// added here to avoid linter warning
|
|
|
|
// int64ToPtr returns the pointer to an int
|
|
func int64ToPtr(i int64) *int64 {
|
|
return &i
|
|
}
|
|
|
|
// float64ToPtr returns the pointer to an float64
|
|
func float64ToPtr(f float64) *float64 {
|
|
return &f
|
|
}
|