124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package alibabacloud
|
|
|
|
import (
|
|
"fmt"
|
|
gamekruiseiov1alpha1 "github.com/openkruise/kruise-game/apis/v1alpha1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"reflect"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func TestSlpSpAllocate(t *testing.T) {
|
|
tests := []struct {
|
|
slbsp *SlbSpPlugin
|
|
pod *corev1.Pod
|
|
podNetConfig *lbSpConfig
|
|
numBackends map[string]int
|
|
podSlbId map[string]string
|
|
expErr error
|
|
}{
|
|
{
|
|
slbsp: &SlbSpPlugin{
|
|
numBackends: make(map[string]int),
|
|
podSlbId: make(map[string]string),
|
|
mutex: sync.RWMutex{},
|
|
},
|
|
pod: &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "pod-name",
|
|
Namespace: "pod-ns",
|
|
Labels: map[string]string{
|
|
"xxx": "xxx",
|
|
},
|
|
},
|
|
},
|
|
podNetConfig: &lbSpConfig{
|
|
lbIds: []string{"lb-xxa"},
|
|
ports: []int{80},
|
|
protocols: []corev1.Protocol{corev1.ProtocolTCP},
|
|
},
|
|
numBackends: map[string]int{"lb-xxa": 1},
|
|
podSlbId: map[string]string{"pod-ns/pod-name": "lb-xxa"},
|
|
expErr: nil,
|
|
},
|
|
|
|
{
|
|
slbsp: &SlbSpPlugin{
|
|
numBackends: map[string]int{"lb-xxa": 200},
|
|
podSlbId: map[string]string{"a-ns/a-name": "lb-xxa"},
|
|
mutex: sync.RWMutex{},
|
|
},
|
|
pod: &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "pod-name",
|
|
Namespace: "pod-ns",
|
|
Labels: map[string]string{
|
|
"xxx": "xxx",
|
|
},
|
|
},
|
|
},
|
|
podNetConfig: &lbSpConfig{
|
|
lbIds: []string{"lb-xxa"},
|
|
ports: []int{80},
|
|
protocols: []corev1.Protocol{corev1.ProtocolTCP},
|
|
},
|
|
numBackends: map[string]int{"lb-xxa": 200},
|
|
podSlbId: map[string]string{"a-ns/a-name": "lb-xxa"},
|
|
expErr: fmt.Errorf(ErrorUpperLimit),
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
slbId, err := test.slbsp.getOrAllocate(test.podNetConfig, test.pod)
|
|
if (err == nil) != (test.expErr == nil) {
|
|
t.Errorf("expect err: %v, but acutal err: %v", test.expErr, err)
|
|
}
|
|
|
|
if test.pod.GetLabels()[SlbIdLabelKey] != slbId {
|
|
t.Errorf("expect pod have slblabel value: %s, but actual value: %s", slbId, test.pod.GetLabels()[SlbIdLabelKey])
|
|
}
|
|
|
|
if !reflect.DeepEqual(test.numBackends, test.slbsp.numBackends) {
|
|
t.Errorf("expect numBackends: %v, but actual: %v", test.numBackends, test.slbsp.numBackends)
|
|
}
|
|
|
|
if !reflect.DeepEqual(test.podSlbId, test.slbsp.podSlbId) {
|
|
t.Errorf("expect numBackends: %v, but actual: %v", test.podSlbId, test.slbsp.podSlbId)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseLbSpConfig(t *testing.T) {
|
|
tests := []struct {
|
|
conf []gamekruiseiov1alpha1.NetworkConfParams
|
|
podNetConfig *lbSpConfig
|
|
}{
|
|
{
|
|
conf: []gamekruiseiov1alpha1.NetworkConfParams{
|
|
{
|
|
Name: PortProtocolsConfigName,
|
|
Value: "80",
|
|
},
|
|
{
|
|
Name: SlbIdsConfigName,
|
|
Value: "lb-xxa",
|
|
},
|
|
},
|
|
podNetConfig: &lbSpConfig{
|
|
lbIds: []string{"lb-xxa"},
|
|
ports: []int{80},
|
|
protocols: []corev1.Protocol{corev1.ProtocolTCP},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
podNetConfig := parseLbSpConfig(test.conf)
|
|
if !reflect.DeepEqual(podNetConfig, test.podNetConfig) {
|
|
t.Errorf("expect podNetConfig: %v, but actual: %v", test.podNetConfig, podNetConfig)
|
|
}
|
|
}
|
|
}
|