package shared 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) }