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         "log"
  21         "os"
  22         "path"
  23         "regexp"
  24         "strings"
  25         "testing"
  26         "time"
  27
  28         "gerrit.automotivelinux.org/gerrit/src/xds/xds-server/lib/xsapiv1"
  29         "github.com/stretchr/testify/assert"
  30 )
  31
  32 var cpt int
  33
  34 /*flush channel with timeout*/
  35 func flushChannelSdk(channel chan xsapiv1.SDK, ms time.Duration) {
  36         timeoutB := false
  37         for !timeoutB {
  38                 select {
  39                 case <-channel:
  40                 case <-time.After(ms * time.Millisecond):
  41                         timeoutB = true
  42                 }
  43         }
  44 }
  45
  46 func ConnectSDKStateChange(t *testing.T, sCli *IOSockClient, channel chan xsapiv1.SDK) {
  47
  48         sCli.Conn.On(xsapiv1.EVTSDKStateChange, func(e xsapiv1.EventMsg) {
  49                 sdk, _ := e.DecodeSDKEvent()
  50                 channel <- sdk
  51         })
  52
  53         args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTSDKStateChange}
  54         assert.Nil(t, HTTPCli.Post("/events/register", args, nil))
  55 }
  56
  57 func DisconnectSDKStateChange(t *testing.T, sCli *IOSockClient) {
  58         args := xsapiv1.EventRegisterArgs{Name: xsapiv1.EVTSDKStateChange}
  59         assert.Nil(t, HTTPCli.Post("/events/unregister", args, nil))
  60 }
  61
  62 func InitSdkDir() {
  63         sdkDir := os.Getenv(envXdtSdk)
  64         os.MkdirAll(sdkDir, 0755)
  65         currentDir, err := os.Getwd()
  66         if err != nil {
  67                 log.Fatal(err)
  68         }
  69         if err := Copy(path.Join(currentDir, sdkFile), path.Join(sdkDir, sdkFileName)); err != nil {
  70                 log.Fatal(err)
  71         }
  72 }
  73
  74 func RemoveSdk(t *testing.T, sdk xsapiv1.SDK, chSdks chan xsapiv1.SDK) {
  75         assert.Nil(t, HTTPCli.Delete("/sdks/"+sdk.ID, &sdk))
  76         sdkFromEvt := <-chSdks //Uninstalling
  77         assert.Equal(t, sdk.ID, sdkFromEvt.ID)
  78         assert.Equal(t, sdkFromEvt.Status, "Un-installing")
  79         sdkFromEvt = <-chSdks //not installed
  80         assert.Equal(t, sdkFromEvt.Status, "Not Installed")
  81         os.RemoveAll(path.Join(os.Getenv(envXdtSdk), sdk.Profile))
  82         os.Remove(strings.Replace(sdk.URL, "file://", "", 1))
  83 }
  84
  85 func TestSdks(t *testing.T) {
  86         cpt = 0
  87         InitSdkDir()
  88         var sdkArray []xsapiv1.SDK
  89         assert.Nil(t, HTTPCli.Get("/sdks", &sdkArray))
  90         _, err := os.Stat(path.Join(os.Getenv(envXdtSdk), "sdks_latest.json"))
  91         time.Sleep(1000 * time.Millisecond)
  92         assert.Nil(t, err)
  93
  94         chSdks := make(chan xsapiv1.SDK)
  95         defer close(chSdks)
  96         ConnectSDKStateChange(t, sCli, chSdks)
  97
  98         for i := 0; i < len(sdkArray); i++ {
  99                 re := regexp.MustCompile("^[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+-[0-9a-z]+$")
 100                 assert.True(t, re.MatchString(sdkArray[i].ID))
 101                 assert.Equal(t, sdkArray[0].Status, "Not Installed")
 102
 103                 var sdk xsapiv1.SDK
 104                 assert.Nil(t, HTTPCli.Get("/sdks/"+sdkArray[i].ID, &sdk))
 105                 assert.Equal(t, sdkArray[i], sdk)
 106
 107                 var sdkRes xsapiv1.SDK
 108                 assert.Nil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 109                 assert.Equal(t, sdkRes.Status, "Installing")
 110                 sdkFromEvent := <-chSdks
 111                 assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 112                 assert.Equal(t, sdkFromEvent.Status, "Installing")
 113
 114                 assert.Nil(t, HTTPCli.Post("/sdks/abortinstall", sdk, &sdkRes))
 115                 assert.Equal(t, sdkRes.Status, "Not Installed")
 116                 sdkFromEvent = <-chSdks
 117                 assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 118                 assert.Equal(t, sdkRes.Status, sdkFromEvent.Status)
 119         }
 120 }
 121
 122 func installFakeSdk(t *testing.T, sdkInstall xsapiv1.SDKInstallArgs, chSdks chan xsapiv1.SDK) xsapiv1.SDK {
 123         InitSdkDir()
 124         var sdkRes xsapiv1.SDK
 125         assert.Nil(t, HTTPCli.Post("/sdks", sdkInstall, &sdkRes))
 126         assert.Equal(t, sdkRes.Status, "Installing")
 127         sdkFromEvent := <-chSdks
 128         assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 129         assert.Equal(t, sdkFromEvent.Status, "Installing")
 130         /*waiting for SDKStateChange event from channel*/
 131         sdkFromEvent = <-chSdks
 132         assert.Equal(t, sdkRes.ID, sdkFromEvent.ID)
 133         assert.Equal(t, "Installed", sdkFromEvent.Status)
 134         assert.Equal(t, sdkFromEvent.LastError, "")
 135         return sdkFromEvent
 136 }
 137
 138 func TestInstallFakeSdk(t *testing.T) {
 139         sdk := xsapiv1.SDKInstallArgs{
 140                 ID:       "",
 141                 Filename: sdkFileName,
 142                 Force:    false,
 143         }
 144
 145         chSdks := make(chan xsapiv1.SDK)
 146         defer close(chSdks)
 147         ConnectSDKStateChange(t, sCli, chSdks)
 148         installFakeSdk(t, sdk, chSdks)
 149
 150         /*test force install*/
 151         sdk.Force = true
 152         sdkRes := installFakeSdk(t, sdk, chSdks)
 153         RemoveSdk(t, sdkRes, chSdks)
 154         DisconnectSDKStateChange(t, sCli)
 155 }
 156
 157 func TestSdksError(t *testing.T) {
 158         sdk := xsapiv1.SDKInstallArgs{
 159                 ID:       "",
 160                 Filename: "",
 161                 Force:    false,
 162         }
 163         var sdkRes xsapiv1.SDK
 164         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 165         sdk.ID = "42885c24-374e-3ef0-9723-0c8a05191aa"
 166         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 167         sdk.Filename = "fake"
 168         assert.NotNil(t, HTTPCli.Post("/sdks", sdk, &sdkRes))
 169 }