mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			921 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			921 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
| Copyright 2014 The Kubernetes Authors All rights reserved.
 | |
| 
 | |
| 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 master
 | |
| 
 | |
| import (
 | |
| 	"bytes"
 | |
| 	"crypto/tls"
 | |
| 	"encoding/json"
 | |
| 	"fmt"
 | |
| 	"io/ioutil"
 | |
| 	"net"
 | |
| 	"net/http"
 | |
| 	"net/http/httptest"
 | |
| 	"reflect"
 | |
| 	"strings"
 | |
| 	"testing"
 | |
| 
 | |
| 	"k8s.io/kubernetes/pkg/api"
 | |
| 	"k8s.io/kubernetes/pkg/api/testapi"
 | |
| 	"k8s.io/kubernetes/pkg/api/unversioned"
 | |
| 	utilnet "k8s.io/kubernetes/pkg/util/net"
 | |
| 
 | |
| 	"k8s.io/kubernetes/pkg/apis/autoscaling"
 | |
| 	"k8s.io/kubernetes/pkg/apis/batch"
 | |
| 	"k8s.io/kubernetes/pkg/apis/extensions"
 | |
| 	"k8s.io/kubernetes/pkg/genericapiserver"
 | |
| 	"k8s.io/kubernetes/pkg/kubelet/client"
 | |
| 	"k8s.io/kubernetes/pkg/registry/endpoint"
 | |
| 	"k8s.io/kubernetes/pkg/registry/namespace"
 | |
| 	"k8s.io/kubernetes/pkg/registry/registrytest"
 | |
| 	"k8s.io/kubernetes/pkg/runtime"
 | |
| 	"k8s.io/kubernetes/pkg/storage"
 | |
| 	etcdstorage "k8s.io/kubernetes/pkg/storage/etcd"
 | |
| 	"k8s.io/kubernetes/pkg/storage/etcd/etcdtest"
 | |
| 	etcdtesting "k8s.io/kubernetes/pkg/storage/etcd/testing"
 | |
| 	"k8s.io/kubernetes/pkg/util/intstr"
 | |
| 
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 	"golang.org/x/net/context"
 | |
| )
 | |
| 
 | |
| // setUp is a convience function for setting up for (most) tests.
 | |
| func setUp(t *testing.T) (Master, *etcdtesting.EtcdTestServer, Config, *assert.Assertions) {
 | |
| 	server := etcdtesting.NewEtcdTestClientServer(t)
 | |
| 
 | |
| 	master := Master{
 | |
| 		GenericAPIServer: &genericapiserver.GenericAPIServer{},
 | |
| 	}
 | |
| 	config := Config{
 | |
| 		Config: &genericapiserver.Config{},
 | |
| 	}
 | |
| 	storageVersions := make(map[string]string)
 | |
| 	storageDestinations := genericapiserver.NewStorageDestinations()
 | |
| 	storageDestinations.AddAPIGroup(
 | |
| 		api.GroupName, etcdstorage.NewEtcdStorage(server.Client, testapi.Default.Codec(), etcdtest.PathPrefix(), false))
 | |
| 	storageDestinations.AddAPIGroup(
 | |
| 		autoscaling.GroupName, etcdstorage.NewEtcdStorage(server.Client, testapi.Autoscaling.Codec(), etcdtest.PathPrefix(), false))
 | |
| 	storageDestinations.AddAPIGroup(
 | |
| 		batch.GroupName, etcdstorage.NewEtcdStorage(server.Client, testapi.Batch.Codec(), etcdtest.PathPrefix(), false))
 | |
| 	storageDestinations.AddAPIGroup(
 | |
| 		extensions.GroupName, etcdstorage.NewEtcdStorage(server.Client, testapi.Extensions.Codec(), etcdtest.PathPrefix(), false))
 | |
| 
 | |
| 	config.StorageDestinations = storageDestinations
 | |
| 	storageVersions[api.GroupName] = testapi.Default.GroupVersion().String()
 | |
| 	storageVersions[autoscaling.GroupName] = testapi.Autoscaling.GroupVersion().String()
 | |
| 	storageVersions[batch.GroupName] = testapi.Batch.GroupVersion().String()
 | |
| 	storageVersions[extensions.GroupName] = testapi.Extensions.GroupVersion().String()
 | |
| 	config.StorageVersions = storageVersions
 | |
| 	config.PublicAddress = net.ParseIP("192.168.10.4")
 | |
| 	master.nodeRegistry = registrytest.NewNodeRegistry([]string{"node1", "node2"}, api.NodeResources{})
 | |
| 
 | |
| 	return master, server, config, assert.New(t)
 | |
| }
 | |
| 
 | |
| func newMaster(t *testing.T) (*Master, *etcdtesting.EtcdTestServer, Config, *assert.Assertions) {
 | |
| 	_, etcdserver, config, assert := setUp(t)
 | |
| 
 | |
| 	config.Serializer = api.Codecs
 | |
| 	config.KubeletClient = client.FakeKubeletClient{}
 | |
| 	config.APIPrefix = "/api"
 | |
| 	config.APIGroupPrefix = "/apis"
 | |
| 
 | |
| 	config.ProxyDialer = func(network, addr string) (net.Conn, error) { return nil, nil }
 | |
| 	config.ProxyTLSClientConfig = &tls.Config{}
 | |
| 
 | |
| 	master, err := New(&config)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Error in bringing up the master: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	return master, etcdserver, config, assert
 | |
| }
 | |
