Ilia Denisov
76 lines
2.5 KiB
Go
76 lines
2.5 KiB
Go
package revokedevicesession
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"galaxy/authsession/internal/domain/common"
|
|
"galaxy/authsession/internal/domain/devicesession"
|
|
"galaxy/authsession/internal/service/shared"
|
|
"galaxy/authsession/internal/testkit"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestExecuteRetriesProjectionPublishUntilSuccess(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
store := &testkit.InMemorySessionStore{}
|
|
publisher := &testkit.RecordingProjectionPublisher{
|
|
Errors: []error{errors.New("publish failed"), nil},
|
|
}
|
|
require.NoError(t, store.Create(context.Background(), activeSessionFixture("device-session-1", "user-1", time.Unix(10, 0).UTC())))
|
|
|
|
service, err := New(store, publisher, testkit.FixedClock{Time: time.Unix(20, 0).UTC()})
|
|
require.NoError(t, err)
|
|
|
|
result, err := service.Execute(context.Background(), Input{
|
|
DeviceSessionID: "device-session-1",
|
|
ReasonCode: "logout_all",
|
|
ActorType: "system",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "revoked", result.Outcome)
|
|
require.Len(t, publisher.PublishedSnapshots(), 2)
|
|
}
|
|
|
|
func TestExecuteRepairsProjectionOnRepeatedAlreadyRevokedRequest(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
store := &testkit.InMemorySessionStore{}
|
|
publisher := &testkit.RecordingProjectionPublisher{Err: errors.New("publish failed")}
|
|
require.NoError(t, store.Create(context.Background(), activeSessionFixture("device-session-1", "user-1", time.Unix(10, 0).UTC())))
|
|
|
|
service, err := New(store, publisher, testkit.FixedClock{Time: time.Unix(20, 0).UTC()})
|
|
require.NoError(t, err)
|
|
|
|
_, err = service.Execute(context.Background(), Input{
|
|
DeviceSessionID: "device-session-1",
|
|
ReasonCode: "logout_all",
|
|
ActorType: "system",
|
|
})
|
|
require.Error(t, err)
|
|
assert.Equal(t, shared.ErrorCodeServiceUnavailable, shared.CodeOf(err))
|
|
require.Len(t, publisher.PublishedSnapshots(), shared.MaxProjectionPublishAttempts)
|
|
|
|
stored, getErr := store.Get(context.Background(), common.DeviceSessionID("device-session-1"))
|
|
require.NoError(t, getErr)
|
|
require.NotNil(t, stored.Revocation)
|
|
assert.Equal(t, devicesession.StatusRevoked, stored.Status)
|
|
|
|
publisher.Err = nil
|
|
|
|
result, err := service.Execute(context.Background(), Input{
|
|
DeviceSessionID: "device-session-1",
|
|
ReasonCode: "logout_all",
|
|
ActorType: "system",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "already_revoked", result.Outcome)
|
|
assert.EqualValues(t, 0, result.AffectedSessionCount)
|
|
require.Len(t, publisher.PublishedSnapshots(), shared.MaxProjectionPublishAttempts+1)
|
|
}
|