mirror of
https://github.com/qdm12/gluetun.git
synced 2026-07-20 09:26:25 +02:00
Quentin McGaw
31 lines
876 B
Go
31 lines
876 B
Go
package settings
|
|
|
|
import gomock "go.uber.org/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
|
|
}
|