Function TestUserIndexScanner_ScanUsers

pkg/util/users/scanner_test.go:98–212

(t *testing.T)

}

func TestUserIndexScanner_ScanUsers(t *testing.T) {
	t.Parallel()

	ctx := context.Background()
	logger := log.NewNopLogger()

	tests := map[string]struct {
		bucketSetup      func(*bucket.ClientMock)
		maxStalePeriod   time.Duration
		expectedActive   []string
		expectedDeleting []string
		expectedDeleted  []string
		expectErr        bool
	}{
		"successful scan from index": {
			bucketSetup: func(b *bucket.ClientMock) {
				index := &UserIndex{
					Version:       userIndexVersion,
					ActiveUsers:   []string{"user-1", "user-2"},
					DeletingUsers: []string{"user-3"},
					DeletedUsers:  []string{"user-4"},
					UpdatedAt:     time.Now().Unix(),
				}
				indexBytes, err := json.Marshal(index)
				require.NoError(t, err)

				var buf bytes.Buffer
				gw := gzip.NewWriter(&buf)
				_, err = gw.Write(indexBytes)
				require.NoError(t, err)
				require.NoError(t, gw.Close())

				b.MockGet(UserIndexCompressedFilename, buf.String(), nil)
			},
			maxStalePeriod:   1 * time.Hour,
			expectedActive:   []string{"user-1", "user-2"},
			expectedDeleting: []string{"user-3"},
			expectedDeleted:  []string{"user-4"},
			expectErr:        false,
		},
		"stale index falls back to base scanner": {
			bucketSetup: func(b *bucket.ClientMock) {
				// Return stale index
				index := &UserIndex{
					Version:       userIndexVersion,
					ActiveUsers:   []string{"user-1"},
					DeletingUsers: []string{},
					DeletedUsers:  []string{},
					UpdatedAt:     time.Now().Add(-2 * time.Hour).Unix(),
				}
				indexBytes, err := json.Marshal(index)
				require.NoError(t, err)

				var buf bytes.Buffer
				gw := gzip.NewWriter(&buf)
				_, err = gw.Write(indexBytes)
				require.NoError(t, err)
				require.NoError(t, gw.Close())