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
83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package totp_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/ory/kratos/identity"
|
|
"github.com/ory/kratos/pkg"
|
|
"github.com/ory/kratos/selfservice/strategy/totp"
|
|
)
|
|
|
|
func TestCountActiveCredentials(t *testing.T) {
|
|
_, reg := pkg.NewFastRegistryWithMocks(t)
|
|
strategy := totp.NewStrategy(reg)
|
|
|
|
key, err := totp.NewKey(t.Context(), "foo", reg)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("first factor", func(t *testing.T) {
|
|
actual, err := strategy.CountActiveFirstFactorCredentials(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 0, actual)
|
|
})
|
|
|
|
t.Run("multi factor", func(t *testing.T) {
|
|
for k, tc := range []struct {
|
|
in map[identity.CredentialsType]identity.Credentials
|
|
expected int
|
|
}{
|
|
{
|
|
in: map[identity.CredentialsType]identity.Credentials{strategy.ID(): {
|
|
Type: strategy.ID(),
|
|
Config: []byte{},
|
|
}},
|
|
expected: 0,
|
|
},
|
|
{
|
|
in: map[identity.CredentialsType]identity.Credentials{strategy.ID(): {
|
|
Type: strategy.ID(),
|
|
Config: []byte(`{"totp_url": ""}`),
|
|
}},
|
|
expected: 0,
|
|
},
|
|
{
|
|
in: map[identity.CredentialsType]identity.Credentials{strategy.ID(): {
|
|
Type: strategy.ID(),
|
|
Identifiers: []string{"foo"},
|
|
Config: []byte(`{"totp_url": "` + key.URL() + `"}`),
|
|
}},
|
|
expected: 1,
|
|
},
|
|
{
|
|
in: map[identity.CredentialsType]identity.Credentials{strategy.ID(): {
|
|
Type: strategy.ID(),
|
|
Config: []byte(`{}`),
|
|
}},
|
|
expected: 0,
|
|
},
|
|
{
|
|
in: nil,
|
|
expected: 0,
|
|
},
|
|
} {
|
|
t.Run(fmt.Sprintf("case=%d", k), func(t *testing.T) {
|
|
cc := map[identity.CredentialsType]identity.Credentials{}
|
|
for _, c := range tc.in {
|
|
cc[c.Type] = c
|
|
}
|
|
|
|
actual, err := strategy.CountActiveMultiFactorCredentials(t.Context(), cc)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tc.expected, actual)
|
|
})
|
|
}
|
|
})
|
|
}
|