forked from kyma-incubator/compass
/
suite.go
137 lines (103 loc) · 3.57 KB
/
suite.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package provisioner
import (
"io/ioutil"
"math/rand"
"os"
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/kyma-incubator/compass/components/provisioner/pkg/gqlschema"
v1meta "k8s.io/apimachinery/pkg/apis/meta/v1"
v1client "k8s.io/client-go/kubernetes/typed/core/v1"
"github.com/kyma-incubator/compass/tests/provisioner-tests/test/testkit"
"github.com/kyma-incubator/compass/tests/provisioner-tests/test/testkit/compass/provisioner"
"github.com/sirupsen/logrus"
"k8s.io/client-go/kubernetes"
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
"k8s.io/client-go/tools/clientcmd"
)
const (
ProvisioningTimeout = 90 * time.Minute
UpgradeTimeout = 90 * time.Minute
DeprovisioningTimeout = 60 * time.Minute
checkInterval = 10 * time.Second
)
type TestSuite struct {
TestId string
ProvisionerClient provisioner.Client
gardenerProviders []string
config testkit.TestConfig
secretsClient v1client.SecretInterface
}
func NewTestSuite(config testkit.TestConfig) (*TestSuite, error) {
rand.Seed(time.Now().UnixNano())
// TODO: Sleep ensures that the Istio Sidecar is up before running the tests. We can consider adding some health endpoint in the service to avoid hardcoded sleep.
time.Sleep(15 * time.Second)
provisionerClient := provisioner.NewProvisionerClient(config.InternalProvisionerURL, config.Tenant, config.QueryLogging)
testId := randStringBytes(8)
return &TestSuite{
TestId: testId,
ProvisionerClient: provisionerClient,
gardenerProviders: config.Gardener.Providers,
config: config,
}, nil
}
func (ts *TestSuite) Setup() error {
logrus.Infof("Setting up environment")
return nil
}
func (ts *TestSuite) Cleanup() {
logrus.Infof("Starting cleanup...")
// TODO: Fetch Provisioner logs if test failed
logrus.Infof("Cleanup completed.")
}
func (ts *TestSuite) Recover() {
if r := recover(); r != nil {
logrus.Warn("Recovered after panic signal: ", r)
}
}
func (ts *TestSuite) WaitUntilOperationIsFinished(timeout time.Duration, operationID string) (gqlschema.OperationStatus, error) {
var operationStatus gqlschema.OperationStatus
var err error
err = testkit.WaitForFunction(checkInterval, timeout, func() bool {
operationStatus, err = ts.ProvisionerClient.RuntimeOperationStatus(operationID)
if err != nil {
logrus.Warnf("Failed to get operation status: %s", err.Error())
return false
}
if operationStatus.State == gqlschema.OperationStateInProgress {
logrus.Infof("Operation '%s': %s in progress", operationStatus.Operation, operationID)
return false
}
return true
})
return operationStatus, err
}
func (ts *TestSuite) KubernetesClientFromRawConfig(t *testing.T, rawConfig string) *kubernetes.Clientset {
tempKubeconfigFile, err := ioutil.TempFile("", "kubeconfig")
require.NoError(t, err)
defer func() {
err := os.RemoveAll(tempKubeconfigFile.Name())
if err != nil {
logrus.Warnf("Failed to delete temporary Kubeconfig file: %s", err.Error())
}
}()
_, err = tempKubeconfigFile.WriteString(rawConfig)
require.NoError(t, err)
k8sConfig, err := clientcmd.BuildConfigFromFlags("", tempKubeconfigFile.Name())
require.NoError(t, err)
k8sClient, err := kubernetes.NewForConfig(k8sConfig)
require.NoError(t, err)
return k8sClient
}
func (ts *TestSuite) removeCredentialsSecret(secretName string) error {
return ts.secretsClient.Delete(secretName, &v1meta.DeleteOptions{})
}
const letterBytes = "abcdefghijklmnopqrstuvwxyz123456789"
func randStringBytes(n int) string {
b := make([]byte, n)
for i := range b {
b[i] = letterBytes[rand.Intn(len(letterBytes))]
}
return string(b)
}