k8s.io/kubernetes@v1.31.0-alpha.0.0.20240520171757-56147500dadc/test/e2e/framework/providers/kubemark/kubemark.go (about)

     1  /*
     2  Copyright 2018 The Kubernetes Authors.
     3  
     4  Licensed under the Apache License, Version 2.0 (the "License");
     5  you may not use this file except in compliance with the License.
     6  You may obtain a copy of the License at
     7  
     8      http://www.apache.org/licenses/LICENSE-2.0
     9  
    10  Unless required by applicable law or agreed to in writing, software
    11  distributed under the License is distributed on an "AS IS" BASIS,
    12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13  See the License for the specific language governing permissions and
    14  limitations under the License.
    15  */
    16  
    17  package kubemark
    18  
    19  import (
    20  	"flag"
    21  	"fmt"
    22  
    23  	"k8s.io/client-go/informers"
    24  	clientset "k8s.io/client-go/kubernetes"
    25  	"k8s.io/client-go/tools/clientcmd"
    26  	"k8s.io/kubernetes/pkg/kubemark"
    27  	"k8s.io/kubernetes/test/e2e/framework"
    28  )
    29  
    30  var (
    31  	kubemarkExternalKubeConfig = flag.String(fmt.Sprintf("%s-%s", "kubemark-external", clientcmd.RecommendedConfigPathFlag), "", "Path to kubeconfig containing embedded authinfo for external cluster.")
    32  )
    33  
    34  func init() {
    35  	framework.RegisterProvider("kubemark", newProvider)
    36  }
    37  
    38  func newProvider() (framework.ProviderInterface, error) {
    39  	// Actual initialization happens when the e2e framework gets constructed.
    40  	return &Provider{}, nil
    41  }
    42  
    43  // Provider is a structure to handle Kubemark cluster for e2e testing
    44  type Provider struct {
    45  	framework.NullProvider
    46  	controller   *kubemark.KubemarkController
    47  	closeChannel chan struct{}
    48  }
    49  
    50  // ResizeGroup resizes an instance group
    51  func (p *Provider) ResizeGroup(group string, size int32) error {
    52  	return p.controller.SetNodeGroupSize(group, int(size))
    53  }
    54  
    55  // GetGroupNodes returns a node name for the specified node group
    56  func (p *Provider) GetGroupNodes(group string) ([]string, error) {
    57  	return p.controller.GetNodeNamesForNodeGroup(group)
    58  }
    59  
    60  // FrameworkBeforeEach prepares clients, configurations etc. for e2e testing
    61  func (p *Provider) FrameworkBeforeEach(f *framework.Framework) {
    62  	if *kubemarkExternalKubeConfig != "" && p.controller == nil {
    63  		externalConfig, err := clientcmd.BuildConfigFromFlags("", *kubemarkExternalKubeConfig)
    64  		externalConfig.QPS = f.Options.ClientQPS
    65  		externalConfig.Burst = f.Options.ClientBurst
    66  		framework.ExpectNoError(err)
    67  		externalClient, err := clientset.NewForConfig(externalConfig)
    68  		framework.ExpectNoError(err)
    69  		f.KubemarkExternalClusterClientSet = externalClient
    70  		p.closeChannel = make(chan struct{})
    71  		externalInformerFactory := informers.NewSharedInformerFactory(externalClient, 0)
    72  		kubemarkInformerFactory := informers.NewSharedInformerFactory(f.ClientSet, 0)
    73  		kubemarkNodeInformer := kubemarkInformerFactory.Core().V1().Nodes()
    74  		go kubemarkNodeInformer.Informer().Run(p.closeChannel)
    75  		p.controller, err = kubemark.NewKubemarkController(externalClient, externalInformerFactory, f.ClientSet, kubemarkNodeInformer)
    76  		framework.ExpectNoError(err)
    77  		externalInformerFactory.Start(p.closeChannel)
    78  		if !p.controller.WaitForCacheSync(p.closeChannel) {
    79  			framework.Failf("Unable to sync caches for %v", p.controller)
    80  		}
    81  		go p.controller.Run(p.closeChannel)
    82  	}
    83  }
    84  
    85  // FrameworkAfterEach cleans up after e2e testing
    86  func (p *Provider) FrameworkAfterEach(f *framework.Framework) {
    87  	if p.closeChannel != nil {
    88  		close(p.closeChannel)
    89  		p.controller = nil
    90  		p.closeChannel = nil
    91  	}
    92  }
    93  
    94  // GroupSize returns the size of an instance group
    95  func (p *Provider) GroupSize(group string) (int, error) {
    96  	return p.controller.GetNodeGroupSize(group)
    97  }