163 lines
3.2 KiB
Go
163 lines
3.2 KiB
Go
package mock
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
)
|
|
|
|
func GenerateUUID() string {
|
|
buf := make([]byte, 16)
|
|
if _, err := crand.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])
|
|
}
|
|
|
|
func Node() *structs.Node {
|
|
node := &structs.Node{
|
|
ID: GenerateUUID(),
|
|
Datacenter: "dc1",
|
|
Name: "foobar",
|
|
Attributes: map[string]string{
|
|
"os": "linux",
|
|
"arch": "x86",
|
|
"version": "0.1.0",
|
|
"driver.docker": "1.0.0",
|
|
},
|
|
Resources: &structs.Resources{
|
|
CPU: 4.0,
|
|
MemoryMB: 8192,
|
|
DiskMB: 100 * 1024,
|
|
IOPS: 150,
|
|
Networks: []*structs.NetworkResource{
|
|
&structs.NetworkResource{
|
|
Public: true,
|
|
CIDR: "192.168.0.100/32",
|
|
ReservedPorts: []int{22},
|
|
MBits: 1000,
|
|
},
|
|
},
|
|
},
|
|
Reserved: &structs.Resources{
|
|
CPU: 0.1,
|
|
MemoryMB: 256,
|
|
DiskMB: 4 * 1024,
|
|
},
|
|
Links: map[string]string{
|
|
"consul": "foobar.dc1",
|
|
},
|
|
Meta: map[string]string{
|
|
"pci-dss": "true",
|
|
},
|
|
NodeClass: "linux-medium-pci",
|
|
Status: structs.NodeStatusReady,
|
|
}
|
|
return node
|
|
}
|
|
|
|
func Job() *structs.Job {
|
|
job := &structs.Job{
|
|
ID: GenerateUUID(),
|
|
Name: "my-job",
|
|
Type: structs.JobTypeService,
|
|
Priority: 50,
|
|
AllAtOnce: false,
|
|
Datacenters: []string{"dc1"},
|
|
Constraints: []*structs.Constraint{
|
|
&structs.Constraint{
|
|
Hard: true,
|
|
LTarget: "$attr.os",
|
|
RTarget: "linux",
|
|
Operand: "=",
|
|
},
|
|
},
|
|
TaskGroups: []*structs.TaskGroup{
|
|
&structs.TaskGroup{
|
|
Name: "web",
|
|
Count: 10,
|
|
Tasks: []*structs.Task{
|
|
&structs.Task{
|
|
Name: "web",
|
|
Driver: "docker",
|
|
Config: map[string]string{
|
|
"image": "hashicorp/web",
|
|
"version": "v1.2.3",
|
|
},
|
|
Resources: &structs.Resources{
|
|
CPU: 0.5,
|
|
MemoryMB: 256,
|
|
},
|
|
},
|
|
},
|
|
Meta: map[string]string{
|
|
"elb_check_type": "http",
|
|
"elb_check_interval": "30s",
|
|
"elb_check_min": "3",
|
|
},
|
|
},
|
|
},
|
|
Meta: map[string]string{
|
|
"owner": "armon",
|
|
},
|
|
Status: structs.JobStatusPending,
|
|
CreateIndex: 42,
|
|
ModifyIndex: 99,
|
|
}
|
|
return job
|
|
}
|
|
|
|
func Eval() *structs.Evaluation {
|
|
eval := &structs.Evaluation{
|
|
ID: GenerateUUID(),
|
|
Priority: 50,
|
|
Type: structs.JobTypeService,
|
|
JobID: GenerateUUID(),
|
|
Status: structs.EvalStatusPending,
|
|
}
|
|
return eval
|
|
}
|
|
|
|
func Alloc() *structs.Allocation {
|
|
alloc := &structs.Allocation{
|
|
ID: GenerateUUID(),
|
|
EvalID: GenerateUUID(),
|
|
NodeID: "foo",
|
|
Resources: &structs.Resources{
|
|
CPU: 1.0,
|
|
MemoryMB: 1024,
|
|
DiskMB: 1024,
|
|
IOPS: 10,
|
|
Networks: []*structs.NetworkResource{
|
|
&structs.NetworkResource{
|
|
Public: true,
|
|
CIDR: "192.168.0.100/32",
|
|
ReservedPorts: []int{12345},
|
|
MBits: 100,
|
|
},
|
|
},
|
|
},
|
|
Job: Job(),
|
|
Status: structs.AllocStatusPending,
|
|
}
|
|
alloc.JobID = alloc.Job.ID
|
|
return alloc
|
|
}
|
|
|
|
func Plan() *structs.Plan {
|
|
return &structs.Plan{
|
|
Priority: 50,
|
|
}
|
|
}
|
|
|
|
func PlanResult() *structs.PlanResult {
|
|
return &structs.PlanResult{}
|
|
}
|