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
|
package integration
import (
"os"
"testing"
"github.com/stretchr/testify/assert"
"github.com/zorkian/go-datadog-api"
)
func TestInvalidAuth(t *testing.T) {
// Override the correct credentials
c := datadog.NewClient("INVALID", "INVALID")
valid, err := c.Validate()
if err != nil {
t.Fatalf("Testing authentication failed when it shouldn't: %s", err)
}
assert.Equal(t, valid, false)
}
func TestValidAuth(t *testing.T) {
valid, err := client.Validate()
if err != nil {
t.Fatalf("Testing authentication failed when it shouldn't: %s", err)
}
assert.Equal(t, valid, true)
}
func TestValidAuthNoAppKey(t *testing.T) {
c := datadog.NewClient(os.Getenv("DATADOG_API_KEY"), "")
valid, err := c.Validate()
if err != nil {
t.Fatalf("Testing authentication failed when it shouldn't: %s", err)
}
assert.Equal(t, valid, true)
}
func TestBaseUrl(t *testing.T) {
t.Run("Base url defaults to https://api.datadoghq.com", func(t *testing.T) {
assert.Empty(t, os.Getenv("DATADOG_HOST"))
c := datadog.NewClient("abc", "def")
assert.Equal(t, "https://api.datadoghq.com", c.GetBaseUrl())
})
t.Run("Base url defaults DATADOG_HOST environment variable if set", func(t *testing.T) {
os.Setenv("DATADOG_HOST", "https://custom.datadoghq.com")
defer os.Unsetenv("DATADOG_HOST")
c := datadog.NewClient("abc", "def")
assert.Equal(t, "https://custom.datadoghq.com", c.GetBaseUrl())
})
t.Run("Base url can be set through the attribute setter", func(t *testing.T) {
c := datadog.NewClient("abc", "def")
c.SetBaseUrl("https://another.datadoghq.com")
assert.Equal(t, "https://another.datadoghq.com", c.GetBaseUrl())
})
}
func TestExtraHeader(t *testing.T) {
t.Run("No Extra Header for backward compatibility", func(t *testing.T) {
c := datadog.NewClient("foo", "bar")
assert.Empty(t, c.ExtraHeader)
})
}
func TestInsertExtraHeader(t *testing.T) {
t.Run("ExtraHeader map should be initialised", func(t *testing.T) {
c := datadog.NewClient("foo", "bar")
c.ExtraHeader["foo"] = "bar"
assert.Equal(t, c.ExtraHeader["foo"], "bar")
})
}
|
