X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;ds=sidebyside;f=test%2Ftarget_test.go;h=e89011bf8534bbcf29877f87e30ed78a99d64243;hb=9cc4aa5bcfa8fbab2d367fde9bc5ba4ff204c5e7;hp=696c8fcf8ad9adbb6b33086b3bc6f2ced876c9a1;hpb=4c01e5e3fad9455b97519380babdf7e8801d8643;p=src%2Fxds%2Fxds-server.git diff --git a/test/target_test.go b/test/target_test.go index 696c8fc..e89011b 100644 --- a/test/target_test.go +++ b/test/target_test.go @@ -17,8 +17,6 @@ package xdsservertest import ( - "os" - "path" "strconv" "strings" "testing" @@ -40,122 +38,295 @@ func flushChannelTerm(channel chan xsapiv1.TerminalOutMsg, ms time.Duration) { } } -func TestTarget(t *testing.T) { - var targetArray []xsapiv1.TargetConfig - assert.Nil(t, HTTPCli.Get("/targets", &targetArray)) - assert.Equal(t, len(targetArray), 0) +func ConnectTargetEvents(t *testing.T, channel chan xsapiv1.TargetConfig) { + sCli.Conn.On(xsapiv1.EVTTargetAdd, func(e xsapiv1.EventMsg) { + target, _ := e.DecodeTargetEvent() + channel <- target + }) - target := xsapiv1.TargetConfig{ - Name: "fakeTarget", - Type: xsapiv1.TypeTgtStandard, - IP: "127.0.0.1", - } - var targetRes, targetResBis xsapiv1.TargetConfig - assert.Nil(t, HTTPCli.Post("/targets", target, &targetRes)) - target.Name = "fakeTargetBis" - assert.Nil(t, HTTPCli.Post("/targets", target, &targetResBis)) - assert.Equal(t, targetRes.Status, "Enable") - assert.Equal(t, targetResBis.Status, "Enable") - - term := xsapiv1.TerminalConfig{ - Name: "terminal", - Type: xsapiv1.TypeTermSSH, - } - var termRes, termResBis xsapiv1.TerminalConfig - assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) - term.Name = "terminalBis" - assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals", term, &termResBis)) - assert.Equal(t, termRes.Status, "Enable") - assert.Equal(t, termResBis.Status, "Enable") - - chTerm := make(chan xsapiv1.TerminalOutMsg) - defer close(chTerm) - sCli, err := NewIoSocketClient(prefixURL, HTTPCli.GetClientID()) - if err != nil { - t.Fatal(err) - } - sCli.Conn.On(xsapiv1.TerminalOutEvent, func(ev xsapiv1.TerminalOutMsg) { - chTerm <- ev + args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetAdd} + assert.Nil(t, HTTPCli.Post("/events/register", args, nil)) + + sCli.Conn.On(xsapiv1.EVTTargetRemove, func(e xsapiv1.EventMsg) { + target, _ := e.DecodeTargetEvent() + channel <- target }) - assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals/"+termRes.ID+"/open", termRes, &termRes)) - assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals/"+termResBis.ID+"/open", termResBis, &termResBis)) - assert.Equal(t, termRes.Status, "Open") - assert.Equal(t, termResBis.Status, "Open") - - termOut := <-chTerm - flushChannelTerm(chTerm, 50) - stdoutMsg := string(termOut.Stdout) - if strings.Contains(stdoutMsg, "Connection refused") { - t.Fatalf("%vYou may have to launch ssh server", stdoutMsg) - } else if strings.Contains(stdoutMsg, "password") { - t.Fatalf("%vcopy your pub key in authorized_keys\ncat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys", stdoutMsg) - } - assert.True(t, strings.Contains(stdoutMsg, "Last login")) + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetRemove} + assert.Nil(t, HTTPCli.Post("/events/register", args, nil)) +} - rootCfgDir := os.Getenv(envRootCfgDir) - totoFile := path.Join(rootCfgDir, "toto") +func DisconnectTargetEvents(t *testing.T) { + args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetAdd} + assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil)) + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetRemove} + assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil)) +} - /*test with 2 terminals*/ - for i := 0; i < 2; i++ { - totoFileCurrent := totoFile + strconv.Itoa(i) - /*send cmd though term*/ - data := []byte("echo helloWorld" + strconv.Itoa(i) + " >> " + totoFileCurrent + "\n") +func ConnectTermEvents(t *testing.T, channel chan xsapiv1.TerminalConfig) { + sCli.Conn.On(xsapiv1.EVTTargetTerminalAdd, func(e xsapiv1.EventMsg) { + termEvt, _ := e.DecodeTerminalEvent() + channel <- termEvt + }) - assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data)) - flushChannelTerm(chTerm, 50) + args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalAdd} + assert.Nil(t, HTTPCli.Post("/events/register", args, nil)) - /*check that toto file is created*/ - _, err = os.Stat(totoFileCurrent) - assert.Nil(t, err) + sCli.Conn.On(xsapiv1.EVTTargetTerminalStateChange, func(e xsapiv1.EventMsg) { + termEvt, _ := e.DecodeTerminalEvent() + channel <- termEvt + }) - data = []byte("cat " + totoFileCurrent + "\n") - assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data)) + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalStateChange} + assert.Nil(t, HTTPCli.Post("/events/register", args, nil)) - <-chTerm //cmd sent - termOut = <-chTerm //result of cat cmd - flushChannelTerm(chTerm, 50) - assert.Equal(t, string(termOut.Stdout), "helloWorld"+strconv.Itoa(i)+"\r\n") + sCli.Conn.On(xsapiv1.EVTTargetTerminalRemove, func(e xsapiv1.EventMsg) { + termEvt, _ := e.DecodeTerminalEvent() + channel <- termEvt + }) + + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalRemove} + assert.Nil(t, HTTPCli.Post("/events/register", args, nil)) +} + +func DisconnectTermEvents(t *testing.T) { + args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalAdd} + assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil)) + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalStateChange} + assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil)) + args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalRemove} + assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil)) +} + +func AddTargets(t *testing.T, nbTargets int, chTarget chan xsapiv1.TargetConfig) []string { + listID := make([]string, nbTargets) + for i := 0; i < nbTargets; i++ { + /*target is local*/ + target := xsapiv1.TargetConfig{ + Name: "fakeTarget" + strconv.Itoa(i), + Type: xsapiv1.TypeTgtStandard, + IP: "127.0.0.1", + } + /*add target*/ + assert.Nil(t, HTTPCli.Post("/targets", target, &target)) + Debugf(t, "add target %v", target.Name) + targetEvt := <-chTarget //waiting for event targetAdd + assert.Equal(t, target.ID, targetEvt.ID) + listID[i] = target.ID } + for i := 0; i < nbTargets; i++ { + var target xsapiv1.TargetConfig + assert.Nil(t, HTTPCli.Get("/targets/"+listID[i], &target)) + assert.Equal(t, target.Status, "Enable") + } + return listID +} - assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals/"+termRes.ID+"/close", termRes, &termRes)) - assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals/"+termResBis.ID+"/close", termResBis, &termResBis)) - assert.Equal(t, termRes.Status, "Close") - assert.Equal(t, termResBis.Status, "Close") +func AddTerms(t *testing.T, nbTerms int, listID []string, chTermEvt chan xsapiv1.TerminalConfig) { + for j := 0; j < len(listID); j++ { + listTermsID := make([]string, nbTerms) + for i := 0; i < nbTerms; i++ { + term := xsapiv1.TerminalConfig{ + Name: "terminal" + strconv.Itoa(i), + Type: xsapiv1.TypeTermSSH, + } + /*add terminal on target*/ + assert.Nil(t, HTTPCli.Post("/targets/"+listID[j]+"/terminals", term, &term)) + Debugf(t, "add terminal %v", term.Name) + termEvt := <-chTermEvt //waiting for event terminalAdd*/ + assert.Equal(t, term.ID, termEvt.ID) + listTermsID[i] = term.ID + } + assert.Equal(t, len(listTermsID), nbTerms) + for i := 0; i < nbTerms; i++ { + var term xsapiv1.TerminalConfig + assert.Nil(t, HTTPCli.Get("/targets/"+listID[j]+"/terminals/"+listTermsID[i], &term)) + assert.Equal(t, term.Status, "Close") + } + } +} - /*remove targets and terms*/ - assert.Nil(t, HTTPCli.Get("/targets", &targetArray)) - for i := 0; i < len(targetArray); i++ { - var termArray []xsapiv1.TerminalConfig - assert.Nil(t, HTTPCli.Get("/targets/"+targetArray[i].ID+"/terminals", &termArray)) - for j := 0; j < len(termArray); j++ { - assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes)) - assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes)) +func PostTerms(t *testing.T, post string, chTermEvt chan xsapiv1.TerminalConfig) { + var status string + switch post { + case "open": + status = "Open" + case "close": + status = "Closing" + } + var targets []xsapiv1.TargetConfig + assert.Nil(t, HTTPCli.Get("/targets", &targets)) + for i := 0; i < len(targets); i++ { + var terms []xsapiv1.TerminalConfig + assert.Nil(t, HTTPCli.Get("/targets/"+targets[i].ID+"/terminals", &terms)) + listTermsID := make([]string, len(terms)) + for j := 0; j < len(terms); j++ { + var term xsapiv1.TerminalConfig + /*post action on term*/ + assert.Nil(t, HTTPCli.Post("/targets/"+targets[i].ID+"/terminals/"+terms[j].ID+"/"+post, terms[j], &term)) + Debugf(t, "%v terminal %v", post, term.Name) + termEvt := <-chTermEvt //waiting for event terminalStateChange + assert.Equal(t, term.ID, termEvt.ID) + assert.Equal(t, term.Status, status) + assert.Equal(t, termEvt.Status, status) + listTermsID[i] = term.ID + } + time.Sleep(10 * time.Millisecond) + for j := 0; j < len(listTermsID); j++ { + var term xsapiv1.TerminalConfig + assert.Nil(t, HTTPCli.Get("/targets/"+targets[i].ID+"/terminals/"+listTermsID[i], &term)) + assert.True(t, strings.EqualFold(term.Status, post)) + Debugf(t, "check that term status %v is %v", term.Name, post) + } + } +} + +func RemoveTermsTargets(t *testing.T, chTarget chan xsapiv1.TargetConfig, chTermEvt chan xsapiv1.TerminalConfig) { + var targets []xsapiv1.TargetConfig + assert.Nil(t, HTTPCli.Get("/targets", &targets)) + for i := 0; i < len(targets); i++ { + var terms []xsapiv1.TerminalConfig + assert.Nil(t, HTTPCli.Get("/targets/"+targets[i].ID+"/terminals", &terms)) + for j := 0; j < len(terms); j++ { + var term xsapiv1.TerminalConfig + assert.Nil(t, HTTPCli.Delete("/targets/"+targets[i].ID+"/terminals/"+terms[j].ID, &term)) + termEvt := <-chTermEvt + assert.Equal(t, term.ID, termEvt.ID) + assert.NotNil(t, HTTPCli.Delete("/targets/"+targets[i].ID+"/terminals/"+terms[j].ID, &term)) + Debugf(t, "remove terminal %v", term.Name) } var tgtRes xsapiv1.TargetConfig - assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes)) - assert.Equal(t, targetArray[i].ID, tgtRes.ID) + assert.Nil(t, HTTPCli.Delete("/targets/"+targets[i].ID, &tgtRes)) + targetEvt := <-chTarget //waiting for remove terminal event + assert.Equal(t, tgtRes.ID, targetEvt.ID) + assert.Equal(t, targets[i].ID, tgtRes.ID) } } +func TestTarget(t *testing.T) { + nbTargets := 3 + nbTermsByTarget := 3 + /*channel for target events*/ + chTarget := make(chan xsapiv1.TargetConfig) + defer close(chTarget) + ConnectTargetEvents(t, chTarget) + + /*channel for terminal events*/ + chTermEvt := make(chan xsapiv1.TerminalConfig) + defer close(chTermEvt) + ConnectTermEvents(t, chTermEvt) + + /*check that targetArray is empty at startup*/ + var targetArray []xsapiv1.TargetConfig + assert.Nil(t, HTTPCli.Get("/targets", &targetArray)) + assert.Equal(t, len(targetArray), 0) + + listID := AddTargets(t, nbTargets, chTarget) + AddTerms(t, nbTermsByTarget, listID, chTermEvt) + + ///*channel for TerminalOutMsg*/ + //chTerm := make(chan xsapiv1.TerminalOutMsg) + //defer close(chTerm) + + ///*connect on terminalOutMsg event*/ + //sCli.Conn.On(xsapiv1.TerminalOutEvent, func(ev xsapiv1.TerminalOutMsg) { + // chTerm <- ev + //}) + + ///*just for the first term*/ + //var terms []xsapiv1.TerminalConfig + //var term xsapiv1.TerminalConfig + //assert.Nil(t, HTTPCli.Get("/targets/"+listID[0]+"/terminals", &terms)) + //assert.Nil(t, HTTPCli.Post("/targets/"+listID[0]+"/terminals/"+terms[0].ID+"/open", terms[0], &term)) + //<-chTermEvt //waiting for event terminalStateChange + //termOut := <-chTerm //waiting for terminalOutMsg + //flushChannelTerm(chTerm, 50) //flushing all terminalOutMsg + //stdoutMsg := string(termOut.Stdout) + //if strings.Contains(stdoutMsg, "Connection refused") { + // t.Fatalf("%vYou may have to launch ssh server", stdoutMsg) + //} else if strings.Contains(stdoutMsg, "password") { + // t.Fatalf("%vcopy your pub key in authorized_keys\ncat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys", stdoutMsg) + //} + //assert.True(t, strings.Contains(stdoutMsg, "Last login")) //first terminal msg should be Last Login + //assert.Nil(t, HTTPCli.Post("/targets/"+listID[0]+"/terminals/"+terms[0].ID+"/close", terms[0], &term)) + //<-chTermEvt //waiting for event terminalStateChange + + ///*open terminals*/ + //PostTerms(t, "open", chTermEvt) + //termOut = <-chTerm //waiting for terminalOutMsg + //flushChannelTerm(chTerm, 50) //flushing all terminalOutMsg + //stdoutMsg = string(termOut.Stdout) + //if strings.Contains(stdoutMsg, "Connection refused") { + // t.Fatalf("%vYou may have to launch ssh server", stdoutMsg) + //} else if strings.Contains(stdoutMsg, "password") { + // t.Fatalf("%vcopy your pub key in authorized_keys\ncat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys", stdoutMsg) + //} + //assert.True(t, strings.Contains(stdoutMsg, "Last login")) //first terminal msg should be Last Login + + ///*create toto file through terminals*/ + //rootCfgDir := os.Getenv(envRootCfgDir) + //totoFile := path.Join(rootCfgDir, "toto") + + ///*test with 2 terminals*/ + //for i := 0; i < 2; i++ { + // totoFileCurrent := totoFile + strconv.Itoa(i) + // /*send cmd though term*/ + // data := []byte("echo helloWorld" + strconv.Itoa(i) + " >> " + totoFileCurrent + "\n") + // Debugf(t, "send following command through terminal: %v", string(data)) + // assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data)) + // flushChannelTerm(chTerm, 50) //waiting for terminal msg + + // /*check that toto file is created*/ + // _, err := os.Stat(totoFileCurrent) + // assert.Nil(t, err) + + // /*send cmd though term*/ + // data = []byte("cat " + totoFileCurrent + "\n") + // Debugf(t, "send following command through terminal: %v", string(data)) + // assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data)) + + // <-chTerm //cmd sent + // termOut = <-chTerm //result of cat cmd + // flushChannelTerm(chTerm, 50) //flushing what remains + // /*check that terminal msg is what was written before*/ + // assert.Equal(t, string(termOut.Stdout), "helloWorld"+strconv.Itoa(i)+"\r\n") + // Debugf(t, "check terminal output msg: %v", string(termOut.Stdout)) + //} + + //PostTerms(t, "close", chTermEvt) + + /*remove targets and terms*/ + RemoveTermsTargets(t, chTarget, chTermEvt) + DisconnectTargetEvents(t) + DisconnectTermEvents(t) +} func TestTargetErrors(t *testing.T) { + /*cannot create empty target*/ target := xsapiv1.TargetConfig{} var targetRes xsapiv1.TargetConfig assert.NotNil(t, HTTPCli.Post("/targets", target, &targetRes)) + Debugf(t, "error while creating empty target") + /*check cannot create target with no IP*/ target.Type = xsapiv1.TypeTgtStandard assert.NotNil(t, HTTPCli.Post("/targets", target, &targetRes)) + Debugf(t, "error while creating target without IP") target.IP = "127.0.0.1" assert.Nil(t, HTTPCli.Post("/targets", target, &targetRes)) + Debugf(t, "create target %v", targetRes.Name) + /*cannot create empty terminal*/ term := xsapiv1.TerminalConfig{} var termRes xsapiv1.TerminalConfig assert.NotNil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) + Debugf(t, "error while creating empty terminal") term.Type = xsapiv1.TypeTermSSH assert.NotNil(t, HTTPCli.Post("/targets/"+"1010"+"/terminals", term, &termRes)) + Debugf(t, "error while creating terminal on an non existing target") assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes)) + Debugf(t, "create several terminals") /*remove targets and terms*/ var targetArray []xsapiv1.TargetConfig @@ -165,11 +336,15 @@ func TestTargetErrors(t *testing.T) { assert.Nil(t, HTTPCli.Get("/targets/"+targetArray[i].ID+"/terminals", &termArray)) for j := 0; j < len(termArray); j++ { assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes)) + Debugf(t, "delete terminal %v", termRes.Name) assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes)) + Debugf(t, "error while deleting an already deleted terminal %v", termRes.Name) } var tgtRes xsapiv1.TargetConfig assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes)) + Debugf(t, "delete target %v", tgtRes.Name) assert.Equal(t, targetArray[i].ID, tgtRes.ID) assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes)) + Debugf(t, "error while deleting an already deleted target %v", tgtRes.Name) } }