269 lines
7.8 KiB
Go
269 lines
7.8 KiB
Go
|
/*
|
||
|
|
Copyright 2014 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 fake
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"regexp"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"k8s.io/apimachinery/pkg/types"
|
||
|
|
"k8s.io/kubernetes/pkg/api/v1"
|
||
|
|
"k8s.io/kubernetes/pkg/cloudprovider"
|
||
|
|
)
|
||
|
|
|
||
|
|
const ProviderName = "fake"
|
||
|
|
|
||
|
|
// FakeBalancer is a fake storage of balancer information
|
||
|
|
type FakeBalancer struct {
|
||
|
|
Name string
|
||
|
|
Region string
|
||
|
|
LoadBalancerIP string
|
||
|
|
Ports []v1.ServicePort
|
||
|
|
Hosts []*v1.Node
|
||
|
|
}
|
||
|
|
|
||
|
|
type FakeUpdateBalancerCall struct {
|
||
|
|
Service *v1.Service
|
||
|
|
Hosts []*v1.Node
|
||
|
|
}
|
||
|
|
|
||
|
|
// FakeCloud is a test-double implementation of Interface, LoadBalancer, Instances, and Routes. It is useful for testing.
|
||
|
|
type FakeCloud struct {
|
||
|
|
Exists bool
|
||
|
|
Err error
|
||
|
|
Calls []string
|
||
|
|
Addresses []v1.NodeAddress
|
||
|
|
ExtID map[types.NodeName]string
|
||
|
|
InstanceTypes map[types.NodeName]string
|
||
|
|
Machines []types.NodeName
|
||
|
|
NodeResources *v1.NodeResources
|
||
|
|
ClusterList []string
|
||
|
|
MasterName string
|
||
|
|
ExternalIP net.IP
|
||
|
|
Balancers map[string]FakeBalancer
|
||
|
|
UpdateCalls []FakeUpdateBalancerCall
|
||
|
|
RouteMap map[string]*FakeRoute
|
||
|
|
Lock sync.Mutex
|
||
|
|
cloudprovider.Zone
|
||
|
|
}
|
||
|
|
|
||
|
|
type FakeRoute struct {
|
||
|
|
ClusterName string
|
||
|
|
Route cloudprovider.Route
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) addCall(desc string) {
|
||
|
|
f.Calls = append(f.Calls, desc)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ClearCalls clears internal record of method calls to this FakeCloud.
|
||
|
|
func (f *FakeCloud) ClearCalls() {
|
||
|
|
f.Calls = []string{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) ListClusters() ([]string, error) {
|
||
|
|
return f.ClusterList, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) Master(name string) (string, error) {
|
||
|
|
return f.MasterName, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) Clusters() (cloudprovider.Clusters, bool) {
|
||
|
|
return f, true
|
||
|
|
}
|
||
|
|
|
||
|
|
// ProviderName returns the cloud provider ID.
|
||
|
|
func (f *FakeCloud) ProviderName() string {
|
||
|
|
return ProviderName
|
||
|
|
}
|
||
|
|
|
||
|
|
// ScrubDNS filters DNS settings for pods.
|
||
|
|
func (f *FakeCloud) ScrubDNS(nameservers, searches []string) (nsOut, srchOut []string) {
|
||
|
|
return nameservers, searches
|
||
|
|
}
|
||
|
|
|
||
|
|
// LoadBalancer returns a fake implementation of LoadBalancer.
|
||
|
|
// Actually it just returns f itself.
|
||
|
|
func (f *FakeCloud) LoadBalancer() (cloudprovider.LoadBalancer, bool) {
|
||
|
|
return f, true
|
||
|
|
}
|
||
|
|
|
||
|
|
// Instances returns a fake implementation of Instances.
|
||
|
|
//
|
||
|
|
// Actually it just returns f itself.
|
||
|
|
func (f *FakeCloud) Instances() (cloudprovider.Instances, bool) {
|
||
|
|
return f, true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) Zones() (cloudprovider.Zones, bool) {
|
||
|
|
return f, true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) Routes() (cloudprovider.Routes, bool) {
|
||
|
|
return f, true
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetLoadBalancer is a stub implementation of LoadBalancer.GetLoadBalancer.
|
||
|
|
func (f *FakeCloud) GetLoadBalancer(clusterName string, service *v1.Service) (*v1.LoadBalancerStatus, bool, error) {
|
||
|
|
status := &v1.LoadBalancerStatus{}
|
||
|
|
status.Ingress = []v1.LoadBalancerIngress{{IP: f.ExternalIP.String()}}
|
||
|
|
|
||
|
|
return status, f.Exists, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// EnsureLoadBalancer is a test-spy implementation of LoadBalancer.EnsureLoadBalancer.
|
||
|
|
// It adds an entry "create" into the internal method call record.
|
||
|
|
func (f *FakeCloud) EnsureLoadBalancer(clusterName string, service *v1.Service, nodes []*v1.Node) (*v1.LoadBalancerStatus, error) {
|
||
|
|
f.addCall("create")
|
||
|
|
if f.Balancers == nil {
|
||
|
|
f.Balancers = make(map[string]FakeBalancer)
|
||
|
|
}
|
||
|
|
|
||
|
|
name := cloudprovider.GetLoadBalancerName(service)
|
||
|
|
spec := service.Spec
|
||
|
|
|
||
|
|
zone, err := f.GetZone()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
region := zone.Region
|
||
|
|
|
||
|
|
f.Balancers[name] = FakeBalancer{name, region, spec.LoadBalancerIP, spec.Ports, nodes}
|
||
|
|
|
||
|
|
status := &v1.LoadBalancerStatus{}
|
||
|
|
status.Ingress = []v1.LoadBalancerIngress{{IP: f.ExternalIP.String()}}
|
||
|
|
|
||
|
|
return status, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// UpdateLoadBalancer is a test-spy implementation of LoadBalancer.UpdateLoadBalancer.
|
||
|
|
// It adds an entry "update" into the internal method call record.
|
||
|
|
func (f *FakeCloud) UpdateLoadBalancer(clusterName string, service *v1.Service, nodes []*v1.Node) error {
|
||
|
|
f.addCall("update")
|
||
|
|
f.UpdateCalls = append(f.UpdateCalls, FakeUpdateBalancerCall{service, nodes})
|
||
|
|
return f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// EnsureLoadBalancerDeleted is a test-spy implementation of LoadBalancer.EnsureLoadBalancerDeleted.
|
||
|
|
// It adds an entry "delete" into the internal method call record.
|
||
|
|
func (f *FakeCloud) EnsureLoadBalancerDeleted(clusterName string, service *v1.Service) error {
|
||
|
|
f.addCall("delete")
|
||
|
|
return f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) AddSSHKeyToAllInstances(user string, keyData []byte) error {
|
||
|
|
return errors.New("unimplemented")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Implementation of Instances.CurrentNodeName
|
||
|
|
func (f *FakeCloud) CurrentNodeName(hostname string) (types.NodeName, error) {
|
||
|
|
return types.NodeName(hostname), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// NodeAddresses is a test-spy implementation of Instances.NodeAddresses.
|
||
|
|
// It adds an entry "node-addresses" into the internal method call record.
|
||
|
|
func (f *FakeCloud) NodeAddresses(instance types.NodeName) ([]v1.NodeAddress, error) {
|
||
|
|
f.addCall("node-addresses")
|
||
|
|
return f.Addresses, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// ExternalID is a test-spy implementation of Instances.ExternalID.
|
||
|
|
// It adds an entry "external-id" into the internal method call record.
|
||
|
|
// It returns an external id to the mapped instance name, if not found, it will return "ext-{instance}"
|
||
|
|
func (f *FakeCloud) ExternalID(nodeName types.NodeName) (string, error) {
|
||
|
|
f.addCall("external-id")
|
||
|
|
return f.ExtID[nodeName], f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// InstanceID returns the cloud provider ID of the node with the specified Name.
|
||
|
|
func (f *FakeCloud) InstanceID(nodeName types.NodeName) (string, error) {
|
||
|
|
f.addCall("instance-id")
|
||
|
|
return f.ExtID[nodeName], nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// InstanceType returns the type of the specified instance.
|
||
|
|
func (f *FakeCloud) InstanceType(instance types.NodeName) (string, error) {
|
||
|
|
f.addCall("instance-type")
|
||
|
|
return f.InstanceTypes[instance], nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// List is a test-spy implementation of Instances.List.
|
||
|
|
// It adds an entry "list" into the internal method call record.
|
||
|
|
func (f *FakeCloud) List(filter string) ([]types.NodeName, error) {
|
||
|
|
f.addCall("list")
|
||
|
|
result := []types.NodeName{}
|
||
|
|
for _, machine := range f.Machines {
|
||
|
|
if match, _ := regexp.MatchString(filter, string(machine)); match {
|
||
|
|
result = append(result, machine)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return result, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) GetZone() (cloudprovider.Zone, error) {
|
||
|
|
f.addCall("get-zone")
|
||
|
|
return f.Zone, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) ListRoutes(clusterName string) ([]*cloudprovider.Route, error) {
|
||
|
|
f.Lock.Lock()
|
||
|
|
defer f.Lock.Unlock()
|
||
|
|
f.addCall("list-routes")
|
||
|
|
var routes []*cloudprovider.Route
|
||
|
|
for _, fakeRoute := range f.RouteMap {
|
||
|
|
if clusterName == fakeRoute.ClusterName {
|
||
|
|
routeCopy := fakeRoute.Route
|
||
|
|
routes = append(routes, &routeCopy)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return routes, f.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) CreateRoute(clusterName string, nameHint string, route *cloudprovider.Route) error {
|
||
|
|
f.Lock.Lock()
|
||
|
|
defer f.Lock.Unlock()
|
||
|
|
f.addCall("create-route")
|
||
|
|
name := clusterName + "-" + nameHint
|
||
|
|
if _, exists := f.RouteMap[name]; exists {
|
||
|
|
f.Err = fmt.Errorf("route %q already exists", name)
|
||
|
|
return f.Err
|
||
|
|
}
|
||
|
|
fakeRoute := FakeRoute{}
|
||
|
|
fakeRoute.Route = *route
|
||
|
|
fakeRoute.Route.Name = name
|
||
|
|
fakeRoute.ClusterName = clusterName
|
||
|
|
f.RouteMap[name] = &fakeRoute
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *FakeCloud) DeleteRoute(clusterName string, route *cloudprovider.Route) error {
|
||
|
|
f.Lock.Lock()
|
||
|
|
defer f.Lock.Unlock()
|
||
|
|
f.addCall("delete-route")
|
||
|
|
name := route.Name
|
||
|
|
if _, exists := f.RouteMap[name]; !exists {
|
||
|
|
f.Err = fmt.Errorf("no route found with name %q", name)
|
||
|
|
return f.Err
|
||
|
|
}
|
||
|
|
delete(f.RouteMap, name)
|
||
|
|
return nil
|
||
|
|
}
|