27 lines
548 B
Go
27 lines
548 B
Go
package scheduler
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/nomad/nomad/state"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
)
|
|
|
|
func testContext(t *testing.T) (*state.StateStore, *EvalContext) {
|
|
state, err := state.NewStateStore(os.Stderr)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
plan := &structs.Plan{
|
|
NodeEvict: make(map[string][]string),
|
|
NodeAllocation: make(map[string][]*structs.Allocation),
|
|
}
|
|
|
|
logger := log.New(os.Stderr, "", log.LstdFlags)
|
|
|
|
ctx := NewEvalContext(state, plan, logger)
|
|
return state, ctx
|
|
}
|