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         "fmt"
  21         "log"
  22         "os"
  23         "path"
  24         "regexp"
  25         "sync"
  26         "testing"
  27         "time"
  28
  29         "gerrit.automotivelinux.org/gerrit/src/xds/xds-server/lib/xsapiv1"
  30         socketio_client "github.com/sebd71/go-socket.io-client"
  31         "github.com/stretchr/testify/assert"
  32 )
  33
  34 var ch chan xsapiv1.SDK
  35
  36 // Disconnection Channel used to notify XDS Server disconnection
  37 //type Disconnection struct {
  38 //      error   string
  39 //      code    int
  40 //      svrConf ServerCfg
  41 //}
  42
  43 // IOSockClient .
  44 type IOSockClient struct {
  45         URL       string
  46         Conn      *socketio_client.Client
  47         Options   *socketio_client.Options
  48         EmitMutex *sync.Mutex
  49         Connected bool
  50         //ServerDiscoChan chan Disconnection
  51         EscapeKeys []byte
  52 }
  53
  54 func NewIoSocketClient(url, clientID string) (*IOSockClient, error) {
  55
  56         var err error
  57
  58         sCli := &IOSockClient{
  59                 URL:       url,
  60                 EmitMutex: &sync.Mutex{},
  61                 Options: &socketio_client.Options{
  62                         Transport: "websocket",
  63                         Header:    make(map[string][]string),
  64                 },
  65                 //ServerDiscoChan: make(chan Disconnection, 1),
  66         }
  67         sCli.Options.Header["XDS-SID"] = []string{clientID}
  68
  69         sCli.Conn, err = socketio_client.NewClient(url, sCli.Options)
  70         if err != nil {
  71                 return nil, fmt.Errorf("IO.socket connection error: " + err.Error())
  72         }
  73
  74         sCli.Conn.On("connection", func() {
  75                 sCli.Connected = true
  76         })
  77
  78         sCli.Conn.On("disconnection", func(err error) {
  79                 log.Printf("WS disconnection event with err: %v\n", err)
  80                 sCli.Connected = false
  81         })
  82
  83         sCli.Conn.On(xsapiv1.EVTSDKStateChange, func(e xsapiv1.EventMsg) {
  84                 sdk, _ := e.DecodeSDKEvent()
  85                 ch <- sdk
  86         })
  87
  88         args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTSDKStateChange}
  89         if err := HTTPCli.Post("/events/register", args, nil); err != nil {
  90                 return sCli, err
  91         }
  92
  93         //sCli.Conn.On(xsapiv1.)
  94
  95         return sCli, nil
  96 }
  97
  98 func TestSdks(t *testing.T) {
  99         var sdkArray []xsapiv1.SDK
 100         assert.Nil(t, HTTPCli.Get("/sdks", &sdkArray))
 101         _, err := os.Stat(path.Join(sdkDir, "sdks_latest.json"))
 102         assert.Nil(t, err)
 103         t.Logf("Connect websocket with url=%v clientId=%v\n", prefixURL, HTTPCli.GetClientID())
 104         _, err = NewIoSocketClient(prefixURL, HTTPCli.GetClientID())
 105         if err != nil {
 106                 t.Fatal(err)
 107         }
 108
 109         ch = make(chan xsapiv1.SDK)
 110
 111         for i := 0; i < len(sdkArray); i++ {
 112                 re := regexp.MustCompile("^[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+$")
 113                 assert.True(t, re.MatchString(sdkArray[i].ID))
 114                 assert.Equal(t, sdkArray[0].Status, "Not Installed")
 115
 116                 var sdk xsapiv1.SDK
 117                 assert.Nil(t, HTTPCli.Get("/sdks/"+sdkArray[i].ID, &sdk))
 118                 assert.Equal(t, sdkArray[i], sdk)
 119
 120                 var sdkRes xsapiv1.SDK
 121                 assert.Nil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 122                 assert.Equal(t, sdkRes.Status, "Installing")
 123                 time.Sleep(1 * time.Millisecond)
 124                 assert.Nil(t, HTTPCli.Post("/sdks/abortinstall", sdk, &sdkRes))
 125                 assert.Equal(t, sdkRes.Status, "Not Installed")
 126                 time.Sleep(1 * time.Millisecond)
 127                 sdkFromEvent := <-ch
 128                 assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 129                 assert.Equal(t, sdkRes.Status, sdkFromEvent.Status)
 130         }
 131 }
 132
 133 func installFakeSdk(t *testing.T, sdkInstall xsapiv1.SDKInstallArgs) xsapiv1.SDK {
 134         var sdkRes xsapiv1.SDK
 135         assert.Nil(t, HTTPCli.Post("/sdks", sdkInstall, &sdkRes))
 136         assert.Equal(t, sdkRes.Status, "Installing")
 137         /*waiting for SDKStateChange event from channel*/
 138         sdkFromEvent := <-ch
 139         t.Log(sdkFromEvent)
 140         assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 141         assert.Equal(t, "Installed", sdkFromEvent.Status)
 142         return sdkFromEvent
 143 }
 144
 145 func TestInstallFakeSdk(t *testing.T) {
 146         sdk := xsapiv1.SDKInstallArgs{
 147                 ID:       "",
 148                 Filename: sdkFileName,
 149                 Force:    false,
 150         }
 151         installFakeSdk(t, sdk)
 152         var sdkRes xsapiv1.SDK
 153         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 154         sdk.Force = true
 155         //installFakeSdk(t, sdk)
 156 }
 157
 158 func TestSdksError(t *testing.T) {
 159         sdk := xsapiv1.SDKInstallArgs{
 160                 ID:       "",
 161                 Filename: "",
 162                 Force:    false,
 163         }
 164         var sdkRes xsapiv1.SDK
 165         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 166         sdk.ID = "42885c24-374e-3ef0-9723-0ce8a05191aa"
 167         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 168         sdk.Filename = "fake"
 169         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 170 }