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
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package totp_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/ory/x/contextx"
|
|
|
|
"github.com/ory/kratos/driver/config"
|
|
"github.com/ory/kratos/pkg"
|
|
"github.com/ory/kratos/selfservice/strategy/totp"
|
|
)
|
|
|
|
func TestGenerator(t *testing.T) {
|
|
_, reg := pkg.NewFastRegistryWithMocks(t)
|
|
|
|
t.Run("no issuer set", func(t *testing.T) {
|
|
key, err := totp.NewKey(t.Context(), "foo", reg)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, reg.Config().SelfPublicURL(t.Context()).Hostname(), key.Issuer(), "if issuer is not set explicitly it should be the public URL")
|
|
})
|
|
|
|
t.Run("custom issuer set", func(t *testing.T) {
|
|
ctx := contextx.WithConfigValue(t.Context(), config.ViperKeyTOTPIssuer, "foobar.com")
|
|
|
|
key, err := totp.NewKey(ctx, "foo", reg)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "foobar.com", key.Issuer(), "if issuer is set explicitly it should be the correct value")
|
|
assert.Equal(t, "foo", key.AccountName())
|
|
})
|
|
|
|
t.Run("generate HTML image", func(t *testing.T) {
|
|
key, err := totp.NewKey(t.Context(), "foo", reg)
|
|
require.NoError(t, err)
|
|
|
|
img, err := totp.KeyToHTMLImage(key)
|
|
require.NoError(t, err)
|
|
assert.Truef(t, strings.HasPrefix(img, "data:image/png;base64,"), "image is a base64 encoded png: %s", img)
|
|
})
|
|
}
|