func TestReplaceTagWithSpace(t *testing.T) {
req := definition.Request{}
val := make(definition.Values)
val["param1"] = []string{"valParam"}
req.QueryStringParameters = val
cookie := make(definition.Cookies)
cookie["cookie1"] = "valCookie"
req.Cookies = cookie
res := definition.Response{}
res.Body = "Request {{request.query.param1}}. Cookie: {{request.cookie.cookie1}}. Random: {{fake.UserName}}"
faker := FakeDataParse{DummyDataFaker{"AleixMG"}}
faker.Parse(&req, &res)
if res.Body != "Request valParam. Cookie: valCookie. Random: AleixMG" {
t.Error("Replaced tags in body not match", res.Body)
}
}
func (this FakeDataParse) Parse(req *definition.Request, res *definition.Response) {
for header, values := range res.Headers {
for i, value := range values {
res.Headers[header][i] = this.replaceVars(req, value)
}
}
for cookie, value := range res.Cookies {
res.Cookies[cookie] = this.replaceVars(req, value)
}
res.Body = this.replaceVars(req, res.Body)
}