mirror of https://github.com/fluxcd/cli-utils.git
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
// Copyright 2020 The Kubernetes Authors.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package stress
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/onsi/gomega/format"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/klog/v2"
|
|
"k8s.io/kubectl/pkg/scheme"
|
|
"sigs.k8s.io/cli-utils/test/e2e/e2eutil"
|
|
"sigs.k8s.io/cli-utils/test/e2e/invconfig"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
|
|
)
|
|
|
|
// Parse optional logging flags
|
|
// Ex: ginkgo ./test/e2e/... -- -v=5
|
|
// Allow init for e2e test (not imported by external code)
|
|
// nolint:gochecknoinits
|
|
func init() {
|
|
klog.InitFlags(nil)
|
|
klog.SetOutput(GinkgoWriter)
|
|
}
|
|
|
|
var defaultTestTimeout = 1 * time.Hour
|
|
var defaultBeforeTestTimeout = 30 * time.Second
|
|
var defaultAfterTestTimeout = 30 * time.Second
|
|
|
|
var c client.Client
|
|
var invConfig invconfig.InventoryConfig
|
|
|
|
var _ = BeforeSuite(func() {
|
|
// increase from 4000 to handle long event lists
|
|
format.MaxLength = 10000
|
|
|
|
cfg, err := ctrl.GetConfig()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
cfg.UserAgent = e2eutil.UserAgent("test/stress")
|
|
|
|
if e2eutil.IsFlowControlEnabled(cfg) {
|
|
// Disable client-side throttling.
|
|
klog.V(3).Infof("Client-side throttling disabled")
|
|
cfg.QPS = -1
|
|
cfg.Burst = -1
|
|
}
|
|
|
|
invConfig = invconfig.NewCustomTypeInvConfig(cfg)
|
|
|
|
mapper, err := apiutil.NewDynamicRESTMapper(cfg)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
c, err = client.New(cfg, client.Options{
|
|
Scheme: scheme.Scheme,
|
|
Mapper: mapper,
|
|
})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), defaultBeforeTestTimeout)
|
|
defer cancel()
|
|
e2eutil.CreateInventoryCRD(ctx, c)
|
|
Expect(ctx.Err()).To(BeNil(), "BeforeSuite context cancelled or timed out")
|
|
})
|
|
|
|
var _ = AfterSuite(func() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), defaultAfterTestTimeout)
|
|
defer cancel()
|
|
e2eutil.DeleteInventoryCRD(ctx, c)
|
|
Expect(ctx.Err()).To(BeNil(), "AfterSuite context cancelled or timed out")
|
|
})
|
|
|
|
var _ = Describe("Stress", func() {
|
|
var namespace *v1.Namespace
|
|
var inventoryName string
|
|
var ctx context.Context
|
|
var cancel context.CancelFunc
|
|
|
|
BeforeEach(func() {
|
|
ctx, cancel = context.WithTimeout(context.Background(), defaultTestTimeout)
|
|
inventoryName = e2eutil.RandomString("test-inv-")
|
|
namespace = e2eutil.CreateRandomNamespace(ctx, c)
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(ctx.Err()).To(BeNil(), "test context cancelled or timed out")
|
|
cancel()
|
|
ctx, cancel = context.WithTimeout(context.Background(), defaultAfterTestTimeout)
|
|
defer cancel()
|
|
// clean up resources created by the tests
|
|
e2eutil.DeleteNamespace(ctx, c, namespace)
|
|
})
|
|
|
|
It("ThousandDeployments", func() {
|
|
thousandDeploymentsTest(ctx, c, invConfig, inventoryName, namespace.GetName())
|
|
})
|
|
|
|
It("ThousandNamespaces", func() {
|
|
thousandNamespacesTest(ctx, c, invConfig, inventoryName, namespace.GetName())
|
|
})
|
|
})
|