RocketChat/Rocket.Chat

import { MockedServerContext, MockedUserContext } from '@rocket.chat/mock-providers';
import { render } from '@testing-library/react';
import { Meteor } from 'meteor/meteor';
import React from 'react';

import RouterContextMock from '../../../tests/mocks/client/RouterContextMock';
import SAMLLoginRoute from './SAMLLoginRoute';

const navigateStub = jest.fn();

beforeEach(() => {
    jest.clearAllMocks();
    navigateStub.mockClear();
    (Meteor.loginWithSamlToken as jest.Mock<any>).mockClear();
});

it('should redirect to /home', async () => {
    render(
        <MockedServerContext>
            <MockedUserContext>
                <RouterContextMock searchParameters={{ redirectUrl: 'http://rocket.chat' }} navigate={navigateStub}>
                    <SAMLLoginRoute />
                </RouterContextMock>
            </MockedUserContext>
        </MockedServerContext>,
        { legacyRoot: true },
    );

    expect(navigateStub).toHaveBeenCalledTimes(1);
    expect(navigateStub).toHaveBeenLastCalledWith(expect.objectContaining({ pathname: '/home' }), expect.anything());
});

it('should redirect to /home when userId is not null', async () => {
    render(
        <MockedServerContext>
            <MockedUserContext>
                <RouterContextMock searchParameters={{ redirectUrl: 'http://rocket.chat' }} navigate={navigateStub}>
                    <SAMLLoginRoute />
                </RouterContextMock>
            </MockedUserContext>
        </MockedServerContext>,
        { legacyRoot: true },
    );

    expect(navigateStub).toHaveBeenCalledTimes(1);
    expect(navigateStub).toHaveBeenLastCalledWith(expect.objectContaining({ pathname: '/home' }), expect.anything());
});

it('should redirect to /home when userId is null and redirectUrl is not within the workspace domain', async () => {
    render(
        <MockedServerContext>
            <RouterContextMock searchParameters={{ redirectUrl: 'http://rocket.chat' }} navigate={navigateStub}>
                <SAMLLoginRoute />
            </RouterContextMock>
        </MockedServerContext>,
        { legacyRoot: true },
    );

    expect(navigateStub).toHaveBeenCalledTimes(1);
    expect(navigateStub).toHaveBeenLastCalledWith(expect.objectContaining({ pathname: '/home' }), expect.anything());
});

it('should redirect to the provided redirectUrl when userId is null and redirectUrl is within the workspace domain', async () => {
    render(
        <MockedServerContext>
            <RouterContextMock searchParameters={{ redirectUrl: 'http://localhost:3000/invite/test' }} navigate={navigateStub}>
                <SAMLLoginRoute />
            </RouterContextMock>
        </MockedServerContext>,
        { legacyRoot: true },
    );

    expect(navigateStub).toHaveBeenCalledTimes(1);
    expect(navigateStub).toHaveBeenLastCalledWith(expect.objectContaining({ pathname: '/invite/test' }), expect.anything());
});

it('should call loginWithSamlToken when component is mounted', async () => {
    render(
        <MockedServerContext>
            <RouterContextMock searchParameters={{ redirectUrl: 'http://rocket.chat' }} navigate={navigateStub}>
                <SAMLLoginRoute />
            </RouterContextMock>
        </MockedServerContext>,
        { legacyRoot: true },
    );

    expect(Meteor.loginWithSamlToken).toHaveBeenCalledTimes(1);
    expect(Meteor.loginWithSamlToken).toHaveBeenLastCalledWith(undefined, expect.any(Function));
});

it('should call loginWithSamlToken with the token when it is present', async () => {
    render(
        <MockedUserContext>
            <RouterContextMock
                searchParameters={{ redirectUrl: 'http://rocket.chat' }}
                routeParameters={{ token: 'testToken' }}
                navigate={navigateStub}
            >
                <SAMLLoginRoute />
            </RouterContextMock>
        </MockedUserContext>,
        { legacyRoot: true },
    );

    expect(Meteor.loginWithSamlToken).toHaveBeenCalledTimes(1);
    expect(Meteor.loginWithSamlToken).toHaveBeenLastCalledWith('testToken', expect.any(Function));
});