cookie.test.ts

/*
 * Copyright (c) 2025 Ping Identity Corporation. All rights reserved.
 *
 * This software may be modified and distributed under the terms
 * of the MIT license. See the LICENSE file for details.
 */
import { it } from '@effect/vitest';
import { expect } from 'vitest';
import {
  getElementFromCookie,
  incrementCookieHeader,
  parseCookieHeaderForIndex,
} from '../cookie.js';
import { HeaderTypes } from '../../types/index.js';
import { Effect, Exit } from 'effect';
import { responseMap } from '../../responses/index.js';
import { returnSuccessResponseRedirect } from '../../responses/return-success-redirect.js';
import { NoSuchElementException } from 'effect/Cause';

it.effect('should parse a cookie header for an index value', () =>
  Effect.gen(function* () {
    const header: HeaderTypes = {
      cookie: 'stepIndex=1',
    };

    const result = yield* parseCookieHeaderForIndex(header);
    expect(result).toEqual('1');
  }),
);

it.effect('should parse a cookie header for an index value', () =>
  Effect.gen(function* () {
    const header: HeaderTypes = {
      cookie: '',
    };

    const result = yield* parseCookieHeaderForIndex(header).pipe(Effect.exit);
    expect(result).toEqual(Exit.fail(new NoSuchElementException()));
  }),
);

it('should increment the cookie header', () => {
  const headers: HeaderTypes = {
    cookie: 'stepIndex=1',
  };

  const result = incrementCookieHeader(headers);
  expect(result).toEqual('2');
});

it('should return 1 if no cookie header passed to incrementCookieHeader', () => {
  const headers: HeaderTypes = {};

  const result = incrementCookieHeader(headers);
  expect(result).toEqual('1');
});

it('should get an element from the response map based off the cookie header', () =>
  Effect.gen(function* () {
    const headers: HeaderTypes = {
      cookie: 'stepIndex=1',
    };

    const expected = responseMap['UsernamePassword'][1];
    const result = yield* getElementFromCookie(responseMap['UsernamePassword'], headers);
    expect(expected).toEqual(result);
  }));
it.effect('should return responesRedirect when we have exceeded the index for a given flow', () =>
  Effect.gen(function* () {
    const headers: HeaderTypes = {
      cookie: 'stepIndex=4',
    };

    const arr = responseMap['UsernamePassword'];
    const expected = returnSuccessResponseRedirect;
    const result = yield* getElementFromCookie(arr, headers);

    expect(expected).toEqual(result);
  }),
);