diff --git a/testutil/rpcapi/rcpapi.go b/testutil/rpcapi/rcpapi.go deleted file mode 100644 index 1eafabccb..000000000 --- a/testutil/rpcapi/rcpapi.go +++ /dev/null @@ -1,160 +0,0 @@ -package rpcapi - -import ( - "net/rpc" - - msgpackrpc "github.com/hashicorp/net-rpc-msgpackrpc" - "github.com/hashicorp/nomad/nomad/structs" -) - -type RPC struct { - Region string - Namespace string - codec rpc.ClientCodec -} - -func NewRPC(codec rpc.ClientCodec) *RPC { - return &RPC{ - Region: "global", - Namespace: structs.DefaultNamespace, - codec: codec, - } -} - -// AllocAll calls Alloc.List + Alloc.GetAllocs to return all allocs. -func (r *RPC) AllocAll() ([]*structs.Allocation, error) { - listResp, err := r.AllocList() - if err != nil { - return nil, err - } - - ids := make([]string, 0, len(listResp.Allocations)) - for _, a := range listResp.Allocations { - ids = append(ids, a.ID) - } - - allocsResp, err := r.AllocGetAllocs(ids) - if err != nil { - return nil, err - } - return allocsResp.Allocs, nil -} - -// Alloc.List RPC -func (r *RPC) AllocList() (*structs.AllocListResponse, error) { - get := &structs.AllocListRequest{ - QueryOptions: structs.QueryOptions{ - Region: r.Region, - Namespace: r.Namespace, - }, - } - - var resp structs.AllocListResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Alloc.List", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Alloc.GetAllocs RPC -func (r *RPC) AllocGetAllocs(ids []string) (*structs.AllocsGetResponse, error) { - get := &structs.AllocsGetRequest{ - AllocIDs: ids, - QueryOptions: structs.QueryOptions{ - Region: r.Region, - Namespace: r.Namespace, - }, - } - var resp structs.AllocsGetResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Alloc.GetAllocs", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Eval.List RPC -func (r *RPC) EvalList() (*structs.EvalListResponse, error) { - get := &structs.EvalListRequest{ - QueryOptions: structs.QueryOptions{ - Region: r.Region, - Namespace: r.Namespace, - }, - } - var resp structs.EvalListResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Eval.List", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Job.List RPC -func (r *RPC) JobList() (*structs.JobListResponse, error) { - get := &structs.JobListRequest{ - QueryOptions: structs.QueryOptions{ - Region: r.Region, - Namespace: r.Namespace, - }, - } - - var resp structs.JobListResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Job.List", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Job.Register RPC -func (r *RPC) JobRegister(j *structs.Job) (*structs.JobRegisterResponse, error) { - req := &structs.JobRegisterRequest{ - Job: j.Copy(), - WriteRequest: structs.WriteRequest{ - Region: r.Region, - Namespace: j.Namespace, - }, - } - - // Fetch the response - var resp structs.JobRegisterResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Job.Register", req, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Node.List RPC -func (r *RPC) NodeList() (*structs.NodeListResponse, error) { - get := &structs.NodeListRequest{ - QueryOptions: structs.QueryOptions{Region: r.Region}, - } - var resp structs.NodeListResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Node.List", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Node.GetAllocs RPC -func (r *RPC) NodeGetAllocs(nodeID string) (*structs.NodeAllocsResponse, error) { - get := &structs.NodeSpecificRequest{ - NodeID: nodeID, - QueryOptions: structs.QueryOptions{Region: r.Region}, - } - var resp structs.NodeAllocsResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Node.GetAllocs", get, &resp); err != nil { - return nil, err - } - return &resp, nil -} - -// Node.GetNode RPC -func (r *RPC) NodeGet(nodeID string) (*structs.SingleNodeResponse, error) { - get := &structs.NodeSpecificRequest{ - NodeID: nodeID, - QueryOptions: structs.QueryOptions{Region: r.Region}, - } - var resp structs.SingleNodeResponse - if err := msgpackrpc.CallWithCodec(r.codec, "Node.GetNode", get, &resp); err != nil { - return nil, err - } - return &resp, nil -}