ba728f8f97
* api: enable support for setting original source alongside job This PR adds support for setting job source material along with the registration of a job. This includes a new HTTP endpoint and a new RPC endpoint for making queries for the original source of a job. The HTTP endpoint is /v1/job/<id>/submission?version=<version> and the RPC method is Job.GetJobSubmission. The job source (if submitted, and doing so is always optional), is stored in the job_submission memdb table, separately from the actual job. This way we do not incur overhead of reading the large string field throughout normal job operations. The server config now includes job_max_source_size for configuring the maximum size the job source may be, before the server simply drops the source material. This should help prevent Bad Things from happening when huge jobs are submitted. If the value is set to 0, all job source material will be dropped. * api: avoid writing var content to disk for parsing * api: move submission validation into RPC layer * api: return an error if updating a job submission without namespace or job id * api: be exact about the job index we associate a submission with (modify) * api: reword api docs scheduling * api: prune all but the last 6 job submissions * api: protect against nil job submission in job validation * api: set max job source size in test server * api: fixups from pr
142 lines
2.9 KiB
Go
142 lines
2.9 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package api
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/shoenig/test/must"
|
|
)
|
|
|
|
func assertQueryMeta(t *testing.T, qm *QueryMeta) {
|
|
t.Helper()
|
|
|
|
must.NotEq(t, 0, qm.LastIndex, must.Sprint("bad index"))
|
|
must.True(t, qm.KnownLeader, must.Sprint("expected a known leader but gone none"))
|
|
}
|
|
|
|
func assertWriteMeta(t *testing.T, wm *WriteMeta) {
|
|
t.Helper()
|
|
must.Positive(t, wm.LastIndex, must.Sprint("expected WriteMeta.LastIndex to be > 0"))
|
|
}
|
|
|
|
func testJob() *Job {
|
|
task := NewTask("task1", "raw_exec").
|
|
SetConfig("command", "/bin/sleep").
|
|
Require(&Resources{
|
|
CPU: pointerOf(100),
|
|
MemoryMB: pointerOf(256),
|
|
}).
|
|
SetLogConfig(&LogConfig{
|
|
MaxFiles: pointerOf(1),
|
|
MaxFileSizeMB: pointerOf(2),
|
|
})
|
|
|
|
group := NewTaskGroup("group1", 1).
|
|
AddTask(task).
|
|
RequireDisk(&EphemeralDisk{
|
|
SizeMB: pointerOf(25),
|
|
})
|
|
|
|
job := NewBatchJob("job1", "redis", "global", 1).
|
|
AddDatacenter("dc1").
|
|
AddTaskGroup(group)
|
|
|
|
return job
|
|
}
|
|
|
|
func testJobWithScalingPolicy() *Job {
|
|
job := testJob()
|
|
job.TaskGroups[0].Scaling = &ScalingPolicy{
|
|
Policy: map[string]interface{}{},
|
|
Min: pointerOf(int64(1)),
|
|
Max: pointerOf(int64(5)),
|
|
Enabled: pointerOf(true),
|
|
}
|
|
return job
|
|
}
|
|
|
|
func testPeriodicJob() *Job {
|
|
job := testJob().AddPeriodicConfig(&PeriodicConfig{
|
|
Enabled: pointerOf(true),
|
|
Spec: pointerOf("*/30 * * * *"),
|
|
SpecType: pointerOf("cron"),
|
|
})
|
|
return job
|
|
}
|
|
|
|
func testRecommendation(job *Job) *Recommendation {
|
|
rec := &Recommendation{
|
|
ID: "",
|
|
Region: *job.Region,
|
|
Namespace: *job.Namespace,
|
|
JobID: *job.ID,
|
|
Group: *job.TaskGroups[0].Name,
|
|
Task: job.TaskGroups[0].Tasks[0].Name,
|
|
Resource: "CPU",
|
|
Value: *job.TaskGroups[0].Tasks[0].Resources.CPU * 2,
|
|
Meta: map[string]interface{}{
|
|
"testing": true,
|
|
"mocked": "also true",
|
|
},
|
|
Stats: map[string]float64{
|
|
"median": 50.0,
|
|
"mean": 51.0,
|
|
"max": 75.5,
|
|
"99": 73.0,
|
|
"min": 0.0,
|
|
},
|
|
EnforceVersion: false,
|
|
}
|
|
return rec
|
|
}
|
|
|
|
func testNamespace() *Namespace {
|
|
return &Namespace{
|
|
Name: "test-namespace",
|
|
Description: "Testing namespaces",
|
|
}
|
|
}
|
|
|
|
func testQuotaSpec() *QuotaSpec {
|
|
return &QuotaSpec{
|
|
Name: "test-quota",
|
|
Description: "Testing namespaces",
|
|
Limits: []*QuotaLimit{
|
|
{
|
|
Region: "global",
|
|
RegionLimit: &Resources{
|
|
CPU: pointerOf(2000),
|
|
MemoryMB: pointerOf(2000),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// conversions utils only used for testing
|
|
// added here to avoid linter warning
|
|
|
|
// float64ToPtr returns the pointer to an float64
|
|
func float64ToPtr(f float64) *float64 {
|
|
return &f
|
|
}
|
|
|
|
// generateUUID generates a uuid useful for testing only
|
|
func generateUUID() string {
|
|
buf := make([]byte, 16)
|
|
if _, err := rand.Read(buf); err != nil {
|
|
panic(fmt.Errorf("failed to read random bytes: %v", err))
|
|
}
|
|
|
|
return fmt.Sprintf("%08x-%04x-%04x-%04x-%12x",
|
|
buf[0:4],
|
|
buf[4:6],
|
|
buf[6:8],
|
|
buf[8:10],
|
|
buf[10:16])
|
|
}
|