tests: better handle websocket client
[src/xds/xds-server.git] / test / sdks_test.go
1 /*
2  * Copyright (C) 2017-2018 "IoT.bzh"
3  * Author Clément Bénier <clement.benier@iot.bzh>
4  *
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
8  *
9  *   http://www.apache.org/licenses/LICENSE-2.0
10  *
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.
16  */
17 package xdsservertest
18
19 import (
20         "os"
21         "path"
22         "regexp"
23         "testing"
24         "time"
25
26         "gerrit.automotivelinux.org/gerrit/src/xds/xds-server/lib/xsapiv1"
27         "github.com/stretchr/testify/assert"
28 )
29
30 func ConnectSDKStateChange(t *testing.T, sCli *IOSockClient, channel chan xsapiv1.SDK) {
31
32         sCli.Conn.On(xsapiv1.EVTSDKStateChange, func(e xsapiv1.EventMsg) {
33                 sdk, _ := e.DecodeSDKEvent()
34                 channel <- sdk
35         })
36
37         args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTSDKStateChange}
38         assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
39 }
40
41 func TestSdks(t *testing.T) {
42         var sdkArray []xsapiv1.SDK
43         assert.Nil(t, HTTPCli.Get("/sdks", &sdkArray))
44         _, err := os.Stat(path.Join(sdkDir, "sdks_latest.json"))
45         assert.Nil(t, err)
46         t.Logf("Connect websocket with url=%v clientId=%v\n", prefixURL, HTTPCli.GetClientID())
47
48         chSdks := make(chan xsapiv1.SDK)
49         ConnectSDKStateChange(t, sCli, chSdks)
50
51         for i := 0; i < len(sdkArray); i++ {
52                 re := regexp.MustCompile("^[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+$")
53                 assert.True(t, re.MatchString(sdkArray[i].ID))
54                 assert.Equal(t, sdkArray[0].Status, "Not Installed")
55
56                 var sdk xsapiv1.SDK
57                 assert.Nil(t, HTTPCli.Get("/sdks/"+sdkArray[i].ID, &sdk))
58                 assert.Equal(t, sdkArray[i], sdk)
59
60                 var sdkRes xsapiv1.SDK
61                 assert.Nil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
62                 assert.Equal(t, sdkRes.Status, "Installing")
63                 time.Sleep(1 * time.Millisecond)
64                 assert.Nil(t, HTTPCli.Post("/sdks/abortinstall", sdk, &sdkRes))
65                 assert.Equal(t, sdkRes.Status, "Not Installed")
66                 time.Sleep(1 * time.Millisecond)
67                 sdkFromEvent := <-chSdks
68                 assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
69                 assert.Equal(t, sdkRes.Status, sdkFromEvent.Status)
70         }
71 }
72
73 func installFakeSdk(t *testing.T, sdkInstall xsapiv1.SDKInstallArgs, chSdks chan xsapiv1.SDK) xsapiv1.SDK {
74         var sdkRes xsapiv1.SDK
75         assert.Nil(t, HTTPCli.Post("/sdks", sdkInstall, &sdkRes))
76         assert.Equal(t, sdkRes.Status, "Installing")
77         /*waiting for SDKStateChange event from channel*/
78         sdkFromEvent := <-chSdks
79         t.Log(sdkFromEvent)
80         assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
81         assert.Equal(t, "Installed", sdkFromEvent.Status)
82         return sdkFromEvent
83 }
84
85 func TestInstallFakeSdk(t *testing.T) {
86         sdk := xsapiv1.SDKInstallArgs{
87                 ID:       "",
88                 Filename: sdkFileName,
89                 Force:    false,
90         }
91
92         chSdks := make(chan xsapiv1.SDK)
93         ConnectSDKStateChange(t, sCli, chSdks)
94         installFakeSdk(t, sdk, chSdks)
95         var sdkRes xsapiv1.SDK
96         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
97         sdk.Force = true
98         //installFakeSdk(t, sdk, chSdks)
99 }
100
101 func TestSdksError(t *testing.T) {
102         sdk := xsapiv1.SDKInstallArgs{
103                 ID:       "",
104                 Filename: "",
105                 Force:    false,
106         }
107         var sdkRes xsapiv1.SDK
108         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
109         sdk.ID = "42885c24-374e-3ef0-9723-0ce8a05191aa"
110         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
111         sdk.Filename = "fake"
112         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
113 }