71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
Var1 = []string{"ONE=1", "TWO=2"}
|
|
)
|
|
|
|
func createTmpFile(content []byte) (string, error) {
|
|
tmpfile, err := ioutil.TempFile(os.TempDir(), "unittest")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if _, err := tmpfile.Write(content); err != nil {
|
|
return "", err
|
|
|
|
}
|
|
if err := tmpfile.Close(); err != nil {
|
|
return "", err
|
|
}
|
|
return tmpfile.Name(), nil
|
|
}
|
|
|
|
func TestValidateSysctl(t *testing.T) {
|
|
strSlice := []string{"net.core.test1=4", "kernel.msgmax=2"}
|
|
result, _ := validateSysctl(strSlice)
|
|
assert.Equal(t, result["net.core.test1"], "4")
|
|
}
|
|
|
|
func TestValidateSysctlBadSysctl(t *testing.T) {
|
|
strSlice := []string{"BLAU=BLUE", "GELB^YELLOW"}
|
|
_, err := validateSysctl(strSlice)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestGetAllLabels(t *testing.T) {
|
|
fileLabels := []string{}
|
|
labels, _ := getAllLabels(fileLabels, Var1)
|
|
assert.Equal(t, len(labels), 2)
|
|
}
|
|
|
|
func TestGetAllLabelsBadKeyValue(t *testing.T) {
|
|
inLabels := []string{"=badValue", "="}
|
|
fileLabels := []string{}
|
|
_, err := getAllLabels(fileLabels, inLabels)
|
|
assert.Error(t, err, assert.AnError)
|
|
}
|
|
|
|
func TestGetAllLabelsBadLabelFile(t *testing.T) {
|
|
fileLabels := []string{"/foobar5001/be"}
|
|
_, err := getAllLabels(fileLabels, Var1)
|
|
assert.Error(t, err, assert.AnError)
|
|
}
|
|
|
|
func TestGetAllLabelsFile(t *testing.T) {
|
|
content := []byte("THREE=3")
|
|
tFile, err := createTmpFile(content)
|
|
defer os.Remove(tFile)
|
|
assert.NoError(t, err)
|
|
fileLabels := []string{tFile}
|
|
result, _ := getAllLabels(fileLabels, Var1)
|
|
assert.Equal(t, len(result), 3)
|
|
}
|