import { expect } from 'chai';

import { getModifiedHttpHeaders } from '../../../../../server/lib/shared/getModifiedHttpHeaders';

const inputVariants = [
	{
		label: 'HTTP path (Headers instance)',
		makeInput: (headers: Record<string, string>) => new Headers(headers),
	},
	{
		label: 'DDP path (plain object)',
		makeInput: (headers: Record<string, string>) => headers,
	},
];

describe('getModifiedHttpHeaders', () => {
	for (const { label, makeInput } of inputVariants) {
		describe(label, () => {
			it('should redact x-auth-token if present', () => {
				const inputHeaders = {
					'x-auth-token': '12345',
					'some-other-header': 'value',
				};
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result['x-auth-token']).to.equal('[redacted]');
				expect(result['some-other-header']).to.equal('value');
			});

			it('should not modify headers if x-auth-token is not present', () => {
				const inputHeaders = {
					'some-other-header': 'value',
				};
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result).to.deep.equal(inputHeaders);
			});

			it('should redact rc_token in cookies if present', () => {
				const inputHeaders = {
					cookie: 'session_id=abc123; rc_token=98765; other_cookie=value',
				};
				const expectedCookies = 'session_id=abc123; rc_token=[redacted]; other_cookie=value';
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result.cookie).to.equal(expectedCookies);
			});

			it('should not modify cookies if rc_token is not present', () => {
				const inputHeaders = {
					cookie: 'session_id=abc123; other_cookie=value',
				};
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result.cookie).to.equal(inputHeaders.cookie);
			});

			it('should return headers unchanged if neither x-auth-token nor cookie are present', () => {
				const inputHeaders = {
					'some-other-header': 'value',
				};
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result).to.deep.equal(inputHeaders);
			});

			it('should handle cases with both x-auth-token and rc_token in cookie', () => {
				const inputHeaders = {
					'x-auth-token': '12345',
					'cookie': 'session_id=abc123; rc_token=98765; other_cookie=value',
				};
				const expectedCookies = 'session_id=abc123; rc_token=[redacted]; other_cookie=value';
				const result = getModifiedHttpHeaders(makeInput(inputHeaders));
				expect(result['x-auth-token']).to.equal('[redacted]');
				expect(result.cookie).to.equal(expectedCookies);
			});
		});
	}
});