constellation/bootstrapper/internal/kubernetes/kubewaiter/kubewaiter_test.go
Daniel Weiße 690b50b29d
dev-docs: Go package docs (#958)
* Remove unused package

* Add Go package docs to most packages

Signed-off-by: Daniel Weiße <dw@edgeless.systems>
Signed-off-by: Fabian Kammel <fk@edgeless.systems>
Signed-off-by: Paul Meyer <49727155+katexochen@users.noreply.github.com>
Co-authored-by: Paul Meyer <49727155+katexochen@users.noreply.github.com>
Co-authored-by: Fabian Kammel <fk@edgeless.systems>
2023-01-19 15:57:50 +01:00

61 lines
1.1 KiB
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package kubewaiter
import (
"context"
"errors"
"testing"
"github.com/stretchr/testify/require"
"go.uber.org/goleak"
corev1 "k8s.io/api/core/v1"
)
func TestMain(m *testing.M) {
goleak.VerifyTestMain(m)
}
func TestCloudKubeAPIWaiter(t *testing.T) {
testCases := map[string]struct {
kubeClient KubernetesClient
wantErr bool
}{
"success": {
kubeClient: &stubKubernetesClient{},
},
"error": {
kubeClient: &stubKubernetesClient{listAllNamespacesErr: errors.New("error")},
wantErr: true,
},
}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
require := require.New(t)
waiter := &CloudKubeAPIWaiter{}
ctx, cancel := context.WithTimeout(context.Background(), 0)
defer cancel()
err := waiter.Wait(ctx, tc.kubeClient)
if tc.wantErr {
require.Error(err)
} else {
require.NoError(err)
}
})
}
}
type stubKubernetesClient struct {
listAllNamespacesErr error
}
func (c *stubKubernetesClient) ListAllNamespaces(ctx context.Context) (*corev1.NamespaceList, error) {
return nil, c.listAllNamespacesErr
}