-
Notifications
You must be signed in to change notification settings - Fork 3
/
script_test.go
76 lines (71 loc) · 1.66 KB
/
script_test.go
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
package georgia
import (
"net/http/httptest"
"testing"
"time"
"github.com/mattn/go-nulltype"
"github.com/stretchr/testify/assert"
"github.com/whitewater-guide/gorge/core"
"github.com/whitewater-guide/gorge/testutils"
)
func setupTestServer() *httptest.Server {
return testutils.SetupFileServer(nil, nil)
}
func TestGeorgia_ListGauges(t *testing.T) {
ts := setupTestServer()
defer ts.Close()
s := scriptGeorgia{
name: "georgia",
url: ts.URL + "/page.html",
}
actual, err := s.ListGauges()
expected := core.Gauge{
GaugeID: core.GaugeID{
Script: "georgia",
Code: "3c598e973566b29359a3821cf3dceceb",
},
LevelUnit: "cm",
Name: "Acharistskali - keda",
URL: ts.URL + "/page.html",
Timezone: "Asia/Tbilisi",
}
if assert.NoError(t, err) {
assert.Len(t, actual, 20)
assert.Contains(t, actual, expected)
}
}
func TestGeorgia_Harvest(t *testing.T) {
ts := setupTestServer()
defer ts.Close()
s := scriptGeorgia{
name: "georgia",
url: ts.URL + "/page.html",
}
now := time.Now().UTC().Truncate(time.Hour)
actual, err := core.HarvestSlice(&s, core.StringSet{}, 0)
expected := &core.Measurement{
GaugeID: core.GaugeID{
Script: "georgia",
Code: "3c598e973566b29359a3821cf3dceceb",
},
Timestamp: core.HTime{
Time: now,
},
Level: nulltype.NullFloat64Of(64),
}
expected2 := &core.Measurement{
GaugeID: core.GaugeID{
Script: "georgia",
Code: "89f8e0acaa48a39321c3d7f69ac5b5ad",
},
Timestamp: core.HTime{
Time: now,
},
Level: nulltype.NullFloat64Of(156),
}
if assert.NoError(t, err) {
assert.Len(t, actual, 20)
assert.Contains(t, actual, expected)
assert.Contains(t, actual, expected2)
}
}