|
@@ -0,0 +1,372 @@
|
|
|
|
+package sinks
|
|
|
|
+
|
|
|
|
+import (
|
|
|
|
+ "encoding/json"
|
|
|
|
+ "fmt"
|
|
|
|
+ "github.com/emqx/kuiper/common"
|
|
|
|
+ "github.com/emqx/kuiper/xstream/contexts"
|
|
|
|
+ "io/ioutil"
|
|
|
|
+ "net/http"
|
|
|
|
+ "net/http/httptest"
|
|
|
|
+ "reflect"
|
|
|
|
+ "testing"
|
|
|
|
+)
|
|
|
|
+
|
|
|
|
+type request struct{
|
|
|
|
+ Method string
|
|
|
|
+ Body string
|
|
|
|
+ ContentType string
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+func TestRestSink_Apply(t *testing.T) {
|
|
|
|
+ var tests = []struct {
|
|
|
|
+ config map[string]interface{}
|
|
|
|
+ data []map[string]interface{}
|
|
|
|
+ result []request
|
|
|
|
+ }{
|
|
|
|
+ {
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"ab":"hello1"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ },{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"ab":"hello2"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `[{"ab":"hello1"},{"ab":"hello2"}]`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "get",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "GET",
|
|
|
|
+ ContentType: "",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "put",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "text",
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "PUT",
|
|
|
|
+ ContentType: "text/plain",
|
|
|
|
+ Body: `[{"ab":"hello1"},{"ab":"hello2"}]`,
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "form",
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `result=%5B%7B%22ab%22%3A%22hello1%22%7D%2C%7B%22ab%22%3A%22hello2%22%7D%5D`,
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "form",
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `ab=hello1`,
|
|
|
|
+ },{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `ab=hello2`,
|
|
|
|
+ }},
|
|
|
|
+ }, {
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "json",
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ "timeout": float64(1000),
|
|
|
|
+ },
|
|
|
|
+ data: []map[string]interface{}{{
|
|
|
|
+ "ab": "hello1",
|
|
|
|
+ }, {
|
|
|
|
+ "ab": "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"ab":"hello1"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }, {
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"ab":"hello2"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },
|
|
|
|
+ }
|
|
|
|
+ fmt.Printf("The test bucket size is %d.\n\n", len(tests))
|
|
|
|
+ contextLogger := common.Log.WithField("rule", "TestRestSink_Apply")
|
|
|
|
+ ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
|
|
|
|
+
|
|
|
|
+ var requests []request
|
|
|
|
+ ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request){
|
|
|
|
+ body, err := ioutil.ReadAll(r.Body)
|
|
|
|
+ if err != nil {
|
|
|
|
+ fmt.Printf("Error reading body: %v", err)
|
|
|
|
+ http.Error(w, "can't read body", http.StatusBadRequest)
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ requests = append(requests, request{
|
|
|
|
+ Method: r.Method,
|
|
|
|
+ Body: string(body),
|
|
|
|
+ ContentType: r.Header.Get("Content-Type"),
|
|
|
|
+ })
|
|
|
|
+ contextLogger.Debugf(string(body))
|
|
|
|
+ fmt.Fprintf(w, string(body))
|
|
|
|
+ }))
|
|
|
|
+ defer ts.Close()
|
|
|
|
+ for i, tt := range tests {
|
|
|
|
+ requests = nil
|
|
|
|
+ s := &RestSink{}
|
|
|
|
+ tt.config["url"] = ts.URL
|
|
|
|
+ s.Configure(tt.config)
|
|
|
|
+ s.Open(ctx)
|
|
|
|
+ input, err := json.Marshal(tt.data)
|
|
|
|
+ if err != nil{
|
|
|
|
+ t.Errorf("Failed to parse the input into []byte]")
|
|
|
|
+ continue
|
|
|
|
+ }
|
|
|
|
+ s.Collect(ctx, input)
|
|
|
|
+ s.Close(ctx)
|
|
|
|
+ if !reflect.DeepEqual(tt.result, requests) {
|
|
|
|
+ t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.result, requests)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+func TestRestSinkTemplate_Apply(t *testing.T) {
|
|
|
|
+ var tests = []struct {
|
|
|
|
+ config map[string]interface{}
|
|
|
|
+ data []map[string]interface{}
|
|
|
|
+ result []request
|
|
|
|
+ }{
|
|
|
|
+ {
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ "dataTemplate": `{"wrapper":"w1","content":{{json .}},"ab":"{{.ab}}"}`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"wrapper":"w1","content":{"ab":"hello1"},"ab":"hello1"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ },{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"wrapper":"w1","content":{"ab":"hello2"},"ab":"hello2"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "dataTemplate": `{"wrapper":"arr","content":{{json .}},"content0":{{json (index . 0)}},ab0":"{{index . 0 "ab"}}"}`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"wrapper":"arr","content":[{"ab":"hello1"},{"ab":"hello2"}],"content0":{"ab":"hello1"},ab0":"hello1"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "get",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "dataTemplate": `{"wrapper":"w1","content":{{json .}},"ab":"{{.ab}}"}`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "GET",
|
|
|
|
+ ContentType: "",
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "put",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "html",
|
|
|
|
+ "dataTemplate": `<div>results</div><ul>{{range .}}<li>{{.ab}}</li>{{end}}</ul>`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "PUT",
|
|
|
|
+ ContentType: "text/html",
|
|
|
|
+ Body: `<div>results</div><ul><li>hello1</li><li>hello2</li></ul>`,
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "form",
|
|
|
|
+ "dataTemplate": `{"content":{{json .}}}`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `content=%5B%7B%22ab%22%3A%22hello1%22%7D%2C%7B%22ab%22%3A%22hello2%22%7D%5D`,
|
|
|
|
+ }},
|
|
|
|
+ },{
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "form",
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ "dataTemplate": `{"newab":"{{.ab}}"}`,
|
|
|
|
+ },
|
|
|
|
+ data:[]map[string]interface{}{{
|
|
|
|
+ "ab" : "hello1",
|
|
|
|
+ },{
|
|
|
|
+ "ab" : "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `newab=hello1`,
|
|
|
|
+ },{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ ContentType: "application/x-www-form-urlencoded;param=value",
|
|
|
|
+ Body: `newab=hello2`,
|
|
|
|
+ }},
|
|
|
|
+ }, {
|
|
|
|
+ config: map[string]interface{}{
|
|
|
|
+ "method": "post",
|
|
|
|
+ //"url": "http://localhost/test", //set dynamically to the test server
|
|
|
|
+ "bodyType": "json",
|
|
|
|
+ "sendSingle": true,
|
|
|
|
+ "timeout": float64(1000),
|
|
|
|
+ "dataTemplate": `{"newab":"{{.ab}}"}`,
|
|
|
|
+ },
|
|
|
|
+ data: []map[string]interface{}{{
|
|
|
|
+ "ab": "hello1",
|
|
|
|
+ }, {
|
|
|
|
+ "ab": "hello2",
|
|
|
|
+ }},
|
|
|
|
+ result: []request{{
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"newab":"hello1"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }, {
|
|
|
|
+ Method: "POST",
|
|
|
|
+ Body: `{"newab":"hello2"}`,
|
|
|
|
+ ContentType: "application/json",
|
|
|
|
+ }},
|
|
|
|
+ },
|
|
|
|
+ }
|
|
|
|
+ fmt.Printf("The test bucket size is %d.\n\n", len(tests))
|
|
|
|
+ contextLogger := common.Log.WithField("rule", "TestRestSink_Apply")
|
|
|
|
+ ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
|
|
|
|
+
|
|
|
|
+ var requests []request
|
|
|
|
+ ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request){
|
|
|
|
+ body, err := ioutil.ReadAll(r.Body)
|
|
|
|
+ if err != nil {
|
|
|
|
+ fmt.Printf("Error reading body: %v", err)
|
|
|
|
+ http.Error(w, "can't read body", http.StatusBadRequest)
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ requests = append(requests, request{
|
|
|
|
+ Method: r.Method,
|
|
|
|
+ Body: string(body),
|
|
|
|
+ ContentType: r.Header.Get("Content-Type"),
|
|
|
|
+ })
|
|
|
|
+ contextLogger.Debugf(string(body))
|
|
|
|
+ fmt.Fprintf(w, string(body))
|
|
|
|
+ }))
|
|
|
|
+ defer ts.Close()
|
|
|
|
+ for i, tt := range tests {
|
|
|
|
+ requests = nil
|
|
|
|
+ s := &RestSink{}
|
|
|
|
+ tt.config["url"] = ts.URL
|
|
|
|
+ s.Configure(tt.config)
|
|
|
|
+ s.Open(ctx)
|
|
|
|
+ input, err := json.Marshal(tt.data)
|
|
|
|
+ if err != nil{
|
|
|
|
+ t.Errorf("Failed to parse the input into []byte]")
|
|
|
|
+ continue
|
|
|
|
+ }
|
|
|
|
+ s.Collect(ctx, input)
|
|
|
|
+ s.Close(ctx)
|
|
|
|
+ if !reflect.DeepEqual(tt.result, requests) {
|
|
|
|
+ t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.result, requests)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+}
|