func TestDispatcher(t *testing.T) {
fmt.Println("Starting")
lm := kamaji.NewLicenseManager()
nm := kamaji.NewNodeManager("", 6666)
go nm.Start()
tm := kamaji.NewTaskManager()
d := kamaji.NewDispatcher(lm, nm, tm)
job_count := 10
task_count := 5
command_count := 20
for i := 0; i < job_count; i++ {
job := kamaji.NewJob(fmt.Sprintf("Job %d", i))
for j := 0; j < task_count; j++ {
task := kamaji.NewTask(fmt.Sprintf("Task %d", j), job, []string{})
for k := 0; k < command_count; k++ {
_ = kamaji.NewCommand(fmt.Sprintf("Command %d", k), task)
}
}
tm.AddJob(job)
}
go d.Start()
client_count := 1
fmt.Println("Starting: ", client_count, " clients.")
quit := make(chan bool, client_count)
for i := 0; i < client_count; i++ {
go cli(i, quit)
time.Sleep(time.Millisecond * 2)
}
time.Sleep(time.Second)
for i := 0; i < client_count; i++ {
quit <- true
}
}
func main() {
fmt.Println("Starting")
// Create all managers
lm := kamaji.NewLicenseManager()
lm.AddLicense("maya", 4)
lm.AddLicense("nuke", 4)
nm := kamaji.NewNodeManager("", 1314)
tm := kamaji.NewTaskManager()
d := kamaji.NewDispatcher(lm, nm, tm)
s := kamaji.NewService("", 8080, lm, nm, tm)
// Create signal handler
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for sig := range c {
switch sig {
case os.Interrupt:
fmt.Printf("\nCatched Interupt Signal, Cleaning up\n")
d.Stop()
}
}
}()
// Create Test Jobs
job_count := 10
task_count := 2
command_count := 50
for i := 1; i < job_count+1; i++ {
lic_name := "maya"
if i > 6 {
lic_name = "nuke"
}
job := kamaji.NewJob(fmt.Sprintf("Job %d | %s", i, lic_name))
for j := 0; j < task_count; j++ {
task := kamaji.NewTask(fmt.Sprintf("Task %d [%s]", j, lic_name), job, []string{lic_name})
for k := 0; k < command_count; k++ {
_ = kamaji.NewCommand(fmt.Sprintf("Command %d", k), task)
}
}
tm.AddJob(job)
}
go nm.Start()
defer nm.Stop()
go tm.Start()
defer tm.Stop()
go s.Start()
defer s.Stop()
d.Start()
}
// Test Bringing up 100 nodes and then stop the nodemanager.
func TestNodeManager(t *testing.T) {
port := 1414
numClients := 100
nm := kamaji.NewNodeManager("", port)
go nm.Start()
time.Sleep(time.Millisecond * 10)
for i := 0; i < numClients; i++ {
go client(t, false, strconv.Itoa(port))
time.Sleep(time.Millisecond * 5)
}
for {
if len(nm.Nodes) == numClients {
break
}
}
nm.Stop()
}