偽クライアントを使用して 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")
}

これらのメソッドを使用すると、より堅牢なテスト シナリオが可能になり、クラスター内構成の取得の成功と失敗の両方の動作をアサートできます。