89 строки
1.9 KiB
Go
89 строки
1.9 KiB
Go
package platform
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-container-networking/cni/log"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
exitCode := m.Run()
|
|
os.Exit(exitCode)
|
|
}
|
|
|
|
func TestGetLastRebootTime(t *testing.T) {
|
|
logger := log.CNILogger.With(zap.String("component", "platform"))
|
|
p := NewExecClient(logger)
|
|
_, err := p.GetLastRebootTime()
|
|
if err != nil {
|
|
t.Errorf("GetLastRebootTime failed :%v", err)
|
|
}
|
|
}
|
|
|
|
func TestGetOSDetails(t *testing.T) {
|
|
_, err := GetOSDetails()
|
|
if err != nil {
|
|
t.Errorf("GetOSDetails failed :%v", err)
|
|
}
|
|
}
|
|
|
|
func TestGetProcessNameByID(t *testing.T) {
|
|
pName, err := GetProcessNameByID(strconv.Itoa(os.Getpid()))
|
|
if err != nil {
|
|
t.Errorf("GetProcessNameByID failed: %v", err)
|
|
}
|
|
|
|
if !strings.Contains(pName, "platform.test") {
|
|
t.Errorf("Incorrect process name:%v\n", pName)
|
|
}
|
|
}
|
|
|
|
func TestReadFileByLines(t *testing.T) {
|
|
lines, err := ReadFileByLines("testfiles/test1")
|
|
if err != nil {
|
|
t.Errorf("ReadFileByLines failed: %v", err)
|
|
}
|
|
|
|
if len(lines) != 2 {
|
|
t.Errorf("Line count %d didn't match expected count", len(lines))
|
|
}
|
|
|
|
lines, err = ReadFileByLines("testfiles/test2")
|
|
if err != nil {
|
|
t.Errorf("ReadFileByLines failed: %v", err)
|
|
}
|
|
|
|
if len(lines) != 1 {
|
|
t.Errorf("Line count %d didn't match expected count", len(lines))
|
|
}
|
|
|
|
lines, err = ReadFileByLines("testfiles/test3")
|
|
if err != nil {
|
|
t.Errorf("ReadFileByLines failed: %v", err)
|
|
}
|
|
|
|
if len(lines) != 2 {
|
|
t.Errorf("Line count %d didn't match expected count", len(lines))
|
|
}
|
|
|
|
if lines[1] != "" {
|
|
t.Errorf("Expected empty line but got %s", lines[1])
|
|
}
|
|
}
|
|
|
|
func TestFileExists(t *testing.T) {
|
|
isExist, err := CheckIfFileExists("testfiles/test1")
|
|
if err != nil || !isExist {
|
|
t.Errorf("Returned file not found %v", err)
|
|
}
|
|
|
|
isExist, err = CheckIfFileExists("testfiles/filenotfound")
|
|
if err != nil || isExist {
|
|
t.Errorf("Returned file found")
|
|
}
|
|
}
|