Files
upstream-ory-kratos/selfservice/flow/internal_context_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

53 lines
1.6 KiB
Go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package flow
import (
"strings"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/ory/x/sqlxx"
)
type fakeInternalContexter struct {
ic sqlxx.JSONRawMessage
}
func (f *fakeInternalContexter) EnsureInternalContext() {
if len(f.ic) == 0 {
f.ic = sqlxx.JSONRawMessage("{}")
}
}
func (f *fakeInternalContexter) GetInternalContext() sqlxx.JSONRawMessage { return f.ic }
func (f *fakeInternalContexter) SetInternalContext(b sqlxx.JSONRawMessage) { f.ic = b }
func TestRequestBaseURLInternalContext(t *testing.T) {
t.Run("round-trips through InternalContext", func(t *testing.T) {
f := &fakeInternalContexter{}
require.NoError(t, SetRequestBaseURL(f, "http://localhost:4000"))
assert.Equal(t, "http://localhost:4000", GetRequestBaseURL(f))
assert.Contains(t, string(f.GetInternalContext()), InternalContextKeyRequestBaseURL)
})
t.Run("empty input is a no-op and leaves no key", func(t *testing.T) {
f := &fakeInternalContexter{ic: sqlxx.JSONRawMessage("{}")}
require.NoError(t, SetRequestBaseURL(f, ""))
assert.Empty(t, GetRequestBaseURL(f))
assert.Equal(t, "{}", string(f.GetInternalContext()))
})
t.Run("oversized input is rejected (row-bloat guard)", func(t *testing.T) {
f := &fakeInternalContexter{}
require.NoError(t, SetRequestBaseURL(f, "https://"+strings.Repeat("a", 8193)))
assert.Empty(t, GetRequestBaseURL(f))
})
t.Run("missing key returns empty", func(t *testing.T) {
assert.Empty(t, GetRequestBaseURL(&fakeInternalContexter{ic: sqlxx.JSONRawMessage(`{"other":"x"}`)}))
})
}