|
9 | 9 | "time"
|
10 | 10 |
|
11 | 11 | "github.com/go-check/check"
|
| 12 | + "github.com/iovisor/kubectl-trace/pkg/version" |
12 | 13 | "gotest.tools/icmd"
|
13 | 14 | "sigs.k8s.io/kind/pkg/cluster"
|
14 | 15 | "sigs.k8s.io/kind/pkg/cluster/config/encoding"
|
@@ -46,10 +47,22 @@ func (k *KubectlTraceSuite) SetUpSuite(c *check.C) {
|
46 | 47 |
|
47 | 48 | clusterName, err := generateClusterName()
|
48 | 49 | c.Assert(err, check.IsNil)
|
49 |
| - ctx := cluster.NewContext(clusterName) |
50 |
| - err = ctx.Create(cfg, retain, wait) |
| 50 | + kctx := cluster.NewContext(clusterName) |
| 51 | + |
| 52 | + err = kctx.Create(cfg, retain, wait) |
| 53 | + c.Assert(err, check.IsNil) |
| 54 | + k.kindContext = kctx |
| 55 | + |
| 56 | + nodes, err := kctx.ListNodes() |
| 57 | + |
51 | 58 | c.Assert(err, check.IsNil)
|
52 |
| - k.kindContext = ctx |
| 59 | + |
| 60 | + // copy the bpftrace image to the nodes |
| 61 | + for _, n := range nodes { |
| 62 | + loadcomm := fmt.Sprintf("docker save %s | docker exec -i %s docker load", version.ImageNameTag(), n.String()) |
| 63 | + res := icmd.RunCommand("bash", "-c", loadcomm) |
| 64 | + c.Assert(res.Error, check.IsNil) |
| 65 | + } |
53 | 66 | }
|
54 | 67 |
|
55 | 68 | func (s *KubectlTraceSuite) TearDownSuite(c *check.C) {
|
|
0 commit comments