如何使用假客戶端為Client-Go 編寫簡單測試

問題概述

在本例中，我們的目標是編寫一個測試函數GetNamespaceCreationTime 函數，用於檢索特定Kubernetes 命名空間的建立時間戳記。然而，您很難找到合適的方法來合併初始化邏輯並與假客戶端互動。

測試實作

為了有效測試 GetNamespaceCreationTime 函數，初始化過程不應駐留在函數本身。相反，Kubernetes 用戶端介面應作為參數傳遞給該函數。

將 GetNamespaceCreationTime 函數中的現有程式碼替換為以下內容：

import (
  "fmt"
  "k8s.io/api/core/v1"
  metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
  "time"
)

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 := rest.InClusterConfig()
  if err != nil {
    log.Warnf("Could not get in-cluster config: %s", err)
    return nil, err
  }

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

  return client, err
}

測試函數

在 TestGetNamespaceCreationTime 函數中，實例化假客戶端並呼叫 GetNamespaceCreationTIme 方法：

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")
}

這些方法支援更強大的測試場景，您可以在其中斷言成功和失敗的叢集內配置檢索的行為。