| 
 | |
| // TestNew verifies that the New function returns a Master
 | |
| // using the configuration properly.
 | |
| func TestNew(t *testing.T) {
 | |
| 	master, etcdserver, config, assert := newMaster(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	// Verify many of the variables match their config counterparts
 | |
| 	assert.Equal(master.enableCoreControllers, config.EnableCoreControllers)
 | |
| 	assert.Equal(master.tunneler, config.Tunneler)
 | |
| 	assert.Equal(master.APIPrefix, config.APIPrefix)
 | |
| 	assert.Equal(master.APIGroupPrefix, config.APIGroupPrefix)
 | |
| 	assert.Equal(master.ApiGroupVersionOverrides, config.APIGroupVersionOverrides)
 | |
| 	assert.Equal(master.RequestContextMapper, config.RequestContextMapper)
 | |
| 	assert.Equal(master.MasterCount, config.MasterCount)
 | |
| 	assert.Equal(master.ClusterIP, config.PublicAddress)
 | |
| 	assert.Equal(master.PublicReadWritePort, config.ReadWritePort)
 | |
| 	assert.Equal(master.ServiceReadWriteIP, config.ServiceReadWriteIP)
 | |
| 
 | |
| 	// These functions should point to the same memory location
 | |
| 	masterDialer, _ := utilnet.Dialer(master.ProxyTransport)
 | |
| 	masterDialerFunc := fmt.Sprintf("%p", masterDialer)
 | |
| 	configDialerFunc := fmt.Sprintf("%p", config.ProxyDialer)
 | |
| 	assert.Equal(masterDialerFunc, configDialerFunc)
 | |
| 
 | |
| 	assert.Equal(master.ProxyTransport.(*http.Transport).TLSClientConfig, config.ProxyTLSClientConfig)
 | |
| }
 | |
| 
 | |
| // TestGetServersToValidate verifies the unexported getServersToValidate function
 | |
| func TestGetServersToValidate(t *testing.T) {
 | |
| 	master, etcdserver, config, assert := setUp(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	servers := master.getServersToValidate(&config)
 | |
| 
 | |
| 	// Expected servers to validate: scheduler, controller-manager and etcd.
 | |
| 	assert.Equal(3, len(servers), "unexpected server list: %#v", servers)
 | |
| 
 | |
| 	for _, server := range []string{"scheduler", "controller-manager", "etcd-0"} {
 | |
| 		if _, ok := servers[server]; !ok {
 | |
| 			t.Errorf("server list missing: %s", server)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // TestFindExternalAddress verifies both pass and fail cases for the unexported
 | |
| // findExternalAddress function
 | |
| func TestFindExternalAddress(t *testing.T) {
 | |
| 	assert := assert.New(t)
 | |
| 	expectedIP := "172.0.0.1"
 | |
| 
 | |
| 	nodes := []*api.Node{new(api.Node), new(api.Node), new(api.Node)}
 | |
| 	nodes[0].Status.Addresses = []api.NodeAddress{{"ExternalIP", expectedIP}}
 | |
| 	nodes[1].Status.Addresses = []api.NodeAddress{{"LegacyHostIP", expectedIP}}
 | |
| 	nodes[2].Status.Addresses = []api.NodeAddress{{"ExternalIP", expectedIP}, {"LegacyHostIP", "172.0.0.2"}}
 | |
| 
 | |
| 	// Pass Case
 | |
| 	for _, node := range nodes {
 | |
| 		ip, err := findExternalAddress(node)
 | |
| 		assert.NoError(err, "error getting node external address")
 | |
| 		assert.Equal(expectedIP, ip, "expected ip to be %s, but was %s", expectedIP, ip)
 | |
| 	}
 | |
| 
 | |
| 	// Fail case
 | |
| 	_, err := findExternalAddress(new(api.Node))
 | |
| 	assert.Error(err, "expected findExternalAddress to fail on a node with missing ip information")
 | |
| }
 | |
| 
 | |
| // TestNewBootstrapController verifies master fields are properly copied into controller
 | |
| func TestNewBootstrapController(t *testing.T) {
 | |
| 	// Tests a subset of inputs to ensure they are set properly in the controller
 | |
| 	master, etcdserver, _, assert := setUp(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	portRange := utilnet.PortRange{Base: 10, Size: 10}
 | |
| 
 | |
| 	master.namespaceRegistry = namespace.NewRegistry(nil)
 | |
| 	master.serviceRegistry = registrytest.NewServiceRegistry()
 | |
| 	master.endpointRegistry = endpoint.NewRegistry(nil)
 | |
| 
 | |
| 	master.ServiceNodePortRange = portRange
 | |
| 	master.MasterCount = 1
 | |
| 	master.ServiceReadWritePort = 1000
 | |
| 	master.PublicReadWritePort = 1010
 | |
| 
 | |
| 	controller := master.NewBootstrapController()
 | |
| 
 | |
| 	assert.Equal(controller.NamespaceRegistry, master.namespaceRegistry)
 | |
| 	assert.Equal(controller.EndpointRegistry, master.endpointRegistry)
 | |
| 	assert.Equal(controller.ServiceRegistry, master.serviceRegistry)
 | |
| 	assert.Equal(controller.ServiceNodePortRange, portRange)
 | |
| 	assert.Equal(controller.MasterCount, master.MasterCount)
 | |
| 	assert.Equal(controller.ServicePort, master.ServiceReadWritePort)
 | |
| 	assert.Equal(controller.PublicServicePort, master.PublicReadWritePort)
 | |
| }
 | |
| 
 | |
| // TestControllerServicePorts verifies master extraServicePorts are
 | |
| // correctly copied into controller
 | |
| func TestControllerServicePorts(t *testing.T) {
 | |
| 	master, etcdserver, _, assert := setUp(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	master.namespaceRegistry = namespace.NewRegistry(nil)
 | |
| 	master.serviceRegistry = registrytest.NewServiceRegistry()
 | |
| 	master.endpointRegistry = endpoint.NewRegistry(nil)
 | |
| 
 | |
| 	master.ExtraServicePorts = []api.ServicePort{
 | |
| 		{
 | |
| 			Name:       "additional-port-1",
 | |
| 			Port:       1000,
 | |
| 			Protocol:   api.ProtocolTCP,
 | |
| 			TargetPort: intstr.FromInt(1000),
 | |
| 		},
 | |
| 		{
 | |
| 			Name:       "additional-port-2",
 | |
| 			Port:       1010,
 | |
| 			Protocol:   api.ProtocolTCP,
 | |
| 			TargetPort: intstr.FromInt(1010),
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	controller := master.NewBootstrapController()
 | |
| 
 | |
| 	assert.Equal(1000, controller.ExtraServicePorts[0].Port)
 | |
| 	assert.Equal(1010, controller.ExtraServicePorts[1].Port)
 | |
| }
 | |
| 
 | |
| // TestGetNodeAddresses verifies that proper results are returned
 | |
| // when requesting node addresses.
 | |
| func TestGetNodeAddresses(t *testing.T) {
 | |
| 	master, etcdserver, _, assert := setUp(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	// Fail case (no addresses associated with nodes)
 | |
| 	nodes, _ := master.nodeRegistry.ListNodes(api.NewDefaultContext(), nil)
 | |
| 	addrs, err := master.getNodeAddresses()
 | |
| 
 | |
| 	assert.Error(err, "getNodeAddresses should have caused an error as there are no addresses.")
 | |
| 	assert.Equal([]string(nil), addrs)
 | |
| 
 | |
| 	// Pass case with External type IP
 | |
| 	nodes, _ = master.nodeRegistry.ListNodes(api.NewDefaultContext(), nil)
 | |
| 	for index := range nodes.Items {
 | |
| 		nodes.Items[index].Status.Addresses = []api.NodeAddress{{Type: api.NodeExternalIP, Address: "127.0.0.1"}}
 | |
| 	}
 | |
| 	addrs, err = master.getNodeAddresses()
 | |
| 	assert.NoError(err, "getNodeAddresses should not have returned an error.")
 | |
| 	assert.Equal([]string{"127.0.0.1", "127.0.0.1"}, addrs)
 | |
| 
 | |
| 	// Pass case with LegacyHost type IP
 | |
| 	nodes, _ = master.nodeRegistry.ListNodes(api.NewDefaultContext(), nil)
 | |
| 	for index := range nodes.Items {
 | |
| 		nodes.Items[index].Status.Addresses = []api.NodeAddress{{Type: api.NodeLegacyHostIP, Address: "127.0.0.2"}}
 | |
| 	}
 | |
| 	addrs, err = master.getNodeAddresses()
 | |
| 	assert.NoError(err, "getNodeAddresses failback should not have returned an error.")
 | |
| 	assert.Equal([]string{"127.0.0.2", "127.0.0.2"}, addrs)
 | |
| }
 | |
| 
 | |
| // Because we need to be backwards compatible with release 1.1, at endpoints
 | |
| // that exist in release 1.1, the responses should have empty APIVersion.
 | |
| func TestAPIVersionOfDiscoveryEndpoints(t *testing.T) {
 | |
| 	master, etcdserver, _, assert := newMaster(t)
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	server := httptest.NewServer(master.HandlerContainer.ServeMux)
 | |
| 
 | |
| 	// /api exists in release-1.1
 | |
| 	resp, err := http.Get(server.URL + "/api")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	apiVersions := unversioned.APIVersions{}
 | |
| 	assert.NoError(decodeResponse(resp, &apiVersions))
 | |
| 	assert.Equal(apiVersions.APIVersion, "")
 | |
| 
 | |
| 	// /api/v1 exists in release-1.1
 | |
| 	resp, err = http.Get(server.URL + "/api/v1")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	resourceList := unversioned.APIResourceList{}
 | |
| 	assert.NoError(decodeResponse(resp, &resourceList))
 | |
| 	assert.Equal(resourceList.APIVersion, "")
 | |
| 
 | |
| 	// /apis exists in release-1.1
 | |
| 	resp, err = http.Get(server.URL + "/apis")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	groupList := unversioned.APIGroupList{}
 | |
| 	assert.NoError(decodeResponse(resp, &groupList))
 | |
| 	assert.Equal(groupList.APIVersion, "")
 | |
| 
 | |
| 	// /apis/extensions exists in release-1.1
 | |
| 	resp, err = http.Get(server.URL + "/apis/extensions")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	group := unversioned.APIGroup{}
 | |
| 	assert.NoError(decodeResponse(resp, &group))
 | |
| 	assert.Equal(group.APIVersion, "")
 | |
| 
 | |
| 	// /apis/extensions/v1beta1 exists in release-1.1
 | |
| 	resp, err = http.Get(server.URL + "/apis/extensions/v1beta1")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	resourceList = unversioned.APIResourceList{}
 | |
| 	assert.NoError(decodeResponse(resp, &resourceList))
 | |
| 	assert.Equal(resourceList.APIVersion, "")
 | |
| 
 | |
| 	// /apis/autoscaling doesn't exist in release-1.1, so the APIVersion field
 | |
| 	// should be non-empty in the results returned by the server.
 | |
| 	resp, err = http.Get(server.URL + "/apis/autoscaling")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	group = unversioned.APIGroup{}
 | |
| 	assert.NoError(decodeResponse(resp, &group))
 | |
| 	assert.Equal(group.APIVersion, "v1")
 | |
| 
 | |
| 	// apis/autoscaling/v1 doesn't exist in release-1.1, so the APIVersion field
 | |
| 	// should be non-empty in the results returned by the server.
 | |
| 
 | |
| 	resp, err = http.Get(server.URL + "/apis/autoscaling/v1")
 | |
| 	if err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 	resourceList = unversioned.APIResourceList{}
 | |
| 	assert.NoError(decodeResponse(resp, &resourceList))
 | |
| 	assert.Equal(resourceList.APIVersion, "v1")
 | |
| 
 | |
| }
 | |
| 
 | |
| func TestDiscoveryAtAPIS(t *testing.T) {
 | |
| 	// TODO(caesarxuchao): make this pass  now that batch is added,
 | |
| 	// and rewrite it so that the indexes do not need to change each time a new api group is added.
 | |
| 	/*
 | |
| 		master, etcdserver, config, assert := newMaster(t)
 | |
| 		defer etcdserver.Terminate(t)
 | |
| 
 | |
| 		server := httptest.NewServer(master.HandlerContainer.ServeMux)
 | |
| 		resp, err := http.Get(server.URL + "/apis")
 | |
| 		if !assert.NoError(err) {
 | |
| 			t.Errorf("unexpected error: %v", err)
 | |
| 		}
 | |
| 
 | |
| 		assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 		groupList := unversioned.APIGroupList{}
 | |
| 		assert.NoError(decodeResponse(resp, &groupList))
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("unexpected error: %v", err)
 | |
| 		}
 | |
| 
 | |
| 		expectGroupNames := []string{autoscaling.GroupName, batch.GroupName, extensions.GroupName}
 | |
| 		expectVersions := [][]unversioned.GroupVersionForDiscovery{
 | |
| 			{
 | |
| 				{
 | |
| 					GroupVersion: testapi.Autoscaling.GroupVersion().String(),
 | |
| 					Version:      testapi.Autoscaling.GroupVersion().Version,
 | |
| 				},
 | |
| 			},
 | |
| 			{
 | |
| 				{
 | |
| 					GroupVersion: testapi.Batch.GroupVersion().String(),
 | |
| 					Version:      testapi.Batch.GroupVersion().Version,
 | |
| 				},
 | |
| 			},
 | |
| 			{
 | |
| 				{
 | |
| 					GroupVersion: testapi.Extensions.GroupVersion().String(),
 | |
| 					Version:      testapi.Extensions.GroupVersion().Version,
 | |
| 				},
 | |
| 			},
 | |
| 		}
 | |
| 		expectPreferredVersion := []unversioned.GroupVersionForDiscovery{
 | |
| 			{
 | |
| 				GroupVersion: config.StorageVersions[autoscaling.GroupName],
 | |
| 				Version:      apiutil.GetVersion(config.StorageVersions[autoscaling.GroupName]),
 | |
| 			},
 | |
| 			{
 | |
| 				GroupVersion: config.StorageVersions[batch.GroupName],
 | |
| 				Version:      apiutil.GetVersion(config.StorageVersions[batch.GroupName]),
 | |
| 			},
 | |
| 			{
 | |
| 				GroupVersion: config.StorageVersions[extensions.GroupName],
 | |
| 				Version:      apiutil.GetVersion(config.StorageVersions[extensions.GroupName]),
 | |
| 			},
 | |
| 		}
 | |
| 
 | |
| 
 | |
| 		assert.Equal(2, len(groupList.Groups))
 | |
| 		assert.Equal(expectGroupNames[0], groupList.Groups[0].Name)
 | |
| 		assert.Equal(expectGroupNames[1], groupList.Groups[1].Name)
 | |
| 
 | |
| 		assert.Equal(expectVersions[0], groupList.Groups[0].Versions)
 | |
| 		assert.Equal(expectVersions[1], groupList.Groups[1].Versions)
 | |
| 
 | |
| 		assert.Equal(expectPreferredVersion[0], groupList.Groups[0].PreferredVersion)
 | |
| 		assert.Equal(expectPreferredVersion[1], groupList.Groups[1].PreferredVersion)
 | |
| 
 | |
| 		thirdPartyGV := unversioned.GroupVersionForDiscovery{GroupVersion: "company.com/v1", Version: "v1"}
 | |
| 		master.addThirdPartyResourceStorage("/apis/company.com/v1", nil,
 | |
| 			unversioned.APIGroup{
 | |
| 				Name:             "company.com",
 | |
| 				Versions:         []unversioned.GroupVersionForDiscovery{thirdPartyGV},
 | |
| 				PreferredVersion: thirdPartyGV,
 | |
| 			})
 | |
| 
 | |
| 		resp, err = http.Get(server.URL + "/apis")
 | |
| 		if !assert.NoError(err) {
 | |
| 			t.Errorf("unexpected error: %v", err)
 | |
| 		}
 | |
| 
 | |
| 		assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 		assert.NoError(decodeResponse(resp, &groupList))
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("unexpected error: %v", err)
 | |
| 		}
 | |
| 
 | |
| 		thirdPartyGroupName := "company.com"
 | |
| 		thirdPartyExpectVersions := []unversioned.GroupVersionForDiscovery{thirdPartyGV}
 | |
| 
 | |
| 		assert.Equal(4, len(groupList.Groups))
 | |
| 		// autoscaling group
 | |
| 		assert.Equal(expectGroupNames[0], groupList.Groups[0].Name)
 | |
| 		assert.Equal(expectVersions[0], groupList.Groups[0].Versions)
 | |
| 		assert.Equal(expectPreferredVersion[0], groupList.Groups[0].PreferredVersion)
 | |
| 		// batch group
 | |
| 		assert.Equal(expectGroupNames[1], groupList.Groups[1].Name)
 | |
| 		assert.Equal(expectVersions[1], groupList.Groups[1].Versions)
 | |
| 		assert.Equal(expectPreferredVersion[1], groupList.Groups[1].PreferredVersion)
 | |
| 		// third party
 | |
| 		assert.Equal(thirdPartyGroupName, groupList.Groups[2].Name)
 | |
| 		assert.Equal(thirdPartyExpectVersions, groupList.Groups[2].Versions)
 | |
| 		assert.Equal(thirdPartyGV, groupList.Groups[2].PreferredVersion)
 | |
| 		// extensions group
 | |
| 		assert.Equal(expectGroupNames[2], groupList.Groups[3].Name)
 | |
| 		assert.Equal(expectVersions[2], groupList.Groups[3].Versions)
 | |
| 		assert.Equal(expectPreferredVersion[2], groupList.Groups[3].PreferredVersion)
 | |
| 	*/
 | |
| }
 | |
| 
 | |
| var versionsToTest = []string{"v1", "v3"}
 | |
| 
 | |
| type Foo struct {
 | |
| 	unversioned.TypeMeta `json:",inline"`
 | |
| 	api.ObjectMeta       `json:"metadata,omitempty" description:"standard object metadata"`
 | |
| 
 | |
| 	SomeField  string `json:"someField"`
 | |
| 	OtherField int    `json:"otherField"`
 | |
| }
 | |
| 
 | |
| type FooList struct {
 | |
| 	unversioned.TypeMeta `json:",inline"`
 | |
| 	unversioned.ListMeta `json:"metadata,omitempty" description:"standard list metadata; see http://releases.k8s.io/HEAD/docs/devel/api-conventions.md#metadata"`
 | |
| 
 | |
| 	Items []Foo `json:"items"`
 | |
| }
 | |
| 
 | |
| func initThirdParty(t *testing.T, version string) (*Master, *etcdtesting.EtcdTestServer, *httptest.Server, *assert.Assertions) {
 | |
| 	master, etcdserver, _, assert := newMaster(t)
 | |
| 
 | |
| 	api := &extensions.ThirdPartyResource{
 | |
| 		ObjectMeta: api.ObjectMeta{
 | |
| 			Name: "foo.company.com",
 | |
| 		},
 | |
| 		Versions: []extensions.APIVersion{
 | |
| 			{
 | |
| 				APIGroup: "group",
 | |
| 				Name:     version,
 | |
| 			},
 | |
| 		},
 | |
| 	}
 | |
| 	master.thirdPartyStorage = etcdstorage.NewEtcdStorage(etcdserver.Client, testapi.Extensions.Codec(), etcdtest.PathPrefix(), false)
 | |
| 	_, master.ServiceClusterIPRange, _ = net.ParseCIDR("10.0.0.0/24")
 | |
| 
 | |
| 	if !assert.NoError(master.InstallThirdPartyResource(api)) {
 | |
| 		t.FailNow()
 | |
| 	}
 | |
| 
 | |
| 	server := httptest.NewServer(master.HandlerContainer.ServeMux)
 | |
| 	return master, etcdserver, server, assert
 | |
| }
 | |
| 
 | |
| func TestInstallThirdPartyAPIList(t *testing.T) {
 | |
| 	for _, version := range versionsToTest {
 | |
| 		testInstallThirdPartyAPIListVersion(t, version)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testInstallThirdPartyAPIListVersion(t *testing.T, version string) {
 | |
| 	tests := []struct {
 | |
| 		items []Foo
 | |
| 	}{
 | |
| 		{},
 | |
| 		{
 | |
| 			items: []Foo{},
 | |
| 		},
 | |
| 		{
 | |
| 			items: []Foo{
 | |
| 				{
 | |
| 					ObjectMeta: api.ObjectMeta{
 | |
| 						Name: "test",
 | |
| 					},
 | |
| 					TypeMeta: unversioned.TypeMeta{
 | |
| 						Kind:       "Foo",
 | |
| 						APIVersion: version,
 | |
| 					},
 | |
| 					SomeField:  "test field",
 | |
| 					OtherField: 10,
 | |
| 				},
 | |
| 				{
 | |
| 					ObjectMeta: api.ObjectMeta{
 | |
| 						Name: "bar",
 | |
| 					},
 | |
| 					TypeMeta: unversioned.TypeMeta{
 | |
| 						Kind:       "Foo",
 | |
| 						APIVersion: version,
 | |
| 					},
 | |
| 					SomeField:  "test field another",
 | |
| 					OtherField: 20,
 | |
| 				},
 | |
| 			},
 | |
| 		},
 | |
| 	}
 | |
| 	for _, test := range tests {
 | |
| 		func() {
 | |
| 			master, etcdserver, server, assert := initThirdParty(t, version)
 | |
| 			// TODO: Uncomment when fix #19254
 | |
| 			// defer server.Close()
 | |
| 			defer etcdserver.Terminate(t)
 | |
| 
 | |
| 			if test.items != nil {
 | |
| 				storeThirdPartyList(master.thirdPartyStorage, "/ThirdPartyResourceData/company.com/foos/default", test.items)
 | |
| 			}
 | |
| 
 | |
| 			resp, err := http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos")
 | |
| 			if !assert.NoError(err) {
 | |
| 				return
 | |
| 			}
 | |
| 			defer resp.Body.Close()
 | |
| 
 | |
| 			assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 			data, err := ioutil.ReadAll(resp.Body)
 | |
| 			assert.NoError(err)
 | |
| 
 | |
| 			list := FooList{}
 | |
| 			if err = json.Unmarshal(data, &list); err != nil {
 | |
| 				t.Errorf("unexpected error: %v", err)
 | |
| 			}
 | |
| 
 | |
| 			if test.items == nil {
 | |
| 				if len(list.Items) != 0 {
 | |
| 					t.Errorf("expected no items, saw: %v", list.Items)
 | |
| 				}
 | |
| 				return
 | |
| 			}
 | |
| 
 | |
| 			if len(list.Items) != len(test.items) {
 | |
| 				t.Errorf("unexpected length: %d vs %d", len(list.Items), len(test.items))
 | |
| 				return
 | |
| 			}
 | |
| 			// The order of elements in LIST is not guaranteed.
 | |
| 			mapping := make(map[string]int)
 | |
| 			for ix := range test.items {
 | |
| 				mapping[test.items[ix].Name] = ix
 | |
| 			}
 | |
| 			for ix := range list.Items {
 | |
| 				// Copy things that are set dynamically on the server
 | |
| 				expectedObj := test.items[mapping[list.Items[ix].Name]]
 | |
| 				expectedObj.SelfLink = list.Items[ix].SelfLink
 | |
| 				expectedObj.ResourceVersion = list.Items[ix].ResourceVersion
 | |
| 				expectedObj.Namespace = list.Items[ix].Namespace
 | |
| 				expectedObj.UID = list.Items[ix].UID
 | |
| 				expectedObj.CreationTimestamp = list.Items[ix].CreationTimestamp
 | |
| 
 | |
| 				// We endure the order of items by sorting them (using 'mapping')
 | |
| 				// so that this function passes.
 | |
| 				if !reflect.DeepEqual(list.Items[ix], expectedObj) {
 | |
| 					t.Errorf("expected:\n%#v\nsaw:\n%#v\n", expectedObj, list.Items[ix])
 | |
| 				}
 | |
| 			}
 | |
| 		}()
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func encodeToThirdParty(name string, obj interface{}) (runtime.Object, error) {
 | |
| 	serial, err := json.Marshal(obj)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	thirdPartyData := extensions.ThirdPartyResourceData{
 | |
| 		ObjectMeta: api.ObjectMeta{Name: name},
 | |
| 		Data:       serial,
 | |
| 	}
 | |
| 	return &thirdPartyData, nil
 | |
| }
 | |
| 
 | |
| func storeThirdPartyObject(s storage.Interface, path, name string, obj interface{}) error {
 | |
| 	data, err := encodeToThirdParty(name, obj)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	return s.Set(context.TODO(), etcdtest.AddPrefix(path), data, nil, 0)
 | |
| }
 | |
| 
 | |
| func storeThirdPartyList(s storage.Interface, path string, list []Foo) error {
 | |
| 	for _, obj := range list {
 | |
| 		if err := storeThirdPartyObject(s, path+"/"+obj.Name, obj.Name, obj); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func decodeResponse(resp *http.Response, obj interface{}) error {
 | |
| 	defer resp.Body.Close()
 | |
| 
 | |
| 	data, err := ioutil.ReadAll(resp.Body)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	if err := json.Unmarshal(data, obj); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func TestInstallThirdPartyAPIGet(t *testing.T) {
 | |
| 	for _, version := range versionsToTest {
 | |
| 		testInstallThirdPartyAPIGetVersion(t, version)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testInstallThirdPartyAPIGetVersion(t *testing.T, version string) {
 | |
| 	master, etcdserver, server, assert := initThirdParty(t, version)
 | |
| 	// TODO: Uncomment when fix #19254
 | |
| 	// defer server.Close()
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	expectedObj := Foo{
 | |
| 		ObjectMeta: api.ObjectMeta{
 | |
| 			Name: "test",
 | |
| 		},
 | |
| 		TypeMeta: unversioned.TypeMeta{
 | |
| 			Kind:       "Foo",
 | |
| 			APIVersion: version,
 | |
| 		},
 | |
| 		SomeField:  "test field",
 | |
| 		OtherField: 10,
 | |
| 	}
 | |
| 	if !assert.NoError(storeThirdPartyObject(master.thirdPartyStorage, "/ThirdPartyResourceData/company.com/foos/default/test", "test", expectedObj)) {
 | |
| 		t.FailNow()
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	resp, err := http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 	item := Foo{}
 | |
| 	assert.NoError(decodeResponse(resp, &item))
 | |
| 	if !assert.False(reflect.DeepEqual(item, expectedObj)) {
 | |
| 		t.Errorf("expected objects to not be equal:\n%v\nsaw:\n%v\n", expectedObj, item)
 | |
| 	}
 | |
| 	// Fill in data that the apiserver injects
 | |
| 	expectedObj.SelfLink = item.SelfLink
 | |
| 	expectedObj.ResourceVersion = item.ResourceVersion
 | |
| 	if !assert.True(reflect.DeepEqual(item, expectedObj)) {
 | |
| 		t.Errorf("expected:\n%#v\nsaw:\n%#v\n", expectedObj, item)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestInstallThirdPartyAPIPost(t *testing.T) {
 | |
| 	for _, version := range versionsToTest {
 | |
| 		testInstallThirdPartyAPIPostForVersion(t, version)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testInstallThirdPartyAPIPostForVersion(t *testing.T, version string) {
 | |
| 	master, etcdserver, server, assert := initThirdParty(t, version)
 | |
| 	// TODO: Uncomment when fix #19254
 | |
| 	// defer server.Close()
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	inputObj := Foo{
 | |
| 		ObjectMeta: api.ObjectMeta{
 | |
| 			Name: "test",
 | |
| 		},
 | |
| 		TypeMeta: unversioned.TypeMeta{
 | |
| 			Kind:       "Foo",
 | |
| 			APIVersion: "company.com/" + version,
 | |
| 		},
 | |
| 		SomeField:  "test field",
 | |
| 		OtherField: 10,
 | |
| 	}
 | |
| 	data, err := json.Marshal(inputObj)
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	resp, err := http.Post(server.URL+"/apis/company.com/"+version+"/namespaces/default/foos", "application/json", bytes.NewBuffer(data))
 | |
| 	if !assert.NoError(err) {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(http.StatusCreated, resp.StatusCode)
 | |
| 
 | |
| 	item := Foo{}
 | |
| 	assert.NoError(decodeResponse(resp, &item))
 | |
| 
 | |
| 	// fill in fields set by the apiserver
 | |
| 	expectedObj := inputObj
 | |
| 	expectedObj.SelfLink = item.SelfLink
 | |
| 	expectedObj.ResourceVersion = item.ResourceVersion
 | |
| 	expectedObj.Namespace = item.Namespace
 | |
| 	expectedObj.UID = item.UID
 | |
| 	expectedObj.CreationTimestamp = item.CreationTimestamp
 | |
| 	if !assert.True(reflect.DeepEqual(item, expectedObj)) {
 | |
| 		t.Errorf("expected:\n%v\nsaw:\n%v\n", expectedObj, item)
 | |
| 	}
 | |
| 
 | |
| 	thirdPartyObj := extensions.ThirdPartyResourceData{}
 | |
| 	err = master.thirdPartyStorage.Get(
 | |
| 		context.TODO(), etcdtest.AddPrefix("/ThirdPartyResourceData/company.com/foos/default/test"),
 | |
| 		&thirdPartyObj, false)
 | |
| 	if !assert.NoError(err) {
 | |
| 		t.FailNow()
 | |
| 	}
 | |
| 
 | |
| 	item = Foo{}
 | |
| 	assert.NoError(json.Unmarshal(thirdPartyObj.Data, &item))
 | |
| 
 | |
| 	if !assert.True(reflect.DeepEqual(item, inputObj)) {
 | |
| 		t.Errorf("expected:\n%v\nsaw:\n%v\n", inputObj, item)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestInstallThirdPartyAPIDelete(t *testing.T) {
 | |
| 	for _, version := range versionsToTest {
 | |
| 		testInstallThirdPartyAPIDeleteVersion(t, version)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testInstallThirdPartyAPIDeleteVersion(t *testing.T, version string) {
 | |
| 	master, etcdserver, server, assert := initThirdParty(t, version)
 | |
| 	// TODO: Uncomment when fix #19254
 | |
| 	// defer server.Close()
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	expectedObj := Foo{
 | |
| 		ObjectMeta: api.ObjectMeta{
 | |
| 			Name:      "test",
 | |
| 			Namespace: "default",
 | |
| 		},
 | |
| 		TypeMeta: unversioned.TypeMeta{
 | |
| 			Kind: "Foo",
 | |
| 		},
 | |
| 		SomeField:  "test field",
 | |
| 		OtherField: 10,
 | |
| 	}
 | |
| 	if !assert.NoError(storeThirdPartyObject(master.thirdPartyStorage, "/ThirdPartyResourceData/company.com/foos/default/test", "test", expectedObj)) {
 | |
| 		t.FailNow()
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	resp, err := http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 	item := Foo{}
 | |
| 	assert.NoError(decodeResponse(resp, &item))
 | |
| 
 | |
| 	// Fill in fields set by the apiserver
 | |
| 	expectedObj.SelfLink = item.SelfLink
 | |
| 	expectedObj.ResourceVersion = item.ResourceVersion
 | |
| 	expectedObj.Namespace = item.Namespace
 | |
| 	if !assert.True(reflect.DeepEqual(item, expectedObj)) {
 | |
| 		t.Errorf("expected:\n%v\nsaw:\n%v\n", expectedObj, item)
 | |
| 	}
 | |
| 
 | |
| 	resp, err = httpDelete(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(http.StatusOK, resp.StatusCode)
 | |
| 
 | |
| 	resp, err = http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	assert.Equal(http.StatusNotFound, resp.StatusCode)
 | |
| 
 | |
| 	expectedDeletedKey := etcdtest.AddPrefix("ThirdPartyResourceData/company.com/foos/default/test")
 | |
| 	thirdPartyObj := extensions.ThirdPartyResourceData{}
 | |
| 	err = master.thirdPartyStorage.Get(
 | |
| 		context.TODO(), expectedDeletedKey, &thirdPartyObj, false)
 | |
| 	if !storage.IsNotFound(err) {
 | |
| 		t.Errorf("expected deletion didn't happen: %v", err)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func httpDelete(url string) (*http.Response, error) {
 | |
| 	req, err := http.NewRequest("DELETE", url, nil)
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	client := &http.Client{}
 | |
| 	return client.Do(req)
 | |
| }
 | |
| 
 | |
| func TestInstallThirdPartyResourceRemove(t *testing.T) {
 | |
| 	for _, version := range versionsToTest {
 | |
| 		testInstallThirdPartyResourceRemove(t, version)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func testInstallThirdPartyResourceRemove(t *testing.T, version string) {
 | |
| 	master, etcdserver, server, assert := initThirdParty(t, version)
 | |
| 	// TODO: Uncomment when fix #19254
 | |
| 	// defer server.Close()
 | |
| 	defer etcdserver.Terminate(t)
 | |
| 
 | |
| 	expectedObj := Foo{
 | |
| 		ObjectMeta: api.ObjectMeta{
 | |
| 			Name: "test",
 | |
| 		},
 | |
| 		TypeMeta: unversioned.TypeMeta{
 | |
| 			Kind: "Foo",
 | |
| 		},
 | |
| 		SomeField:  "test field",
 | |
| 		OtherField: 10,
 | |
| 	}
 | |
| 	if !assert.NoError(storeThirdPartyObject(master.thirdPartyStorage, "/ThirdPartyResourceData/company.com/foos/default/test", "test", expectedObj)) {
 | |
| 		t.FailNow()
 | |
| 		return
 | |
| 	}
 | |
| 	secondObj := expectedObj
 | |
| 	secondObj.Name = "bar"
 | |
| 	if !assert.NoError(storeThirdPartyObject(master.thirdPartyStorage, "/ThirdPartyResourceData/company.com/foos/default/bar", "bar", secondObj)) {
 | |
| 		t.FailNow()
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	resp, err := http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		t.FailNow()
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if resp.StatusCode != http.StatusOK {
 | |
| 		t.Errorf("unexpected status: %v", resp)
 | |
| 	}
 | |
| 
 | |
| 	item := Foo{}
 | |
| 	if err := decodeResponse(resp, &item); err != nil {
 | |
| 		t.Errorf("unexpected error: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	// TODO: validate etcd set things here
 | |
| 	item.ObjectMeta = expectedObj.ObjectMeta
 | |
| 
 | |
| 	if !assert.True(reflect.DeepEqual(item, expectedObj)) {
 | |
| 		t.Errorf("expected:\n%v\nsaw:\n%v\n", expectedObj, item)
 | |
| 	}
 | |
| 
 | |
| 	path := makeThirdPartyPath("company.com")
 | |
| 	master.RemoveThirdPartyResource(path)
 | |
| 
 | |
| 	resp, err = http.Get(server.URL + "/apis/company.com/" + version + "/namespaces/default/foos/test")
 | |
| 	if !assert.NoError(err) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if resp.StatusCode != http.StatusNotFound {
 | |
| 		t.Errorf("unexpected status: %v", resp)
 | |
| 	}
 | |
| 
 | |
| 	expectedDeletedKeys := []string{
 | |
| 		etcdtest.AddPrefix("/ThirdPartyResourceData/company.com/foos/default/test"),
 | |
| 		etcdtest.AddPrefix("/ThirdPartyResourceData/company.com/foos/default/bar"),
 | |
| 	}
 | |
| 	for _, key := range expectedDeletedKeys {
 | |
| 		thirdPartyObj := extensions.ThirdPartyResourceData{}
 | |
| 		err := master.thirdPartyStorage.Get(context.TODO(), key, &thirdPartyObj, false)
 | |
| 		if !storage.IsNotFound(err) {
 | |
| 			t.Errorf("expected deletion didn't happen: %v", err)
 | |
| 		}
 | |
| 	}
 | |
| 	installed := master.ListThirdPartyResources()
 | |
| 	if len(installed) != 0 {
 | |
| 		t.Errorf("Resource(s) still installed: %v", installed)
 | |
| 	}
 | |
| 	services := master.HandlerContainer.RegisteredWebServices()
 | |
| 	for ix := range services {
 | |
| 		if strings.HasPrefix(services[ix].RootPath(), "/apis/company.com") {
 | |
| 			t.Errorf("Web service still installed at %s: %#v", services[ix].RootPath(), services[ix])
 | |
| 		}
 | |
| 	}
 | |
| }
 |