Huawei Cloud Provider implementation (#241)
* add huawei CCI provider * add readme * add vender * add huawei provider mock test
This commit is contained in:
Fei Xu
212
providers/huawei/cci_test.go
Normal file
212
providers/huawei/cci_test.go
Normal file
@@ -0,0 +1,212 @@
|
||||
package huawei
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/util/uuid"
|
||||
)
|
||||
|
||||
const (
|
||||
fakeAppKey = "Whj8f5RAHsvQahveqCdo"
|
||||
fakeAppSecret = "ymW5JgrdwrIvRS76YxyIqHNXe9s5ocIhaWWvPUhx"
|
||||
fakeRegion = "southchina"
|
||||
fakeService = "default"
|
||||
fakeProject = "vk-project"
|
||||
fakeNodeName = "vk"
|
||||
)
|
||||
|
||||
// TestCreateProject test create project.
|
||||
func TestCreateProject(t *testing.T) {
|
||||
cciServerMocker, provider, err := prepareMocks()
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("Unable to prepare the mocks", err)
|
||||
}
|
||||
|
||||
cciServerMocker.OnCreateProject = func(ns *v1.Namespace) (int, interface{}) {
|
||||
assert.NotNil(t, ns, "Project is nil")
|
||||
assert.Equal(t, fakeProject, ns.Name, "pod.Annotations[\"virtual-kubelet-podname\"] is not expected")
|
||||
return http.StatusOK, &v1.Namespace{
|
||||
TypeMeta: metav1.TypeMeta{
|
||||
Kind: "Namespace",
|
||||
APIVersion: "v1",
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: fakeProject,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
if err := provider.createProject(); err != nil {
|
||||
t.Fatal("Failed to create project", err)
|
||||
}
|
||||
}
|
||||
|
||||
// TestCreatePod test create pod.
|
||||
func TestCreatePod(t *testing.T) {
|
||||
cciServerMocker, provider, err := prepareMocks()
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("Unable to prepare the mocks", err)
|
||||
}
|
||||
podName := "pod-" + string(uuid.NewUUID())
|
||||
podNamespace := "ns-" + string(uuid.NewUUID())
|
||||
|
||||
cciServerMocker.OnCreatePod = func(pod *v1.Pod) (int, interface{}) {
|
||||
assert.NotNil(t, pod, "Pod is nil")
|
||||
assert.NotNil(t, pod.Annotations, "pod.Annotations is expected")
|
||||
assert.Equal(t, podName, pod.Annotations[podAnnotationPodNameKey], "pod.Annotations[\"virtual-kubelet-podname\"] is not expected")
|
||||
assert.Equal(t, podNamespace, pod.Annotations[podAnnotationNamespaceKey], "pod.Annotations[\"virtual-kubelet-namespace\"] is not expected")
|
||||
assert.Equal(t, 1, len(pod.Spec.Containers), "1 Container is expected")
|
||||
assert.Equal(t, "nginx", pod.Spec.Containers[0].Name, "Container nginx is expected")
|
||||
return http.StatusOK, pod
|
||||
}
|
||||
|
||||
pod := &v1.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: podName,
|
||||
Namespace: podNamespace,
|
||||
},
|
||||
Spec: v1.PodSpec{
|
||||
Containers: []v1.Container{
|
||||
v1.Container{
|
||||
Name: "nginx",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
if err := provider.CreatePod(pod); err != nil {
|
||||
t.Fatal("Failed to create pod", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Tests get pod.
|
||||
func TestGetPod(t *testing.T) {
|
||||
cciServerMocker, provider, err := prepareMocks()
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("Unable to prepare the mocks", err)
|
||||
}
|
||||
|
||||
podName := "pod-" + string(uuid.NewUUID())
|
||||
podNamespace := "ns-" + string(uuid.NewUUID())
|
||||
|
||||
cciServerMocker.OnGetPod = func(namespace, name string) (int, interface{}) {
|
||||
annotations := map[string]string{
|
||||
podAnnotationPodNameKey: "podname",
|
||||
podAnnotationNamespaceKey: "podnamespaces",
|
||||
podAnnotationUIDkey: "poduid",
|
||||
podAnnotationClusterNameKey: "podclustername",
|
||||
podAnnotationNodeName: "podnodename",
|
||||
}
|
||||
|
||||
return http.StatusOK, &v1.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: podName,
|
||||
Namespace: podNamespace,
|
||||
Annotations: annotations,
|
||||
},
|
||||
Spec: v1.PodSpec{
|
||||
Containers: []v1.Container{
|
||||
v1.Container{
|
||||
Name: "nginx",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
pod, err := provider.GetPod(podNamespace, podName)
|
||||
if err != nil {
|
||||
t.Fatal("Failed to get pod", err)
|
||||
}
|
||||
|
||||
assert.NotNil(t, pod, "Response pod should not be nil")
|
||||
assert.NotNil(t, pod.Spec.Containers, "Containers should not be nil")
|
||||
assert.Equal(t, pod.Name, "podname", "Pod name is not expected")
|
||||
assert.Equal(t, pod.Namespace, "podnamespaces", "Pod namespace is not expected")
|
||||
assert.Nil(t, pod.Annotations, "Pod Annotations should be nil")
|
||||
assert.Equal(t, string(pod.UID), "poduid", "Pod UID is not expected")
|
||||
assert.Equal(t, pod.ClusterName, "podclustername", "Pod clustername is not expected")
|
||||
assert.Equal(t, pod.Spec.NodeName, "podnodename", "Pod node name is not expected")
|
||||
}
|
||||
|
||||
// Tests get pod.
|
||||
func TestGetPods(t *testing.T) {
|
||||
cciServerMocker, provider, err := prepareMocks()
|
||||
|
||||
if err != nil {
|
||||
t.Fatal("Unable to prepare the mocks", err)
|
||||
}
|
||||
|
||||
podName := "pod-" + string(uuid.NewUUID())
|
||||
podNamespace := "ns-" + string(uuid.NewUUID())
|
||||
|
||||
cciServerMocker.OnGetPods = func() (int, interface{}) {
|
||||
annotations := map[string]string{
|
||||
podAnnotationPodNameKey: "podname",
|
||||
podAnnotationNamespaceKey: "podnamespaces",
|
||||
podAnnotationUIDkey: "poduid",
|
||||
podAnnotationClusterNameKey: "podclustername",
|
||||
podAnnotationNodeName: "podnodename",
|
||||
}
|
||||
|
||||
pod := v1.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: podName,
|
||||
Namespace: podNamespace,
|
||||
Annotations: annotations,
|
||||
},
|
||||
Spec: v1.PodSpec{
|
||||
Containers: []v1.Container{
|
||||
v1.Container{
|
||||
Name: "nginx",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
return http.StatusOK, []v1.Pod{pod}
|
||||
}
|
||||
pods, err := provider.GetPods()
|
||||
if err != nil {
|
||||
t.Fatal("Failed to get pods", err)
|
||||
}
|
||||
|
||||
pod := pods[0]
|
||||
assert.NotNil(t, pod, "Response pod should not be nil")
|
||||
assert.NotNil(t, pod.Spec.Containers, "Containers should not be nil")
|
||||
assert.Equal(t, pod.Name, "podname", "Pod name is not expected")
|
||||
assert.Equal(t, pod.Namespace, "podnamespaces", "Pod namespace is not expected")
|
||||
assert.Nil(t, pod.Annotations, "Pod Annotations should be nil")
|
||||
assert.Equal(t, string(pod.UID), "poduid", "Pod UID is not expected")
|
||||
assert.Equal(t, pod.ClusterName, "podclustername", "Pod clustername is not expected")
|
||||
assert.Equal(t, pod.Spec.NodeName, "podnodename", "Pod node name is not expected")
|
||||
}
|
||||
|
||||
func prepareMocks() (*CCIMock, *CCIProvider, error) {
|
||||
cciServerMocker := NewCCIMock()
|
||||
|
||||
os.Setenv("CCI_APP_KEP", fakeAppKey)
|
||||
os.Setenv("CCI_APP_SECRET", fakeAppSecret)
|
||||
|
||||
defaultApiEndpoint = cciServerMocker.GetServerURL()
|
||||
provider, err := NewCCIProvider("cci.toml", nil, fakeNodeName, "Linux", "0.0.0.0", 10250)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
provider.project = fakeProject
|
||||
provider.client.Signer = &fakeSigner{
|
||||
AppKey: fakeAppKey,
|
||||
AppSecret: fakeAppSecret,
|
||||
Region: fakeRegion,
|
||||
Service: fakeService,
|
||||
}
|
||||
|
||||
return cciServerMocker, provider, nil
|
||||
}
|
||||
Reference in New Issue
Block a user