登录
首页 >  Golang >  Go问答

如何使用假客户端为客户端编写简单的测试?

来源:Golang技术栈

时间:2023-05-01 15:03:04 361浏览 收藏

来到golang学习网的大家,相信都是编程学习爱好者,希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《如何使用假客户端为客户端编写简单的测试?》,介绍一下golang,希望对大家的知识积累有所帮助,助力实战开发!

问题内容

问题

我正在寻找正确的方法来测试下面的代码,我找不到任何示例我该怎么做。只有[main_test.go](https://github.com/kubernetes/client- go/blob/master/examples/fake-client/),但它缺少 main.go,对我来说,如何使用它并不明显。我也经历了 Github[问题](https://github.com/kubernetes/client- go/issues?utf8=%E2%9C%93&q=fake+client+),但找不到任何有用的东西。

功能:

import (
  "fmt"
  "k8s.io/api/core/v1"
  metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  fake "k8s.io/client-go/kubernetes/fake"
  "time"
)

func GetNamespaceCreationTime(namespace string) int64 {
    config, err := rest.InClusterConfig()
    if err != nil {
        panic(err.Error())
    }
    clientset, err := kubernetes.NewForConfig(config)
    if err != nil {
        panic(err.Error())
    }

    ns, err := clientset.CoreV1().Namespaces().Get(namespace, metav1.GetOptions{})
    if err != nil {
        panic(err.Error())
    }
    fmt.Printf("%v \n", ns.CreationTimestamp)
    return (ns.GetCreationTimestamp().Unix())
}

测试:

我认为我的测试应该如下所示,但是我该如何使用config := fake.NewSimpleClientset(),我应该将它传递给GetNamespaceCreationTime函数吗?

func TestGetNamespaceCreationTime(t *testing.T) {
  config := fake.NewSimpleClientset()
  got := GetNamespaceCreationTime("default")
  want := int64(1257894000)

  nsMock := config.CoreV1().Namespaces()
  nsMock.Create(&v1.Namespace{
    ObjectMeta: metav1.ObjectMeta{
      Name:              "default",
      CreationTimestamp: metav1.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
    },
  })

  if got != want {
    t.Errorf("got %q want %q", got, want)
  }

正确答案

您需要将初始化部分移到 GetNamespaceCreationTime 函数之外。您可以将 kubernetes.Interface 作为参数传递给该函数。

    import (
      "fmt"
      "k8s.io/api/core/v1"
      metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
      fake "k8s.io/client-go/kubernetes/fake"
      "k8s.io/client-go/kubernetes"
      "time"
    )

var getInclusterConfigFunc = rest.InClusterConfig
var getNewKubeClientFunc = dynamic.NewForConfig
   
    
    func GetNamespaceCreationTime(kubeClient kubernetes.Interface, namespace string) int64 {
    
        ns, err := kubeClient.CoreV1().Namespaces().Get(namespace, metav1.GetOptions{})
        if err != nil {
            panic(err.Error())
        }
        fmt.Printf("%v \n", ns.CreationTimestamp)
        return (ns.GetCreationTimestamp().Unix())
    }
    
    func GetClientSet() kubernetes.Interface {
      config, err := getInclusterConfigFunc()
      if err != nil {
        log.Warnf("Could not get in-cluster config: %s", err)
        return nil, err
      }

      client, err := getNewKubeClientFunc(config)
      if err != nil {
        log.Warnf("Could not connect to in-cluster API server: %s", err)
        return nil, err
      }

      return client, err
    }

然后您的测试函数可以实例化假客户端并调用该方法。

    func TestGetNamespaceCreationTime(t *testing.T) {
      kubeClient := fake.NewSimpleClientset()
      got := GetNamespaceCreationTime(kubeClient, "default")
      want := int64(1257894000)
    
      nsMock := config.CoreV1().Namespaces()
      nsMock.Create(&v1.Namespace{
        ObjectMeta: metav1.ObjectMeta{
          Name:              "default",
          CreationTimestamp: metav1.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC),
        },
      })
    
      if got != want {
        t.Errorf("got %q want %q", got, want)
      }

func fakeGetInclusterConfig() (*rest.Config, error) {
    return nil, nil
}

func fakeGetInclusterConfigWithError() (*rest.Config, error) {
    return nil, errors.New("fake error getting in-cluster config")
}

func TestGetInclusterKubeClient(t *testing.T) {
    origGetInclusterConfig := getInclusterConfigFunc
    getInclusterConfigFunc = fakeGetInclusterConfig
    origGetNewKubeClient := getNewKubeClientFunc
    getNewKubeClientFunc = fakeGetNewKubeClient

    defer func() {
        getInclusterConfigFunc = origGetInclusterConfig
        getNewKubeClientFunc = origGetNewKubeClient
    }()

    client, err := GetClientSet()
    assert.Nil(t, client, "Client is not nil")
    assert.Nil(t, err, "error is not nil")
}

func TestGetInclusterKubeClient_ConfigError(t *testing.T) {
    origGetInclusterConfig := getInclusterConfigFunc
    getInclusterConfigFunc = fakeGetInclusterConfigWithError

    defer func() {
        getInclusterConfigFunc = origGetInclusterConfig
    }()

    client, err := GetInclusterKubeClient()
    assert.Nil(t, client, "Client is not nil")
    assert.NotNil(t, err, "error is nil")
}

终于介绍完啦!小伙伴们,这篇关于《如何使用假客户端为客户端编写简单的测试?》的介绍应该让你收获多多了吧!欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识,快来关注吧!

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>