gluetun/internal/configuration/settings/helpers_test.go

package settings

import gomock "github.com/golang/mock/gomock"

type sourceKeyValue struct {
	key   string
	value string
}

func newMockSource(ctrl *gomock.Controller, keyValues []sourceKeyValue) *MockSource {
	source := NewMockSource(ctrl)
	var previousCall *gomock.Call
	for _, keyValue := range keyValues {
		transformedKey := keyValue.key
		keyTransformCall := source.EXPECT().KeyTransform(keyValue.key).Return(transformedKey)
		if previousCall != nil {
			keyTransformCall.After(previousCall)
		}
		isSet := keyValue.value != ""
		previousCall = source.EXPECT().Get(transformedKey).
			Return(keyValue.value, isSet).After(keyTransformCall)
		if isSet {
			previousCall = source.EXPECT().KeyTransform(keyValue.key).
				Return(transformedKey).After(previousCall)
			previousCall = source.EXPECT().String().
				Return("mock source").After(previousCall)
		}
	}
	return source
}