[英]How to write simple tests for client-go using a fake client?
我正在尋找測試下面我的代碼的正確方法,我找不到任何示例我該怎么做。 只有main_test.go但它缺少 main.go 對我來說,如何使用它並不明顯。 我也經歷了 Github 問題,但我找不到任何有用的東西。
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")
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.