test/main_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         "io"
  22         "log"
  23         "os"
  24         "os/exec"
  25         "sync"
  26         "testing"
  27         "time"
  28
  29         common "gerrit.automotivelinux.org/gerrit/src/xds/xds-common.git/golib"
  30         "gerrit.automotivelinux.org/gerrit/src/xds/xds-server/lib/xsapiv1"
  31         socketio_client "github.com/sebd71/go-socket.io-client"
  32 )
  33
  34 // IOSockClient
  35 type IOSockClient struct {
  36         URL       string
  37         Conn      *socketio_client.Client
  38         Options   *socketio_client.Options
  39         EmitMutex *sync.Mutex
  40         Connected bool
  41         //ServerDiscoChan chan Disconnection
  42         EscapeKeys []byte
  43 }
  44
  45 //global client
  46 var HTTPCli *common.HTTPClient
  47 var logDir string
  48 var sCli *IOSockClient
  49
  50 func Debug(t *testing.T, args ...interface{}) {
  51         if os.Getenv("VERBOSE") != "" {
  52                 t.Log(args)
  53         }
  54 }
  55
  56 func Debugf(t *testing.T, format string, args ...interface{}) {
  57         if os.Getenv("VERBOSE") != "" {
  58                 t.Logf(format, args)
  59         }
  60 }
  61
  62 func Copy(src, dst string) error {
  63         in, err := os.Open(src)
  64         if err != nil {
  65                 return err
  66         }
  67         defer in.Close()
  68
  69         out, err := os.Create(dst)
  70         if err != nil {
  71                 return err
  72         }
  73         defer out.Close()
  74
  75         _, err = io.Copy(out, in)
  76         if err != nil {
  77                 return err
  78         }
  79         return out.Close()
  80 }
  81
  82 func initEnv(launchProcess bool) {
  83         if launchProcess {
  84                 /*kill xds-server if needed*/
  85                 cmd := exec.Command("killall", "-9", "xds-server")
  86                 if err := cmd.Start(); err != nil {
  87                         log.Fatal(err)
  88                 }
  89                 cmd.Wait()
  90         }
  91         /*set environment variable*/
  92         rootTestLog := "/tmp/xds-server-test"
  93         if err := os.Setenv(envRootCfgDir, rootTestLog); err != nil {
  94                 log.Fatal(err)
  95         }
  96         sdkDir := rootTestLog + "/sdks/"
  97         if err := os.Setenv(envXdtSdk, sdkDir); err != nil {
  98                 log.Fatal(err)
  99         }
 100         if err := os.Setenv(envXdsServerWorkspaceDir, rootTestLog); err != nil {
 101                 log.Fatal(err)
 102         }
 103         if err := os.Setenv(envXdsServerRootCfgDir, rootTestLog); err != nil {
 104                 log.Fatal(err)
 105         }
 106         if err := os.Setenv("XDS_LOG_SILLY", "1"); err != nil {
 107                 log.Fatal(err)
 108         }
 109         /*remove and recreate working directories*/
 110         os.RemoveAll(rootTestLog)
 111         os.MkdirAll(rootTestLog, 0755)
 112         logDir = rootTestLog + "/logs/"
 113         os.MkdirAll(logDir, 0755)
 114 }
 115
 116 /*prepare xds-server process*/
 117 func launchXdsServer(proc **os.Process) *os.File {
 118         logFile := logDir + logFileXdsServer
 119         file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY, 0644)
 120         if err != nil {
 121                 log.Fatal(err)
 122         }
 123         tmpProc, err := os.StartProcess(argsProcess[0], argsProcess, &os.ProcAttr{
 124                 Files: []*os.File{os.Stdin, file, file},
 125         })
 126         if err != nil {
 127                 log.Fatal(err)
 128         }
 129         *proc = tmpProc
 130         return file
 131 }
 132
 133 func getHTTPClient(lvl int) (*common.HTTPClient, *os.File) {
 134         logFile := logDir + logFileClient
 135         file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY, 0644)
 136         if err != nil {
 137                 log.Fatal(err)
 138         }
 139         conf := common.HTTPClientConfig{
 140                 URLPrefix:           "/api/v1",
 141                 HeaderClientKeyName: "Xds-Sid",
 142                 CsrfDisable:         true,
 143                 LogOut:              file,
 144                 LogPrefix:           "XDSSERVERTEST: ",
 145                 LogLevel:            lvl,
 146         }
 147         cli, err := common.HTTPNewClient(prefixURL, conf)
 148         if err != nil {
 149                 log.Fatal(err)
 150         }
 151         log.Printf("HTTP session ID : %v", cli.GetClientID())
 152         var ver xsapiv1.Version
 153         err = cli.Get("/version", &ver)
 154         if err != nil {
 155                 log.Fatal(err)
 156         }
 157         return cli, file
 158 }
 159
 160 func NewIoSocketClient(url, clientID string) (*IOSockClient, error) {
 161         var err error
 162
 163         sCli := &IOSockClient{
 164                 URL:       url,
 165                 EmitMutex: &sync.Mutex{},
 166                 Options: &socketio_client.Options{
 167                         Transport: "websocket",
 168                         Header:    make(map[string][]string),
 169                 },
 170         }
 171         sCli.Options.Header["XDS-SID"] = []string{clientID}
 172
 173         sCli.Conn, err = socketio_client.NewClient(url, sCli.Options)
 174         if err != nil {
 175                 return nil, fmt.Errorf("IO.socket connection error: " + err.Error())
 176         }
 177
 178         sCli.Conn.On("connection", func() {
 179                 sCli.Connected = true
 180         })
 181
 182         sCli.Conn.On("disconnection", func(err error) {
 183                 log.Printf("WS disconnection event with err: %v\n", err)
 184                 sCli.Connected = false
 185         })
 186
 187         log.Printf("Connect websocket with url=%v clientId=%v\n", prefixURL, HTTPCli.GetClientID())
 188         return sCli, nil
 189 }
 190 func TestMain(m *testing.M) {
 191         /* useful for debugging, preventing from launching xds-server
 192          * it can be launch separetly */
 193         launchProcess := true
 194         log.Printf("TestMain: launchProcess is %v, so launching xds-server", launchProcess)
 195         initEnv(launchProcess)
 196
 197         var proc *os.Process
 198         var fileXdsServer *os.File
 199         if launchProcess {
 200                 fileXdsServer = launchXdsServer(&proc)
 201                 go func(p *os.Process) {
 202                         log.Print("xds-server is launching")
 203                         if status, err := p.Wait(); err != nil {
 204                                 log.Fatalf("status=%v\n err=%v\n", status, err)
 205                         }
 206                 }(proc)
 207                 defer fileXdsServer.Close()
 208         }
 209         time.Sleep(1 * time.Second)
 210
 211         lvl := common.HTTPLogLevelDebug
 212         var fileHTTPClient *os.File
 213         HTTPCli, fileHTTPClient = getHTTPClient(lvl)
 214         defer fileHTTPClient.Close()
 215         var err error
 216         sCli, err = NewIoSocketClient(prefixURL, HTTPCli.GetClientID())
 217         if err != nil {
 218                 log.Fatal(err)
 219         }
 220
 221         if HTTPCli == nil {
 222                 log.Fatal("HTTPCLi is nil")
 223         }
 224         res := m.Run()
 225         defer os.Exit(res)
 226         proc.Kill()
 227 }