/
mw_transform_test.go
47 lines (43 loc) · 1.25 KB
/
mw_transform_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
package main
import (
"io/ioutil"
"strings"
"testing"
"text/template"
"github.com/TykTechnologies/tyk/apidef"
)
func TestTransformNonAscii(t *testing.T) {
in := `<?xml version="1.0" encoding="utf-8"?>
<names>
<name>Jyväskylä</name>
<name>Hyvinkää</name>
</names>`
want := `["Jyväskylä", "Hyvinkää"]`
tmpl := `[{{range $x, $s := .names.name}}"{{$s}}"{{if not $x}}, {{end}}{{end}}]`
r := testReq(t, "GET", "/", in)
tmeta := &TransformSpec{}
tmeta.TemplateData.Input = apidef.RequestXML
tmeta.Template = template.Must(template.New("blob").Parse(tmpl))
if err := transformBody(r, tmeta, false); err != nil {
t.Fatalf("wanted nil error, got %v", err)
}
gotBs, err := ioutil.ReadAll(r.Body)
if err != nil {
t.Fatal(err)
}
if got := string(gotBs); got != want {
t.Fatalf("wanted body %q, got %q", want, got)
}
}
func TestTransformXMLCrash(t *testing.T) {
// mxj.NewMapXmlReader used to take forever and crash the
// process by eating up all the memory.
in := strings.NewReader("")
r := testReq(t, "GET", "/", in)
tmeta := &TransformSpec{}
tmeta.TemplateData.Input = apidef.RequestXML
tmeta.Template = template.Must(template.New("blob").Parse(""))
if err := transformBody(r, tmeta, false); err == nil {
t.Fatalf("wanted error, got nil")
}
}