wavefront-sdk-go/senders/live_test.go at main · bvsvas/wavefront-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
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
package senders

import (
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func skipUnlessVarsAreSet(t *testing.T) {
	if os.Getenv("LIVE_TEST_HOST") == "" {
		t.Skip()
	}
}

func TestCSP_LIVE(t *testing.T) {
	skipUnlessVarsAreSet(t)

	sender, err := NewSender(os.Getenv("LIVE_TEST_HOST"),
		CSPClientCredentials(
			os.Getenv("LIVE_TEST_CSP_CLIENT_ID"),
			os.Getenv("LIVE_TEST_CSP_CLIENT_SECRET"),
			CSPBaseURL(os.Getenv("LIVE_TEST_CSP_BASE_URL")),
		))
	assert.NoError(t, err)
	assert.NoError(t, sender.SendMetric("test.go-metrics.can-send", 1, 0, "go test",
		map[string]string{"scenario": "direct-csp-server-to-server"}))
	assert.NoError(t, sender.Flush())
	sender.Close()
}

func TestCSP_API_TOKEN_LIVE(t *testing.T) {
	skipUnlessVarsAreSet(t)

	sender, err := NewSender(os.Getenv("LIVE_TEST_HOST"),
		CSPAPIToken(
			os.Getenv("LIVE_TEST_CSP_API_TOKEN"),
			CSPBaseURL(os.Getenv("LIVE_TEST_CSP_BASE_URL")),
		))
	assert.NoError(t, err)
	assert.NoError(t, sender.SendMetric("test.go-metrics.can-send", 1, 0, "go test",
		map[string]string{"scenario": "direct-csp-api-token"}))
	assert.NoError(t, sender.Flush())
	sender.Close()
}

func TestWF_API_TOKEN_LIVE(t *testing.T) {
	skipUnlessVarsAreSet(t)

	sender, err := NewSender(
		os.Getenv("LIVE_TEST_HOST"),
		APIToken(os.Getenv("LIVE_TEST_WF_API_TOKEN")),
	)
	assert.NoError(t, err)
	assert.NoError(t, sender.SendMetric("test.go-metrics.can-send", 1, 0, "go test",
		map[string]string{"scenario": "direct-wf-token"}))
	assert.NoError(t, sender.Flush())
	sender.Close()
}

func TestEventDirectSend_LIVE(t *testing.T) {
	skipUnlessVarsAreSet(t)

	sender, err := NewSender(
		os.Getenv("LIVE_TEST_HOST"),
		APIToken(os.Getenv("LIVE_TEST_WF_API_TOKEN")),
	)
	assert.NoError(t, err)
	assert.NoError(t, sender.SendEvent("test.an-event", time.Now().Add(-30*time.Minute).UnixMilli(), time.Now().Add(5*time.Minute).UnixMilli(), "go test",
		map[string]string{"scenario": "send-event-direct"}))
	assert.NoError(t, sender.Flush())
	sender.Close()
}

func TestEventProxySend_LIVE(t *testing.T) {
	skipUnlessVarsAreSet(t)

	sender, err := NewSender(
		os.Getenv("LIVE_TEST_PROXY_HOST"),
	)
	assert.NoError(t, err)
	assert.NoError(t, sender.SendEvent("test.an-event", time.Now().Add(-30*time.Minute).UnixMilli(), time.Now().Add(5*time.Minute).UnixMilli(), "go test",
		map[string]string{"scenario": "send-event-proxy"}))
	assert.NoError(t, sender.Flush())
	sender.Close()
}