1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
|
package allocator
import (
"testing"
"github.com/docker/swarmkit/api"
"github.com/stretchr/testify/assert"
)
func TestUpdatePortsInHostPublishMode(t *testing.T) {
service := api.Service{
Spec: api.ServiceSpec{
Endpoint: &api.EndpointSpec{
Ports: []*api.PortConfig{
{
Protocol: api.ProtocolTCP,
TargetPort: 80,
PublishedPort: 10000,
PublishMode: api.PublishModeHost,
},
},
},
},
Endpoint: &api.Endpoint{
Ports: []*api.PortConfig{
{
Protocol: api.ProtocolTCP,
TargetPort: 80,
PublishedPort: 15000,
PublishMode: api.PublishModeHost,
},
},
},
}
updatePortsInHostPublishMode(&service)
assert.Equal(t, len(service.Endpoint.Ports), 1)
assert.Equal(t, service.Endpoint.Ports[0].PublishedPort, uint32(10000))
assert.Equal(t, service.Endpoint.Spec.Ports[0].PublishedPort, uint32(10000))
}
|