iago/iago_test.go at master · relab/iago · GitHub

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
package iago_test

import (
	"context"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/relab/iago"
	"github.com/relab/iago/iagotest"
)

func TestIago(t *testing.T) {
	dir := t.TempDir()
	wd, err := os.Getwd()
	if err != nil {
		t.Fatal(err)
	}

	g := iagotest.CreateSSHGroup(t, 4, false)

	g.ErrorHandler = func(e error) {
		t.Error(e)
	}

	g.Run("Custom Shell Command", func(ctx context.Context, host iago.Host) (err error) {
		var sb strings.Builder
		err = iago.Shell{
			Command: "lsb_release -a",
			Stdout:  &sb,
		}.Apply(ctx, host)
		if err != nil {
			return err
		}
		t.Log(sb.String())
		return nil
	})

	g.Run("Read distribution name", iago.Shell{Command: "grep '^ID=' /etc/os-release > $HOME/os"}.Apply)

	g.Run("Upload a file", func(ctx context.Context, host iago.Host) error {
		src, err := iago.NewPath(wd, "LICENSE")
		if err != nil {
			return err
		}
		dest, err := iago.NewPath(iago.Expand(host, "$HOME"), "LICENSE")
		if err != nil {
			return err
		}
		return iago.Upload{
			Src:  src,
			Dest: dest,
		}.Apply(ctx, host)
	})

	g.Run("Download files", func(ctx context.Context, host iago.Host) error {
		src, err := iago.NewPath(iago.Expand(host, "$HOME"), "os")
		if err != nil {
			return err
		}
		dest, err := iago.NewPath(dir, "os")
		if err != nil {
			return err
		}
		return iago.Download{
			Src:  src,
			Dest: dest,
		}.Apply(ctx, host)
	})

	for _, h := range g.Hosts {
		f, err := os.ReadFile(filepath.Join(dir, "os."+h.Name()))
		if err != nil {
			t.Fatal(err)
		}
		t.Log(string(f))
	}
}

func TestIagoDownloadExample(t *testing.T) {
	dir := t.TempDir()

	// The iagotest package provides a helper function that automatically
	// builds and starts docker containers with an exposed SSH port for testing.
	g := iagotest.CreateSSHGroup(t, 2, false)

	g.Run("Download files", func(ctx context.Context, host iago.Host) error {
		src, err := iago.NewPath("/etc", "os-release")
		if err != nil {
			return err
		}
		t.Logf("Downloading %s from %s", src, host.Name())
		dest, err := iago.NewPath(dir, "os")
		if err != nil {
			return err
		}
		t.Logf("Saving to %s", dest)
		return iago.Download{
			Src:  src,
			Dest: dest,
			Perm: iago.NewPerm(0o644),
		}.Apply(ctx, host)
	})

	filesInDir, err := os.ReadDir(dir)
	if err != nil {
		t.Fatal(err)
	}
	files := make([]string, len(filesInDir))
	for i, file := range filesInDir {
		files[i] = file.Name()
	}
	t.Logf("Copied files: %s", strings.Join(files, ", "))

	for _, h := range g.Hosts {
		t.Logf("Reading file %s for host %s", filepath.Join(dir, "os."+h.Name()), h.Name())
		f, err := os.ReadFile(filepath.Join(dir, "os."+h.Name()))
		if err != nil {
			t.Fatal(err)
		}
		t.Log(string(f))
	}
}