127 lines
4.1 KiB
Go
127 lines
4.1 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package envvars_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/kubernetes/pkg/api/v1"
|
|
"k8s.io/kubernetes/pkg/kubelet/envvars"
|
|
)
|
|
|
|
func TestFromServices(t *testing.T) {
|
|
sl := []*v1.Service{
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo-bar"},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"bar": "baz"},
|
|
ClusterIP: "1.2.3.4",
|
|
Ports: []v1.ServicePort{
|
|
{Port: 8080, Protocol: "TCP"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "abc-123"},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"bar": "baz"},
|
|
ClusterIP: "5.6.7.8",
|
|
Ports: []v1.ServicePort{
|
|
{Name: "u-d-p", Port: 8081, Protocol: "UDP"},
|
|
{Name: "t-c-p", Port: 8081, Protocol: "TCP"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "q-u-u-x"},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"bar": "baz"},
|
|
ClusterIP: "9.8.7.6",
|
|
Ports: []v1.ServicePort{
|
|
{Port: 8082, Protocol: "TCP"},
|
|
{Name: "8083", Port: 8083, Protocol: "TCP"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "svrc-clusterip-none"},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"bar": "baz"},
|
|
ClusterIP: "None",
|
|
Ports: []v1.ServicePort{
|
|
{Port: 8082, Protocol: "TCP"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "svrc-clusterip-empty"},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"bar": "baz"},
|
|
ClusterIP: "",
|
|
Ports: []v1.ServicePort{
|
|
{Port: 8082, Protocol: "TCP"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
vars := envvars.FromServices(sl)
|
|
expected := []v1.EnvVar{
|
|
{Name: "FOO_BAR_SERVICE_HOST", Value: "1.2.3.4"},
|
|
{Name: "FOO_BAR_SERVICE_PORT", Value: "8080"},
|
|
{Name: "FOO_BAR_PORT", Value: "tcp://1.2.3.4:8080"},
|
|
{Name: "FOO_BAR_PORT_8080_TCP", Value: "tcp://1.2.3.4:8080"},
|
|
{Name: "FOO_BAR_PORT_8080_TCP_PROTO", Value: "tcp"},
|
|
{Name: "FOO_BAR_PORT_8080_TCP_PORT", Value: "8080"},
|
|
{Name: "FOO_BAR_PORT_8080_TCP_ADDR", Value: "1.2.3.4"},
|
|
{Name: "ABC_123_SERVICE_HOST", Value: "5.6.7.8"},
|
|
{Name: "ABC_123_SERVICE_PORT", Value: "8081"},
|
|
{Name: "ABC_123_SERVICE_PORT_U_D_P", Value: "8081"},
|
|
{Name: "ABC_123_SERVICE_PORT_T_C_P", Value: "8081"},
|
|
{Name: "ABC_123_PORT", Value: "udp://5.6.7.8:8081"},
|
|
{Name: "ABC_123_PORT_8081_UDP", Value: "udp://5.6.7.8:8081"},
|
|
{Name: "ABC_123_PORT_8081_UDP_PROTO", Value: "udp"},
|
|
{Name: "ABC_123_PORT_8081_UDP_PORT", Value: "8081"},
|
|
{Name: "ABC_123_PORT_8081_UDP_ADDR", Value: "5.6.7.8"},
|
|
{Name: "ABC_123_PORT_8081_TCP", Value: "tcp://5.6.7.8:8081"},
|
|
{Name: "ABC_123_PORT_8081_TCP_PROTO", Value: "tcp"},
|
|
{Name: "ABC_123_PORT_8081_TCP_PORT", Value: "8081"},
|
|
{Name: "ABC_123_PORT_8081_TCP_ADDR", Value: "5.6.7.8"},
|
|
{Name: "Q_U_U_X_SERVICE_HOST", Value: "9.8.7.6"},
|
|
{Name: "Q_U_U_X_SERVICE_PORT", Value: "8082"},
|
|
{Name: "Q_U_U_X_SERVICE_PORT_8083", Value: "8083"},
|
|
{Name: "Q_U_U_X_PORT", Value: "tcp://9.8.7.6:8082"},
|
|
{Name: "Q_U_U_X_PORT_8082_TCP", Value: "tcp://9.8.7.6:8082"},
|
|
{Name: "Q_U_U_X_PORT_8082_TCP_PROTO", Value: "tcp"},
|
|
{Name: "Q_U_U_X_PORT_8082_TCP_PORT", Value: "8082"},
|
|
{Name: "Q_U_U_X_PORT_8082_TCP_ADDR", Value: "9.8.7.6"},
|
|
{Name: "Q_U_U_X_PORT_8083_TCP", Value: "tcp://9.8.7.6:8083"},
|
|
{Name: "Q_U_U_X_PORT_8083_TCP_PROTO", Value: "tcp"},
|
|
{Name: "Q_U_U_X_PORT_8083_TCP_PORT", Value: "8083"},
|
|
{Name: "Q_U_U_X_PORT_8083_TCP_ADDR", Value: "9.8.7.6"},
|
|
}
|
|
if len(vars) != len(expected) {
|
|
t.Errorf("Expected %d env vars, got: %+v", len(expected), vars)
|
|
return
|
|
}
|
|
for i := range expected {
|
|
if !reflect.DeepEqual(vars[i], expected[i]) {
|
|
t.Errorf("expected %#v, got %#v", vars[i], expected[i])
|
|
}
|
|
}
|
|
}
|