controller-sdk-go/api/config_test.go at 17d5b081d7e1d26abd7e1cfdbd4a06af7fd2e6da · drycc/controller-sdk-go · 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
package api

import (
	"strings"
	"testing"
)

func TestHealthcheckString(t *testing.T) {
	h := Healthcheck{}

	expected := strings.TrimSpace(`Initial Delay (seconds): 0
Timeout (seconds): 0
Period (seconds): 0
Success Threshold: 0
Failure Threshold: 0
Exec Probe: N/A
HTTP GET Probe: N/A
TCP Socket Probe: N/A`)

	if strings.TrimSpace(h.String()) != expected {
		t.Errorf("Expected:\n\n%s\n\nGot:\n\n%s", expected, h.String())
	}

	h.HTTPGet = &HTTPGetProbe{
		Path:        "/",
		Port:        80,
		HTTPHeaders: []*KVPair{{Key: "X-DEIS-IS", Value: "AWESOME"}},
	}

	expected = strings.TrimSpace(`Initial Delay (seconds): 0
Timeout (seconds): 0
Period (seconds): 0
Success Threshold: 0
Failure Threshold: 0
Exec Probe: N/A
HTTP GET Probe: Path="/" Port=80 HTTPHeaders=[X-DEIS-IS=AWESOME]
TCP Socket Probe: N/A`)

	if strings.TrimSpace(h.String()) != expected {
		t.Errorf("Expected:\n\n%s\n\nGot:\n\n%s", expected, h.String())
	}

	h.Exec = &ExecProbe{Command: []string{"echo", "hi"}}

	h.TCPSocket = &TCPSocketProbe{
		Port: 80,
	}

	expected = strings.TrimSpace(`Initial Delay (seconds): 0
Timeout (seconds): 0
Period (seconds): 0
Success Threshold: 0
Failure Threshold: 0
Exec Probe: Command=[echo hi]
HTTP GET Probe: Path="/" Port=80 HTTPHeaders=[X-DEIS-IS=AWESOME]
TCP Socket Probe: Port=80`)

	if strings.TrimSpace(h.String()) != expected {
		t.Errorf("Expected:\n\n%s\n\nGot:\n\n%s", expected, h.String())
	}
}