mirror of https://github.com/portainer/portainer
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
|
package docker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/docker/docker/api/types/network"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestApplyVersionConstraint(t *testing.T) {
|
||
|
|
initialNet := network.NetworkingConfig{
|
||
|
|
EndpointsConfig: map[string]*network.EndpointSettings{
|
||
|
|
"key1": {
|
||
|
|
MacAddress: "mac1",
|
||
|
|
EndpointID: "endpointID1",
|
||
|
|
},
|
||
|
|
"key2": {
|
||
|
|
MacAddress: "mac2",
|
||
|
|
EndpointID: "endpointID2",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
f := func(currentVer string, constraint string, success, emptyMac bool) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
transformedNet, err := applyVersionConstraint(currentVer, constraint, initialNet, clearMacAddrs)
|
||
|
|
if success {
|
||
|
|
require.NoError(t, err)
|
||
|
|
} else {
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Len(t, transformedNet.EndpointsConfig, len(initialNet.EndpointsConfig))
|
||
|
|
|
||
|
|
for k := range initialNet.EndpointsConfig {
|
||
|
|
if emptyMac {
|
||
|
|
require.NotEqual(t, initialNet.EndpointsConfig[k], transformedNet.EndpointsConfig[k])
|
||
|
|
require.Empty(t, transformedNet.EndpointsConfig[k].MacAddress)
|
||
|
|
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, initialNet.EndpointsConfig[k], transformedNet.EndpointsConfig[k])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
f("1.45", "< 1.44", true, false) // No transformation
|
||
|
|
f("1.43", "< 1.44", true, true) // Transformation
|
||
|
|
f("a.b.", "< 1.44", true, false) // Invalid current version
|
||
|
|
f("1.45", "z 1.44", false, false) // Invalid version constraint
|
||
|
|
}
|