mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			376 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			376 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
| Copyright 2015 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 app
 | |
| 
 | |
| import (
 | |
| 	"encoding/json"
 | |
| 	"fmt"
 | |
| 	"io/ioutil"
 | |
| 	"net/http"
 | |
| 	"os"
 | |
| 	"testing"
 | |
| 	"time"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 
 | |
| 	fed_v1b1 "k8s.io/kubernetes/federation/apis/federation/v1beta1"
 | |
| 	"k8s.io/kubernetes/federation/cmd/federation-apiserver/app"
 | |
| 	"k8s.io/kubernetes/federation/cmd/federation-apiserver/app/options"
 | |
| 	"k8s.io/kubernetes/pkg/api/v1"
 | |
| 	batch_v1 "k8s.io/kubernetes/pkg/apis/batch/v1"
 | |
| 	ext_v1b1 "k8s.io/kubernetes/pkg/apis/extensions/v1beta1"
 | |
| 	metav1 "k8s.io/kubernetes/pkg/apis/meta/v1"
 | |
| 	"k8s.io/kubernetes/pkg/runtime/schema"
 | |
| )
 | |
| 
 | |
| var securePort = 6443 + 2
 | |
| var insecurePort = 8080 + 2
 | |
| var serverIP = fmt.Sprintf("http://localhost:%v", insecurePort)
 | |
| var groupVersions = []schema.GroupVersion{
 | |
| 	fed_v1b1.SchemeGroupVersion,
 | |
| 	ext_v1b1.SchemeGroupVersion,
 | |
| 	batch_v1.SchemeGroupVersion,
 | |
| }
 | |
| 
 | |
| func TestRun(t *testing.T) {
 | |
| 	certDir, err := ioutil.TempDir("", "")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Failed to create temporary certificate directory: %v", err)
 | |
| 	}
 | |
| 	defer os.RemoveAll(certDir)
 | |
| 
 | |
| 	s := options.NewServerRunOptions()
 | |
| 	s.SecureServing.ServingOptions.BindPort = securePort
 | |
| 	s.InsecureServing.BindPort = insecurePort
 | |
| 	s.Etcd.StorageConfig.ServerList = []string{"http://localhost:2379"}
 | |
| 	s.SecureServing.ServerCert.CertDirectory = certDir
 | |
| 
 | |
| 	go func() {
 | |
| 		if err := app.Run(s); err != nil {
 | |
| 			t.Fatalf("Error in bringing up the server: %v", err)
 | |
| 		}
 | |
| 	}()
 | |
| 	if err := waitForApiserverUp(); err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	testSwaggerSpec(t)
 | |
| 	testSupport(t)
 | |
| 	testAPIGroupList(t)
 | |
| 	testAPIGroup(t)
 | |
| 	testAPIResourceList(t)
 | |
| }
 | |
| 
 | |
| func waitForApiserverUp() error {
 | |
| 	for start := time.Now(); time.Since(start) < time.Minute; time.Sleep(5 * time.Second) {
 | |
| 		_, err := http.Get(serverIP)
 | |
| 		if err == nil {
 | |
| 			return nil
 | |
| 		}
 | |
| 	}
 | |
| 	return fmt.Errorf("waiting for apiserver timed out")
 | |
| }
 | |
| 
 | |
