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
|
package integration
import (
"context"
"testing"
"time"
"github.com/linode/linodego"
)
func TestNodeBalancerStats_Get(t *testing.T) {
client, nodebalancer, teardown, err := setupNodeBalancer(t, "fixtures/TestNodeBalancerStats_Get", nil)
defer teardown()
if err != nil {
t.Error(err)
}
ticker := time.NewTicker(10 * time.Second)
timer := time.NewTimer(120 * time.Second)
defer ticker.Stop()
poll:
for {
select {
case <-ticker.C:
_, err = client.GetNodeBalancerStats(context.Background(), nodebalancer.ID)
if err != nil {
// Possible that the call succeeded but that stats aren't available (HTTP: 4XX)
if v, ok := err.(*linodego.Error); ok {
if v.Code == 400 && v.Message == "Stats are unavailable at this time." {
break poll
}
// Otherwise, let's call it fatal
t.Fatal(err)
}
}
if err == nil { // stats are now returning
break poll
}
case <-timer.C:
t.Fatal("Error getting stats, polling timed out")
}
}
}
|
