2 * Copyright (C) 2017-2018 "IoT.bzh"
3 * Author Clément Bénier <clement.benier@iot.bzh>
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
27 "gerrit.automotivelinux.org/gerrit/src/xds/xds-server/lib/xsapiv1"
28 "github.com/stretchr/testify/assert"
31 /*flush channel with timeout*/
32 func flushChannelTerm(channel chan xsapiv1.TerminalOutMsg, ms time.Duration) {
37 case <-time.After(ms * time.Millisecond):
43 func ConnectTargetEvents(t *testing.T, channel chan xsapiv1.TargetConfig) {
44 sCli.Conn.On(xsapiv1.EVTTargetAdd, func(e xsapiv1.EventMsg) {
45 target, _ := e.DecodeTargetEvent()
49 args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetAdd}
50 assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
52 sCli.Conn.On(xsapiv1.EVTTargetRemove, func(e xsapiv1.EventMsg) {
53 target, _ := e.DecodeTargetEvent()
57 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetRemove}
58 assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
61 func DisconnectTargetEvents(t *testing.T) {
62 args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetAdd}
63 assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
64 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetRemove}
65 assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
68 func ConnectTermEvents(t *testing.T, channel chan xsapiv1.TerminalConfig) {
69 sCli.Conn.On(xsapiv1.EVTTargetTerminalAdd, func(e xsapiv1.EventMsg) {
70 termEvt, _ := e.DecodeTerminalEvent()
74 args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalAdd}
75 assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
77 sCli.Conn.On(xsapiv1.EVTTargetTerminalStateChange, func(e xsapiv1.EventMsg) {
78 termEvt, _ := e.DecodeTerminalEvent()
82 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalStateChange}
83 assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
85 sCli.Conn.On(xsapiv1.EVTTargetTerminalRemove, func(e xsapiv1.EventMsg) {
86 termEvt, _ := e.DecodeTerminalEvent()
90 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalRemove}
91 assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
94 func DisconnectTermEvents(t *testing.T) {
95 args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalAdd}
96 assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
97 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalStateChange}
98 assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
99 args = xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTTargetTerminalRemove}
100 assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
103 func TestTarget(t *testing.T) {
104 chTarget := make(chan xsapiv1.TargetConfig)
105 defer close(chTarget)
106 ConnectTargetEvents(t, chTarget)
108 chTermEvt := make(chan xsapiv1.TerminalConfig)
109 defer close(chTermEvt)
110 ConnectTermEvents(t, chTermEvt)
112 var targetArray []xsapiv1.TargetConfig
113 assert.Nil(t, HTTPCli.Get("/targets", &targetArray))
114 assert.Equal(t, len(targetArray), 0)
116 target := xsapiv1.TargetConfig{
118 Type: xsapiv1.TypeTgtStandard,
121 var targetRes, targetResBis xsapiv1.TargetConfig
122 assert.Nil(t, HTTPCli.Post("/targets", target, &targetRes))
123 targetEvt := <-chTarget
124 assert.Equal(t, targetRes.ID, targetEvt.ID)
125 target.Name = "fakeTargetBis"
126 assert.Nil(t, HTTPCli.Post("/targets", target, &targetResBis))
127 targetEvt = <-chTarget
128 assert.Equal(t, targetResBis.ID, targetEvt.ID)
129 assert.Equal(t, targetRes.Status, "Enable")
130 assert.Equal(t, targetResBis.Status, "Enable")
132 term := xsapiv1.TerminalConfig{
134 Type: xsapiv1.TypeTermSSH,
136 var termRes, termResBis xsapiv1.TerminalConfig
137 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
138 termEvt := <-chTermEvt
139 assert.Equal(t, termRes.ID, termEvt.ID)
140 term.Name = "terminalBis"
141 assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals", term, &termResBis))
142 termEvt = <-chTermEvt
143 assert.Equal(t, termResBis.ID, termEvt.ID)
144 assert.Equal(t, termRes.Status, "Enable")
145 assert.Equal(t, termResBis.Status, "Enable")
147 chTerm := make(chan xsapiv1.TerminalOutMsg)
150 sCli.Conn.On(xsapiv1.TerminalOutEvent, func(ev xsapiv1.TerminalOutMsg) {
154 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals/"+termRes.ID+"/open", termRes, &termRes))
155 termEvt = <-chTermEvt
156 assert.Equal(t, termRes.ID, termEvt.ID)
157 assert.Equal(t, termEvt.Status, "Open")
158 assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals/"+termResBis.ID+"/open", termResBis, &termResBis))
159 termEvt = <-chTermEvt
160 assert.Equal(t, termResBis.ID, termEvt.ID)
161 assert.Equal(t, termEvt.Status, "Open")
162 assert.Equal(t, termRes.Status, "Open")
163 assert.Equal(t, termResBis.Status, "Open")
166 flushChannelTerm(chTerm, 50)
167 stdoutMsg := string(termOut.Stdout)
168 if strings.Contains(stdoutMsg, "Connection refused") {
169 t.Fatalf("%vYou may have to launch ssh server", stdoutMsg)
170 } else if strings.Contains(stdoutMsg, "password") {
171 t.Fatalf("%vcopy your pub key in authorized_keys\ncat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys", stdoutMsg)
173 assert.True(t, strings.Contains(stdoutMsg, "Last login"))
175 rootCfgDir := os.Getenv(envRootCfgDir)
176 totoFile := path.Join(rootCfgDir, "toto")
178 /*test with 2 terminals*/
179 for i := 0; i < 2; i++ {
180 totoFileCurrent := totoFile + strconv.Itoa(i)
181 /*send cmd though term*/
182 data := []byte("echo helloWorld" + strconv.Itoa(i) + " >> " + totoFileCurrent + "\n")
184 assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data))
185 flushChannelTerm(chTerm, 50)
187 /*check that toto file is created*/
188 _, err := os.Stat(totoFileCurrent)
191 data = []byte("cat " + totoFileCurrent + "\n")
192 assert.Nil(t, sCli.Conn.Emit(xsapiv1.TerminalInEvent, data))
195 termOut = <-chTerm //result of cat cmd
196 flushChannelTerm(chTerm, 50)
197 assert.Equal(t, string(termOut.Stdout), "helloWorld"+strconv.Itoa(i)+"\r\n")
200 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals/"+termRes.ID+"/close", termRes, &termRes))
201 termEvt = <-chTermEvt
202 assert.Equal(t, termRes.ID, termEvt.ID)
203 assert.Equal(t, termEvt.Status, "Close")
204 assert.Nil(t, HTTPCli.Post("/targets/"+targetResBis.ID+"/terminals/"+termResBis.ID+"/close", termResBis, &termResBis))
205 termEvt = <-chTermEvt
206 assert.Equal(t, termResBis.ID, termEvt.ID)
207 assert.Equal(t, termEvt.Status, "Close")
208 assert.Equal(t, termRes.Status, "Close")
209 assert.Equal(t, termResBis.Status, "Close")
211 /*remove targets and terms*/
212 assert.Nil(t, HTTPCli.Get("/targets", &targetArray))
213 for i := 0; i < len(targetArray); i++ {
214 var termArray []xsapiv1.TerminalConfig
215 assert.Nil(t, HTTPCli.Get("/targets/"+targetArray[i].ID+"/terminals", &termArray))
216 for j := 0; j < len(termArray); j++ {
217 assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes))
218 termEvt = <-chTermEvt
219 assert.Equal(t, termRes.ID, termEvt.ID)
220 assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes))
222 var tgtRes xsapiv1.TargetConfig
223 assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes))
224 targetEvt = <-chTarget
225 assert.Equal(t, tgtRes.ID, targetEvt.ID)
226 assert.Equal(t, targetArray[i].ID, tgtRes.ID)
228 DisconnectTargetEvents(t)
229 DisconnectTermEvents(t)
232 func TestTargetErrors(t *testing.T) {
233 target := xsapiv1.TargetConfig{}
234 var targetRes xsapiv1.TargetConfig
235 assert.NotNil(t, HTTPCli.Post("/targets", target, &targetRes))
236 target.Type = xsapiv1.TypeTgtStandard
237 assert.NotNil(t, HTTPCli.Post("/targets", target, &targetRes))
238 target.IP = "127.0.0.1"
239 assert.Nil(t, HTTPCli.Post("/targets", target, &targetRes))
241 term := xsapiv1.TerminalConfig{}
242 var termRes xsapiv1.TerminalConfig
243 assert.NotNil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
244 term.Type = xsapiv1.TypeTermSSH
245 assert.NotNil(t, HTTPCli.Post("/targets/"+"1010"+"/terminals", term, &termRes))
246 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
247 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
248 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
249 assert.Nil(t, HTTPCli.Post("/targets/"+targetRes.ID+"/terminals", term, &termRes))
251 /*remove targets and terms*/
252 var targetArray []xsapiv1.TargetConfig
253 assert.Nil(t, HTTPCli.Get("/targets", &targetArray))
254 for i := 0; i < len(targetArray); i++ {
255 var termArray []xsapiv1.TerminalConfig
256 assert.Nil(t, HTTPCli.Get("/targets/"+targetArray[i].ID+"/terminals", &termArray))
257 for j := 0; j < len(termArray); j++ {
258 assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes))
259 assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID+"/terminals/"+termArray[j].ID, &termRes))
261 var tgtRes xsapiv1.TargetConfig
262 assert.Nil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes))
263 assert.Equal(t, targetArray[i].ID, tgtRes.ID)
264 assert.NotNil(t, HTTPCli.Delete("/targets/"+targetArray[i].ID, &tgtRes))