| func readResponse(serverURL string) ([]byte, error) {
 | |
| 	response, err := http.Get(serverURL)
 | |
| 	if err != nil {
 | |
| 		return nil, fmt.Errorf("Error in fetching %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	defer response.Body.Close()
 | |
| 	if response.StatusCode != http.StatusOK {
 | |
| 		return nil, fmt.Errorf("unexpected status: %d for URL: %s, expected status: %d", response.StatusCode, serverURL, http.StatusOK)
 | |
| 	}
 | |
| 	contents, err := ioutil.ReadAll(response.Body)
 | |
| 	if err != nil {
 | |
| 		return nil, fmt.Errorf("Error reading response from %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	return contents, nil
 | |
| }
 | |
| 
 | |
| func testSwaggerSpec(t *testing.T) {
 | |
| 	serverURL := serverIP + "/swaggerapi"
 | |
| 	_, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testSupport(t *testing.T) {
 | |
| 	serverURL := serverIP + "/version"
 | |
| 	_, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func findGroup(groups []metav1.APIGroup, groupName string) *metav1.APIGroup {
 | |
| 	for _, group := range groups {
 | |
| 		if group.Name == groupName {
 | |
| 			return &group
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func testAPIGroupList(t *testing.T) {
 | |
| 	groupVersionForDiscoveryMap := make(map[string]metav1.GroupVersionForDiscovery)
 | |
| 	for _, groupVersion := range groupVersions {
 | |
| 		groupVersionForDiscoveryMap[groupVersion.Group] = metav1.GroupVersionForDiscovery{
 | |
| 			GroupVersion: groupVersion.String(),
 | |
| 			Version:      groupVersion.Version,
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	serverURL := serverIP + "/apis"
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiGroupList metav1.APIGroupList
 | |
| 	err = json.Unmarshal(contents, &apiGroupList)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 
 | |
| 	for _, groupVersion := range groupVersions {
 | |
| 		found := findGroup(apiGroupList.Groups, groupVersion.Group)
 | |
| 		assert.NotNil(t, found)
 | |
| 		assert.Equal(t, groupVersion.Group, found.Name)
 | |
| 		assert.Equal(t, 1, len(found.Versions))
 | |
| 		groupVersionForDiscovery := groupVersionForDiscoveryMap[groupVersion.Group]
 | |
| 		assert.Equal(t, groupVersionForDiscovery, found.Versions[0])
 | |
| 		assert.Equal(t, groupVersionForDiscovery, found.PreferredVersion)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testAPIGroup(t *testing.T) {
 | |
| 	for _, groupVersion := range groupVersions {
 | |
| 		serverURL := serverIP + "/apis/" + groupVersion.Group
 | |
| 		contents, err := readResponse(serverURL)
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("%v", err)
 | |
| 		}
 | |
| 		var apiGroup metav1.APIGroup
 | |
| 		err = json.Unmarshal(contents, &apiGroup)
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 		}
 | |
| 		// empty APIVersion for extensions group
 | |
| 		if groupVersion.Group == "extensions" {
 | |
| 			assert.Equal(t, "", apiGroup.APIVersion)
 | |
| 		} else {
 | |
| 			assert.Equal(t, "v1", apiGroup.APIVersion)
 | |
| 		}
 | |
| 		assert.Equal(t, apiGroup.Name, groupVersion.Group)
 | |
| 		assert.Equal(t, 1, len(apiGroup.Versions))
 | |
| 		assert.Equal(t, groupVersion.String(), apiGroup.Versions[0].GroupVersion)
 | |
| 		assert.Equal(t, groupVersion.Version, apiGroup.Versions[0].Version)
 | |
| 		assert.Equal(t, apiGroup.PreferredVersion, apiGroup.Versions[0])
 | |
| 	}
 | |
| 
 | |
| 	testCoreAPIGroup(t)
 | |
| }
 | |
| 
 | |
| func testCoreAPIGroup(t *testing.T) {
 | |
| 	serverURL := serverIP + "/api"
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiVersions metav1.APIVersions
 | |
| 	err = json.Unmarshal(contents, &apiVersions)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	assert.Equal(t, 1, len(apiVersions.Versions))
 | |
| 	assert.Equal(t, "v1", apiVersions.Versions[0])
 | |
| 	assert.NotEmpty(t, apiVersions.ServerAddressByClientCIDRs)
 | |
| }
 | |
| 
 | |
| func findResource(resources []metav1.APIResource, resourceName string) *metav1.APIResource {
 | |
| 	for _, resource := range resources {
 | |
| 		if resource.Name == resourceName {
 | |
| 			return &resource
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func testAPIResourceList(t *testing.T) {
 | |
| 	testFederationResourceList(t)
 | |
| 	testCoreResourceList(t)
 | |
| 	testExtensionsResourceList(t)
 | |
| 	testBatchResourceList(t)
 | |
| }
 | |
| 
 | |
| func testFederationResourceList(t *testing.T) {
 | |
| 	serverURL := serverIP + "/apis/" + fed_v1b1.SchemeGroupVersion.String()
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiResourceList metav1.APIResourceList
 | |
| 	err = json.Unmarshal(contents, &apiResourceList)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	assert.Equal(t, "v1", apiResourceList.APIVersion)
 | |
| 	assert.Equal(t, fed_v1b1.SchemeGroupVersion.String(), apiResourceList.GroupVersion)
 | |
| 	// Assert that there are exactly 2 resources.
 | |
| 	assert.Equal(t, 2, len(apiResourceList.APIResources))
 | |
| 
 | |
| 	found := findResource(apiResourceList.APIResources, "clusters")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.False(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "clusters/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.False(t, found.Namespaced)
 | |
| }
 | |
| 
 | |
| func testCoreResourceList(t *testing.T) {
 | |
| 	serverURL := serverIP + "/api/" + v1.SchemeGroupVersion.String()
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiResourceList metav1.APIResourceList
 | |
| 	err = json.Unmarshal(contents, &apiResourceList)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	assert.Equal(t, "", apiResourceList.APIVersion)
 | |
| 	assert.Equal(t, v1.SchemeGroupVersion.String(), apiResourceList.GroupVersion)
 | |
| 	// Assert that there are exactly 7 resources.
 | |
| 	assert.Equal(t, 8, len(apiResourceList.APIResources))
 | |
| 
 | |
| 	// Verify services.
 | |
| 	found := findResource(apiResourceList.APIResources, "services")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "services/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify namespaces.
 | |
| 	found = findResource(apiResourceList.APIResources, "namespaces")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.False(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "namespaces/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.False(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "namespaces/finalize")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.False(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify events.
 | |
| 	found = findResource(apiResourceList.APIResources, "events")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify secrets.
 | |
| 	found = findResource(apiResourceList.APIResources, "secrets")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify config maps.
 | |
| 	found = findResource(apiResourceList.APIResources, "configmaps")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| }
 | |
| 
 | |
| func testExtensionsResourceList(t *testing.T) {
 | |
| 	serverURL := serverIP + "/apis/" + ext_v1b1.SchemeGroupVersion.String()
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiResourceList metav1.APIResourceList
 | |
| 	err = json.Unmarshal(contents, &apiResourceList)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	// empty APIVersion for extensions group
 | |
| 	assert.Equal(t, "", apiResourceList.APIVersion)
 | |
| 	assert.Equal(t, ext_v1b1.SchemeGroupVersion.String(), apiResourceList.GroupVersion)
 | |
| 	// Assert that there are exactly 11 resources.
 | |
| 	assert.Equal(t, 11, len(apiResourceList.APIResources))
 | |
| 
 | |
| 	// Verify replicasets.
 | |
| 	found := findResource(apiResourceList.APIResources, "replicasets")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "replicasets/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "replicasets/scale")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify ingress.
 | |
| 	found = findResource(apiResourceList.APIResources, "ingresses")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "ingresses/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify daemonsets.
 | |
| 	found = findResource(apiResourceList.APIResources, "daemonsets")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "daemonsets/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 
 | |
| 	// Verify deployments.
 | |
| 	found = findResource(apiResourceList.APIResources, "deployments")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "deployments/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "deployments/scale")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "deployments/rollback")
 | |
| }
 | |
| 
 | |
| func testBatchResourceList(t *testing.T) {
 | |
| 	serverURL := serverIP + "/apis/" + batch_v1.SchemeGroupVersion.String()
 | |
| 	contents, err := readResponse(serverURL)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("%v", err)
 | |
| 	}
 | |
| 	var apiResourceList metav1.APIResourceList
 | |
| 	err = json.Unmarshal(contents, &apiResourceList)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in unmarshalling response from server %s: %v", serverURL, err)
 | |
| 	}
 | |
| 	// empty APIVersion for extensions group
 | |
| 	assert.Equal(t, "v1", apiResourceList.APIVersion)
 | |
| 	assert.Equal(t, batch_v1.SchemeGroupVersion.String(), apiResourceList.GroupVersion)
 | |
| 	// Assert that there are exactly this number of resources.
 | |
| 	assert.Equal(t, 2, len(apiResourceList.APIResources))
 | |
| 
 | |
| 	// Verify jobs
 | |
| 	found := findResource(apiResourceList.APIResources, "jobs")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| 	found = findResource(apiResourceList.APIResources, "jobs/status")
 | |
| 	assert.NotNil(t, found)
 | |
| 	assert.True(t, found.Namespaced)
 | |
| }
 |