...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package provider
17
18 import (
19 "context"
20 "os"
21 "path/filepath"
22 "testing"
23 "time"
24
25 "github.com/go-logr/logr"
26 . "github.com/onsi/ginkgo/v2"
27 . "github.com/onsi/gomega"
28 "go.uber.org/fx"
29 k8sScheme "k8s.io/client-go/kubernetes/scheme"
30 "k8s.io/client-go/rest"
31 ctrl "sigs.k8s.io/controller-runtime"
32 "sigs.k8s.io/controller-runtime/pkg/client"
33 "sigs.k8s.io/controller-runtime/pkg/envtest"
34 logf "sigs.k8s.io/controller-runtime/pkg/log"
35
36 "github.com/chaos-mesh/chaos-mesh/api/v1alpha1"
37 "github.com/chaos-mesh/chaos-mesh/controllers/utils/test/manager"
38 "github.com/chaos-mesh/chaos-mesh/pkg/log"
39 )
40
41
42
43
44 var app *fx.App
45 var cfg *rest.Config
46 var k8sClient client.Client
47 var mgr ctrl.Manager
48 var testEnv *envtest.Environment
49 var setupLog = ctrl.Log.WithName("setup")
50
51 func TestProvider(t *testing.T) {
52 RegisterFailHandler(Fail)
53
54 RunSpecs(t, "Provider suit")
55 }
56
57 var _ = BeforeSuite(func(ctx SpecContext) {
58 logf.SetLogger(log.NewZapLoggerWithWriter(GinkgoWriter))
59 By("bootstrapping test environment")
60 t := true
61 if os.Getenv("USE_EXISTING_CLUSTER") == "true" {
62 testEnv = &envtest.Environment{
63 UseExistingCluster: &t,
64 }
65 } else {
66 testEnv = &envtest.Environment{
67 CRDDirectoryPaths: []string{filepath.Join("..", "..", "config", "crd", "bases")},
68 }
69 }
70
71 err := v1alpha1.SchemeBuilder.AddToScheme(k8sScheme.Scheme)
72 Expect(err).NotTo(HaveOccurred())
73
74 cfg, err = testEnv.Start()
75 Expect(err).ToNot(HaveOccurred())
76 Expect(cfg).ToNot(BeNil())
77
78 rootLogger, err := log.NewDefaultZapLogger()
79 Expect(err).ToNot(HaveOccurred())
80
81 app = fx.New(
82 fx.Options(
83 fx.Supply(cfg),
84 fx.Supply(rootLogger),
85 fx.Provide(
86 NewOption,
87 NewClient,
88 manager.NewTestManager,
89 NewAuthCli,
90 NewScheme,
91 ),
92 ),
93 fx.Populate(&k8sClient),
94 fx.Populate(&mgr),
95 fx.Invoke(Run),
96 )
97 startCtx, cancel := context.WithTimeout(context.Background(), app.StartTimeout())
98 defer cancel()
99 Expect(err).ToNot(HaveOccurred())
100 Expect(k8sClient).ToNot(BeNil())
101
102 if err := app.Start(startCtx); err != nil {
103 setupLog.Error(err, "fail to start manager")
104 }
105 Expect(err).ToNot(HaveOccurred())
106
107 }, NodeTimeout(60*time.Second))
108
109 var _ = AfterSuite(func() {
110 By("tearing down the test environment")
111 stopCtx, cancel := context.WithTimeout(context.Background(), app.StopTimeout())
112 defer cancel()
113
114 if err := app.Stop(stopCtx); err != nil {
115 setupLog.Error(err, "fail to stop manager")
116 }
117 err := testEnv.Stop()
118 Expect(err).ToNot(HaveOccurred())
119 })
120
121 type RunParams struct {
122 fx.In
123
124 Mgr ctrl.Manager
125 Logger logr.Logger
126 Client client.Client
127 }
128
129 func Run(params RunParams) error {
130 return nil
131 }
132