forked from sensorbee/sensorbee
/
common.go
114 lines (101 loc) · 2.59 KB
/
common.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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package testutil
import (
"bytes"
"github.com/mattn/go-scan"
"gopkg.in/pfnet/jasco.v1"
"gopkg.in/sensorbee/sensorbee.v0/data"
"gopkg.in/sensorbee/sensorbee.v0/server"
"gopkg.in/sensorbee/sensorbee.v0/server/config"
"io/ioutil"
"net/http"
"net/http/httptest"
)
var (
// TestAPIWithRealHTTPServer is a flag to switch the use of httptest.server.
// If this flag is true, all tests will be executed with a real HTTP server.
// If it is false, a server will not be created and tests are done by directly
// calling ServeHTTP of a handler.
TestAPIWithRealHTTPServer = false
)
// Server is a temporary HTTP server for test.
type Server struct {
// server has the information of the server
server struct {
realServer *httptest.Server
router http.Handler
url string
}
}
// Close closes the server.
func (s *Server) Close() {
if s.server.realServer != nil {
s.server.realServer.Close()
}
}
// URL returns the URL of the server.
func (s *Server) URL() string {
return s.server.url
}
// HTTPClient returns the HTTP client to send requests to the server.
func (s *Server) HTTPClient() *http.Client {
if s.server.realServer != nil {
return http.DefaultClient
}
return &http.Client{
Transport: &testTransport{s},
}
}
type testTransport struct {
s *Server
}
func (t *testTransport) RoundTrip(req *http.Request) (*http.Response, error) {
if req.Body == nil {
req.Body = ioutil.NopCloser(bytes.NewReader([]byte{}))
}
rc := httptest.NewRecorder()
t.s.server.router.ServeHTTP(rc, req)
return &http.Response{
Status: http.StatusText(rc.Code),
StatusCode: rc.Code,
Proto: "HTTP/1.0",
ProtoMajor: 1,
ProtoMinor: 0,
Header: rc.HeaderMap,
Body: ioutil.NopCloser(rc.Body),
ContentLength: int64(rc.Body.Len()),
}, nil
}
// NewServer returns a temporary running server.
func NewServer() *Server {
s := &Server{}
c, err := config.New(data.Map{})
if err != nil {
panic(err)
}
gvars, err := server.SetUpContextGlobalVariables(c)
if err != nil {
panic(err)
}
jascoRoot := jasco.New("/", nil)
root, err := server.SetUpContextAndRouter("/", jascoRoot, gvars)
if err != nil {
panic(err)
}
server.SetUpAPIRouter("/", root, nil)
if TestAPIWithRealHTTPServer {
s.server.realServer = httptest.NewServer(jascoRoot)
s.server.url = s.server.realServer.URL
} else {
s.server.router = jascoRoot
s.server.url = "http://172.0.0.1:0602"
}
return s
}
// JScan traverses a json object tree.
func JScan(js interface{}, path string) interface{} {
var v interface{}
if err := scan.ScanTree(js, path, &v); err != nil {
return nil
}
return v
}