test: Add test cases for service ref manager and control interface (#615

) * vendor: Update Signed-off-by: Ce Gao <gaoce@caicloud.io> * test: Add Signed-off-by: Ce Gao <gaoce@caicloud.io> * test: Add ref manager test Signed-off-by: Ce Gao <gaoce@caicloud.io>
kubeflow · Jun 1, 2018 · 924e927 · 924e927
1 parent 485d430
commit 924e927
Show file tree

Hide file tree

Showing 408 changed files with 52,329 additions and 6 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/pkg/controller.v2/service_control_test.go b/pkg/controller.v2/service_control_test.go
@@ -0,0 +1,79 @@
+// Copyright 2018 The Kubeflow 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 controller
+
+import (
+	"encoding/json"
+	"net/http/httptest"
+	"testing"
+
+	"k8s.io/api/core/v1"
+	apiequality "k8s.io/apimachinery/pkg/api/equality"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	clientset "k8s.io/client-go/kubernetes"
+	restclient "k8s.io/client-go/rest"
+	"k8s.io/client-go/tools/record"
+	utiltesting "k8s.io/client-go/util/testing"
+	"k8s.io/kubernetes/pkg/api/testapi"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateService(t *testing.T) {
+	ns := metav1.NamespaceDefault
+	body := runtime.EncodeOrDie(testapi.Default.Codec(), &v1.Service{ObjectMeta: metav1.ObjectMeta{Name: "empty_service"}})
+	fakeHandler := utiltesting.FakeHandler{
+		StatusCode:   200,
+		ResponseBody: string(body),
+	}
+	testServer := httptest.NewServer(&fakeHandler)
+	defer testServer.Close()
+	clientset := clientset.NewForConfigOrDie(&restclient.Config{
+		Host: testServer.URL,
+		ContentConfig: restclient.ContentConfig{
+			GroupVersion: &v1.SchemeGroupVersion,
+		},
+	})
+
+	serviceControl := RealServiceControl{
+		KubeClient: clientset,
+		Recorder:   &record.FakeRecorder{},
+	}
+
+	tfJob := newTFJob(1, 0)
+
+	testName := "service-name"
+	service := newBaseService(testName, tfJob, t)
+
+	// Make sure createReplica sends a POST to the apiserver with a pod from the controllers pod template
+	err := serviceControl.CreateServices(ns, service, tfJob)
+	assert.NoError(t, err, "unexpected error: %v", err)
+
+	expectedService := v1.Service{
+		ObjectMeta: metav1.ObjectMeta{
+			Labels:          genLabels(getKey(tfJob, t)),
+			Name:            testName,
+			Namespace:       ns,
+			OwnerReferences: []metav1.OwnerReference{*metav1.NewControllerRef(tfJob, controllerKind)},
+		},
+	}
+	fakeHandler.ValidateRequest(t, testapi.Default.ResourcePath("services", metav1.NamespaceDefault, ""), "POST", nil)
+	var actualService = &v1.Service{}
+	err = json.Unmarshal([]byte(fakeHandler.RequestBody), actualService)
+	assert.NoError(t, err, "unexpected error: %v", err)
+	assert.True(t, apiequality.Semantic.DeepDerivative(&expectedService, actualService),
+		"Body: %s", fakeHandler.RequestBody)
+}