Files
upstream-ory-kratos/ui/container/container_test.go
upstream-mirror c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
snapshot of https://github.com/ory/kratos.git (master, 2026-09-15; blobs > 131072B omitted)
2026-09-15 21:19:12 +05:00

336 lines
13 KiB
Go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package container
import (
"bytes"
"context"
"encoding/json"
"fmt"
"net/http"
"net/url"
"sort"
"testing"
"github.com/ory/kratos/ui/node"
"github.com/pkg/errors"
"github.com/ory/jsonschema/v3"
"github.com/ory/herodot"
"github.com/ory/x/decoderx"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/ory/kratos/schema"
"github.com/ory/kratos/text"
)
func newJSONRequest(t *testing.T, j string) *http.Request {
req, err := http.NewRequest("POST", "/", bytes.NewBufferString(j))
require.NoError(t, err)
req.Header.Set("Content-Type", "application/json")
return req
}
func newFormRequest(t *testing.T, values url.Values) *http.Request {
req, err := http.NewRequest("POST", "/", bytes.NewBufferString(values.Encode()))
require.NoError(t, err)
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
return req
}
func TestContainer(t *testing.T) {
ctx := context.Background()
t.Run("method=NewFromJSON", func(t *testing.T) {
for k, tc := range []struct {
r string
prefix string
expect *Container
}{
{
r: `{"numby":1.5,"stringy":"foobar","objy":{"objy":{},"numby":1.5,"stringy":"foobar"}}`,
expect: &Container{
Nodes: node.Nodes{
node.NewInputFieldFromJSON("numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.stringy", "foobar", node.DefaultGroup),
node.NewInputFieldFromJSON("stringy", "foobar", node.DefaultGroup),
},
},
},
{
r: `{"numby":1.5,"stringy":"foobar","objy":{"objy":{},"numby":1.5,"stringy":"foobar"}}`,
prefix: "traits",
expect: &Container{
Nodes: node.Nodes{
node.NewInputFieldFromJSON("traits.numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("traits.objy.numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("traits.objy.stringy", "foobar", node.DefaultGroup),
node.NewInputFieldFromJSON("traits.stringy", "foobar", node.DefaultGroup),
},
},
},
} {
t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
actual := NewFromJSON("action", node.DefaultGroup, json.RawMessage(tc.r), tc.prefix)
// sort actual.fields lexicographically to have a deterministic order
sort.SliceStable(actual.Nodes, func(i, j int) bool {
return actual.Nodes[i].ID() < actual.Nodes[j].ID()
})
assert.Equal(t, "action", actual.Action)
assert.EqualValues(t, tc.expect.Nodes, actual.Nodes)
})
}
})
t.Run("method=NewFromHTTPRequest", func(t *testing.T) {
for k, tc := range []struct {
ref string
r *http.Request
expect *Container
}{
{
ref: "./stub/simple.schema.json",
r: newJSONRequest(t, `{"numby":1.5,"stringy":"foobar","objy":{"objy":{},"numby":1.5,"stringy":"foobar"}}`),
expect: &Container{
Nodes: node.Nodes{
node.NewInputFieldFromJSON("numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.stringy", "foobar", node.DefaultGroup),
node.NewInputFieldFromJSON("stringy", "foobar", node.DefaultGroup),
},
},
},
{
ref: "./stub/simple.schema.json",
r: newFormRequest(t, url.Values{
"numby": {"1.5"},
"stringy": {"foobar"},
"objy.numby": {"1.5"},
"objy.stringy": {"foobar"},
}),
expect: &Container{
Nodes: node.Nodes{
node.NewInputFieldFromJSON("numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.numby", 1.5, node.DefaultGroup),
node.NewInputFieldFromJSON("objy.stringy", "foobar", node.DefaultGroup),
node.NewInputFieldFromJSON("stringy", "foobar", node.DefaultGroup),
},
},
},
// FIXME https://github.com/ory/kratos/issues/1316
//
//{
// ref: "./stub/complex.schema.json",
// r: newFormRequest(t, url.Values{
// "meal.chef": {"aeneas"},
// }),
// expect: &Container{
// Nodes: node.Nodes{
// node.NewInputFieldFromJSON("meal.chef", "aeneas", node.DefaultGroup),
// &node.Node{Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "meal.name", Type: node.InputAttributeTypeText}, Messages: text.Messages{*text.NewValidationErrorRequired("name")}, Meta: &node.Meta{}},
// },
// },
//},
} {
t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
actual, err := NewFromHTTPRequest(tc.r, node.DefaultGroup, "action", decoderx.HTTPJSONSchemaCompiler(tc.ref, nil))
require.NoError(t, err)
// sort actual.fields lexicographically to have a deterministic order
sort.SliceStable(actual.Nodes, func(i, j int) bool {
return actual.Nodes[i].ID() < actual.Nodes[j].ID()
})
assert.Equal(t, "action", actual.Action)
assert.EqualValues(t, tc.expect.Nodes, actual.Nodes)
})
}
})
t.Run("method=NewFromJSONSchema", func(t *testing.T) {
for k, tc := range []struct {
ref string
prefix string
expect *Container
}{
{
ref: "./stub/simple.schema.json",
prefix: "",
expect: &Container{
Nodes: node.Nodes{
node.NewInputField("numby", nil, node.DefaultGroup, node.InputAttributeTypeNumber),
node.NewInputField("objy.numby", nil, node.DefaultGroup, node.InputAttributeTypeNumber),
node.NewInputField("objy.stringy", nil, node.DefaultGroup, node.InputAttributeTypeText),
node.NewInputField("stringy", nil, node.DefaultGroup, node.InputAttributeTypeText),
},
},
},
{
ref: "./stub/simple.schema.json",
prefix: "traits",
expect: &Container{
Nodes: node.Nodes{
node.NewInputField("traits.numby", nil, node.DefaultGroup, node.InputAttributeTypeNumber),
node.NewInputField("traits.objy.numby", nil, node.DefaultGroup, node.InputAttributeTypeNumber),
node.NewInputField("traits.objy.stringy", nil, node.DefaultGroup, node.InputAttributeTypeText),
node.NewInputField("traits.stringy", nil, node.DefaultGroup, node.InputAttributeTypeText),
},
},
},
{
ref: "./stub/complex.schema.json",
expect: &Container{
Nodes: node.Nodes{
node.NewInputField("fruits", nil, node.DefaultGroup, node.InputAttributeTypeText),
node.NewInputField("meal.chef", nil, node.DefaultGroup, node.InputAttributeTypeText),
node.NewInputField("meal.name", nil, node.DefaultGroup, node.InputAttributeTypeText, node.WithRequiredInputAttribute),
// FIXME https://github.com/ory/kratos/issues/1316
//
//node.NewInputField("vegetables.veggieName", nil, node.DefaultGroup, node.InputAttributeTypeText),
//node.NewInputField("vegetables.veggieLike", nil, node.DefaultGroup, node.InputAttributeTypeText),
//node.NewInputField("vegetables.veggieAmount", nil, node.DefaultGroup, node.InputAttributeTypeText),
},
},
},
} {
t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
actual, err := NewFromJSONSchema(ctx, "action",
node.DefaultGroup, tc.ref, tc.prefix, nil, false)
require.NoError(t, err)
assert.Equal(t, "action", actual.Action)
assert.EqualValues(t, tc.expect.Messages, actual.Messages)
assert.EqualValues(t, tc.expect.Nodes, actual.Nodes)
})
}
})
t.Run("method=ParseError", func(t *testing.T) {
for k, tc := range []struct {
err error
expectErr bool
expect Container
}{
{err: errors.New("foo"), expectErr: true},
{err: herodot.ErrNotFound(), expectErr: true},
{err: herodot.ErrBadRequest().WithReason("tests"), expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewValidationErrorGeneric("tests")}}},
{err: schema.NewInvalidCredentialsError(), expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewErrorValidationInvalidCredentials()}}},
{err: &jsonschema.ValidationError{Message: "test", InstancePtr: "#/foo/bar/baz"}, expect: Container{Nodes: node.Nodes{
&node.Node{Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "foo.bar.baz", Type: node.InputAttributeTypeText}, Messages: text.Messages{*text.NewValidationErrorGeneric("test")}, Meta: new(node.Meta)},
}}},
{err: &jsonschema.ValidationError{Message: "test", InstancePtr: ""}, expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewValidationErrorGeneric("test")}}},
{err: &jsonschema.ValidationError{Message: `"asd" is not valid "email"`, InstancePtr: "", SchemaPtr: "#/email/format"}, expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewErrorValidationEmail("asd")}}},
{err: &jsonschema.ValidationError{Message: `"asd" is not valid "tel"`, InstancePtr: "", SchemaPtr: "#/phone/format"}, expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewErrorValidationPhone("asd")}}},
{err: &jsonschema.ValidationError{Message: `"asd" is not valid "unknown-format"`, InstancePtr: "", SchemaPtr: "#/unknown/format"}, expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewValidationErrorGeneric(`"asd" is not valid "unknown-format"`)}}},
// Malformed format error message
{err: &jsonschema.ValidationError{Message: `"asd" is not valid "unknown-format`, InstancePtr: "", SchemaPtr: "#/unknown/format"}, expect: Container{Nodes: node.Nodes{}, Messages: text.Messages{*text.NewValidationErrorGeneric(`"asd" is not valid "unknown-format`)}}},
} {
t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
for _, in := range []error{tc.err, errors.WithStack(tc.err)} {
c := New("")
err := c.ParseError(node.DefaultGroup, in)
if tc.expectErr {
require.Error(t, err)
return
}
require.NoError(t, err)
assert.EqualValues(t, tc.expect.Messages, c.Messages)
assert.EqualValues(t, tc.expect.Nodes, c.Nodes)
}
})
}
})
t.Run("method=SetValue", func(t *testing.T) {
c := Container{
Nodes: node.Nodes{
node.NewInputField("1", "foo", node.DefaultGroup, node.InputAttributeTypeText),
node.NewInputField("2", "", node.DefaultGroup, node.InputAttributeTypeText),
},
}
assert.Len(t, c.Nodes, 2)
c.SetValue("1", node.NewInputFieldFromJSON("1", "baz1", node.DefaultGroup))
c.SetValue("2", node.NewInputFieldFromJSON("2", "baz2", node.DefaultGroup))
c.SetValue("3", node.NewInputFieldFromJSON("3", "baz3", node.DefaultGroup))
assert.Len(t, c.Nodes, 3)
for _, k := range []string{"1", "2", "3"} {
assert.EqualValues(t, fmt.Sprintf("baz%s", k), c.Nodes.Find(k).Attributes.GetValue(), "%+v", c)
}
})
t.Run("method=SetCSRF", func(t *testing.T) {
f := &Container{Nodes: node.Nodes{}}
f.SetCSRF("csrf-token")
assert.Contains(t, f.Nodes, node.NewCSRFNode("csrf-token"))
})
t.Run("method=AddMessage", func(t *testing.T) {
c := Container{
Nodes: node.Nodes{
&node.Node{Messages: text.Messages{*text.NewValidationErrorGeneric("foo")}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "1", Type: node.InputAttributeTypeText, FieldValue: ""}},
&node.Node{Messages: text.Messages{}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "2", Type: node.InputAttributeTypeText, FieldValue: ""}},
},
}
assert.Len(t, c.Nodes, 2)
c.AddMessage(node.DefaultGroup, &text.Message{Text: "baz1"}, "1")
c.AddMessage(node.DefaultGroup, &text.Message{Text: "baz2"}, "2")
c.AddMessage(node.DefaultGroup, &text.Message{Text: "baz3"}, "3")
c.AddMessage(node.DefaultGroup, &text.Message{Text: "baz"}, "4", "5", "6")
c.AddMessage(node.DefaultGroup, &text.Message{Text: "rootbar"})
assert.Len(t, c.Nodes, 6)
for _, k := range []string{"1", "2", "3"} {
n := c.Nodes.Find(k)
assert.EqualValues(t, fmt.Sprintf("baz%s", k), n.Messages[len(n.Messages)-1].Text, "%+v", c)
}
for _, k := range []string{"4", "5", "6"} {
assert.EqualValues(t, "baz", c.Nodes.Find(k).Messages[0].Text, "%+v", c)
}
assert.Len(t, c.Messages, 1)
assert.Equal(t, "rootbar", c.Messages[0].Text)
})
t.Run("method=Reset", func(t *testing.T) {
c := Container{
Nodes: node.Nodes{
&node.Node{Messages: text.Messages{{Text: "foo"}}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "1", Type: node.InputAttributeTypeText, FieldValue: "foo"}},
&node.Node{Messages: text.Messages{{Text: "bar"}}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "2", Type: node.InputAttributeTypeText, FieldValue: "bar"}},
},
Messages: text.Messages{{Text: ""}},
}
c.Reset()
assert.Empty(t, c.Messages)
assert.Empty(t, c.Nodes.Find("1").Messages)
assert.Empty(t, c.Nodes.Find("1").Attributes.(*node.InputAttributes).FieldValue)
assert.Empty(t, c.Nodes.Find("2").Messages)
assert.Empty(t, c.Nodes.Find("2").Attributes.(*node.InputAttributes).FieldValue)
})
t.Run("method=remove", func(t *testing.T) {
c := Container{
Nodes: node.Nodes{
&node.Node{Messages: text.Messages{{Text: "foo"}}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "1", Type: node.InputAttributeTypeText, FieldValue: "foo"}},
&node.Node{Messages: text.Messages{{Text: "bar"}}, Group: node.DefaultGroup, Type: node.Input, Attributes: &node.InputAttributes{Name: "2", Type: node.InputAttributeTypeText, FieldValue: "bar"}},
},
Messages: text.Messages{{Text: ""}},
}
require.Len(t, c.Nodes, 2)
c.GetNodes().Remove("1")
require.Len(t, c.Nodes, 1)
require.EqualValues(t, "bar", c.Nodes[0].Attributes.GetValue())
})
}