73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package dns
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/qdm12/golibs/command/mock_command"
|
|
"github.com/qdm12/golibs/logging/mock_logging"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/qdm12/private-internet-access-docker/internal/constants"
|
|
)
|
|
|
|
func Test_Start(t *testing.T) {
|
|
t.Parallel()
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
logger := mock_logging.NewMockLogger(mockCtrl)
|
|
logger.EXPECT().Info("%s: starting unbound", logPrefix).Times(1)
|
|
commander := mock_command.NewMockCommander(mockCtrl)
|
|
commander.EXPECT().Start("unbound", "-d", "-c", string(constants.UnboundConf), "-vv").
|
|
Return(nil, nil, nil, nil).Times(1)
|
|
c := &configurator{commander: commander, logger: logger}
|
|
stdout, waitFn, err := c.Start(2)
|
|
assert.Nil(t, stdout)
|
|
assert.Nil(t, waitFn)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_Version(t *testing.T) {
|
|
t.Parallel()
|
|
tests := map[string]struct {
|
|
runOutput string
|
|
runErr error
|
|
version string
|
|
err error
|
|
}{
|
|
"no data": {
|
|
err: fmt.Errorf(`unbound version was not found in ""`),
|
|
},
|
|
"2 lines with version": {
|
|
runOutput: "Version \nVersion 1.0-a hello\n",
|
|
version: "1.0-a",
|
|
},
|
|
"run error": {
|
|
runErr: fmt.Errorf("error"),
|
|
err: fmt.Errorf("unbound version: error"),
|
|
},
|
|
}
|
|
for name, tc := range tests {
|
|
tc := tc
|
|
t.Run(name, func(t *testing.T) {
|
|
t.Parallel()
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
commander := mock_command.NewMockCommander(mockCtrl)
|
|
commander.EXPECT().Run("unbound", "-V").
|
|
Return(tc.runOutput, tc.runErr).Times(1)
|
|
c := &configurator{commander: commander}
|
|
version, err := c.Version()
|
|
if tc.err != nil {
|
|
require.Error(t, err)
|
|
assert.Equal(t, tc.err.Error(), err.Error())
|
|
} else {
|
|
assert.NoError(t, err)
|
|
}
|
|
assert.Equal(t, tc.version, version)
|
|
})
|
|
}
|
|
}
|