가짜 클라이언트를 사용하여 간단한 Client-Go 테스트 작성 방법

문제 개요

이 경우 테스트 함수를 구성하는 것을 목표로 합니다. 특정 Kubernetes 네임스페이스에 대한 생성 타임스탬프를 검색하기 위한 GetNamespaceCreationTime 함수의 경우. 그러나 초기화 논리를 통합하고 가짜 클라이언트와 상호 작용하는 데 적합한 접근 방식을 찾는 데 어려움이 있습니다.

테스트 구현

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

이 방법을 사용하면 클러스터 내 구성 검색의 성공 및 실패 모두에 대한 동작을 어설션할 수 있는 보다 강력한 테스트 시나리오가 가능합니다.