-
Notifications
You must be signed in to change notification settings - Fork 0
/
integration.go
92 lines (81 loc) · 1.89 KB
/
integration.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
package integration
import (
"context"
"os"
"testing"
"time"
"github.com/containerd/containerd"
"github.com/containerd/containerd/defaults"
"github.com/containerd/containerd/namespaces"
"google.golang.org/grpc"
)
type (
Test struct {
t *testing.T
dependsOn []*Service
}
TestOption func(*Test)
)
var (
ctx = namespaces.NamespaceFromEnv(context.Background())
clientOpts = containerd.WithDialOpts([]grpc.DialOption{grpc.WithTimeout(time.Second * 2), grpc.WithInsecure()})
client *containerd.Client
)
func createGlobalClient() error {
if client == nil {
address := os.Getenv("CONTAINERD_ADDRESS")
if address == "" {
address = defaults.DefaultAddress
}
c, err := containerd.New(address, clientOpts)
if err != nil {
return err
}
client = c
}
return nil
}
func NewIntegrationTest(t *testing.T, opts ...TestOption) *Test {
it := &Test{t: t}
for _, o := range opts {
o(it)
}
err := createGlobalClient()
if err != nil {
t.Fatalf("[integration] couldn't create containerd client: %v", err)
}
return it
}
func (it *Test) Run(f func(ctx context.Context, t *testing.T)) {
for _, dep := range it.dependsOn {
running, err := dep.IsRunning()
if err != nil {
it.t.Fatalf("[integration] couldn't check if service is running: %v", err)
}
if running {
continue
}
if err := dep.Start(); err != nil {
it.t.Fatalf("[integration] couldn't create service: %v", err)
}
it.t.Logf("[integration] service %s is running", dep.name)
}
defer func() {
for _, dep := range it.dependsOn {
if !dep.cleanup {
continue
}
err := dep.Stop()
if err != nil {
it.t.Fatalf("[integration] couldn't stop service: %v", err)
}
it.t.Logf("[integration] service %s is stopped", dep.name)
}
}()
f(context.TODO(), it.t)
}
func DependsOn(svc *Service) TestOption {
return func(test *Test) {
test.dependsOn = append(test.dependsOn, svc)
}
}