default_test.go

package config

import (
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func prepareProbeBinary(t *testing.T, expectedOutput string) (path string) {
	f, err := ioutil.TempFile("", "conmon-")
	require.Nil(t, err)
	defer func() { require.Nil(t, f.Close()) }()

	err = f.Chmod(os.FileMode(0o755))
	require.Nil(t, err)

	_, err = f.WriteString(fmt.Sprintf("#!/usr/bin/env sh\necho %q", expectedOutput))
	require.Nil(t, err)

	return f.Name()
}

func TestProbeConmon(t *testing.T) {
	t.Parallel()

	for _, tc := range []struct {
		msg    string
		output string
		assert func(error, string)
	}{
		{
			msg:    "success conmon 2",
			output: "conmon version 2.1.0",
			assert: func(err error, msg string) {
				assert.Nil(t, err, msg)
			},
		},
		{
			msg:    "success conmon 3",
			output: "conmon version 3.0.0-dev",
			assert: func(x error, msg string) {
				assert.Nil(t, x, msg)
			},
		},
		{
			msg:    "failure outdated version",
			output: "conmon version 1.0.0",
			assert: func(err error, msg string) {
				assert.Equal(t, err, ErrConmonOutdated, msg)
			},
		},
		{
			msg:    "failure invalid format",
			output: "invalid",
			assert: func(err error, msg string) {
				expectedErr := fmt.Errorf(_conmonVersionFormatErr, errors.New("invalid version format"))
				assert.Equal(t, err, expectedErr, msg)
			},
		},
	} {
		filePath := prepareProbeBinary(t, tc.output)
		defer os.RemoveAll(filePath)
		err := probeConmon(filePath)
		tc.assert(err, tc.msg)
	}
}

func TestMachineVolumes(t *testing.T) {
	t.Parallel()

	os.Setenv("env1", "/test1")
	os.Setenv("env2", "/test2")
	for _, tc := range []struct {
		msg     string
		volumes []string
		output  []string
		assert  func(err error, in, out []string, msg string)
	}{
		{
			volumes: []string{},
			output:  []string{},
			assert: func(err error, in, out []string, msg string) {
				assert.Equal(t, in, out)
				assert.Nil(t, err, msg)
			},
		},
		{
			volumes: []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "$env1:/env1", "$env1:$env1", "$env2:$env1"},
			output:  []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "/test1:/env1", "/test1:/test1", "/test2:/test1"},
			assert: func(err error, in, out []string, msg string) {
				assert.Equal(t, in, out)
				assert.Nil(t, err, msg)
			},
		},
		{
			volumes: []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "$env1:/env1", "$env1:$env1", "$env3:$env1"},
			output:  []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "/test1:/env1", "/test1:/test1", "/test2:/test1"},
			assert: func(err error, in, out []string, msg string) {
				assert.EqualError(t, err, "invalid machine volume $env3:$env1, fields must container data", msg)
			},
		},
		{
			volumes: []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "$env1:/env1", "$env1:$env4", "$env1:$env1"},
			output:  []string{"/foobar:/foobar", "/foobar1:/foobardest:ro", "/test1:/env1", "/test1:/test1", "/test2:/test1"},
			assert: func(err error, in, out []string, msg string) {
				assert.EqualError(t, err, "invalid machine volume $env1:$env4, fields must container data", msg)
			},
		},
		{
			msg:     "This is broken",
			volumes: []string{"/foobar:/foobar:ro:badopt"},
			output:  []string{"/foobar:/foobar:ro"},
			assert: func(err error, in, out []string, msg string) {
				assert.EqualError(t, err, "invalid machine volume /foobar:/foobar:ro:badopt, 2 or 3 fields required", msg)
			},
		},
		{
			msg:     "This is broken2",
			volumes: []string{"/foobar"},
			output:  []string{"/foobar:/foobar:ro"},
			assert: func(err error, in, out []string, msg string) {
				assert.EqualError(t, err, "invalid machine volume /foobar, 2 or 3 fields required", msg)
			},
		},
	} {
		output, err := machineVolumes(tc.volumes)
		tc.assert(err, tc.output, output, tc.msg)
	}
}