test/usecases/oruclosedlooprecovery/goversion/internal/repository/csvhelp_test.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2021: Nordix Foundation
   6 //   %%
   7 //   Licensed under the Apache License, Version 2.0 (the "License");
   8 //   you may not use this file except in compliance with the License.
   9 //   You may obtain a copy of the License at
  10 //
  11 //        http://www.apache.org/licenses/LICENSE-2.0
  12 //
  13 //   Unless required by applicable law or agreed to in writing, software
  14 //   distributed under the License is distributed on an "AS IS" BASIS,
  15 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 //   See the License for the specific language governing permissions and
  17 //   limitations under the License.
  18 //   ========================LICENSE_END===================================
  19 //
  20
  21 package repository
  22
  23 import (
  24         "os"
  25         "reflect"
  26         "testing"
  27 )
  28
  29 func TestCsvFileHelperImpl_GetCsvFromFile(t *testing.T) {
  30         filePath := createTempCsvFile()
  31         defer os.Remove(filePath)
  32         type args struct {
  33                 name string
  34         }
  35         tests := []struct {
  36                 name       string
  37                 fileHelper *CsvFileHelperImpl
  38                 args       args
  39                 want       [][]string
  40                 wantErr    bool
  41         }{
  42                 {
  43                         name:       "Read from file should return array of content",
  44                         fileHelper: &CsvFileHelperImpl{},
  45                         args: args{
  46                                 name: filePath,
  47                         },
  48                         want:    [][]string{{"O-RU-ID", "O-DU-ID"}},
  49                         wantErr: false,
  50                 },
  51                 {
  52                         name:       "File missing should return error",
  53                         fileHelper: &CsvFileHelperImpl{},
  54                         args: args{
  55                                 name: "nofile.csv",
  56                         },
  57                         want:    nil,
  58                         wantErr: true,
  59                 },
  60         }
  61         for _, tt := range tests {
  62                 t.Run(tt.name, func(t *testing.T) {
  63                         h := &CsvFileHelperImpl{}
  64                         got, err := h.GetCsvFromFile(tt.args.name)
  65                         if (err != nil) != tt.wantErr {
  66                                 t.Errorf("CsvFileHelperImpl.GetCsvFromFile() error = %v, wantErr %v", err, tt.wantErr)
  67                                 return
  68                         }
  69                         if !reflect.DeepEqual(got, tt.want) {
  70                                 t.Errorf("CsvFileHelperImpl.GetCsvFromFile() = %v, want %v", got, tt.want)
  71                         }
  72                 })
  73         }
  74 }
  75
  76 func createTempCsvFile() string {
  77         csvFile, _ := os.CreateTemp("", "test*.csv")
  78         filePath := csvFile.Name()
  79         csvFile.Write([]byte("O-RU-ID,O-DU-ID"))
  80         csvFile.Close()
  81         return filePath
  82 }