mattermost-mobile/test/intl-test-helper.tsx

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

import {DatabaseProvider} from '@nozbe/watermelondb/react';
import {render, type RenderOptions} from '@testing-library/react-native';
import React, {type ReactElement} from 'react';
import {IntlProvider} from 'react-intl';
import {SafeAreaProvider} from 'react-native-safe-area-context';

import ServerUrlProvider from '@context/server';
import {ThemeContext, getDefaultThemeByAppearance} from '@context/theme';
import {getTranslations} from '@i18n';

import type Database from '@nozbe/watermelondb/Database';

export function renderWithIntl(ui: ReactElement, {locale = 'en', ...renderOptions} = {}) {
    function Wrapper({children}: {children: ReactElement}) {
        return (
            <IntlProvider
                locale={locale}
                messages={getTranslations(locale)}
            >
                <SafeAreaProvider>
                    {children}
                </SafeAreaProvider>
            </IntlProvider>
        );
    }

    return render(ui, {wrapper: Wrapper, ...renderOptions});
}

export function renderWithIntlAndTheme(ui: ReactElement, {locale = 'en', ...renderOptions} = {}) {
    function Wrapper({children}: {children: ReactElement}) {
        return (
            <IntlProvider
                locale={locale}
                messages={getTranslations(locale)}
            >
                <ThemeContext.Provider value={getDefaultThemeByAppearance()}>
                    <SafeAreaProvider>
                        {children}
                    </SafeAreaProvider>
                </ThemeContext.Provider>
            </IntlProvider>
        );
    }

    return render(ui, {wrapper: Wrapper, ...renderOptions});
}

export function renderWithEverything(ui: ReactElement, {locale = 'en', database, serverUrl, ...renderOptions}: {locale?: string; database?: Database; serverUrl?: string; renderOptions?: RenderOptions} = {}) {
    function Wrapper({children}: {children: ReactElement}) {
        if (!database) {
            return null;
        }

        const wrapper = (
            <DatabaseProvider database={database}>
                <IntlProvider
                    locale={locale}
                    messages={getTranslations(locale)}
                >
                    <ThemeContext.Provider value={getDefaultThemeByAppearance()}>
                        <SafeAreaProvider>
                            {children}
                        </SafeAreaProvider>
                    </ThemeContext.Provider>
                </IntlProvider>
            </DatabaseProvider>
        );

        if (serverUrl) {
            return (
                <ServerUrlProvider server={{displayName: serverUrl, url: serverUrl}}>
                    {wrapper}
                </ServerUrlProvider>
            );
        }

        return wrapper;
    }

    return render(ui, {wrapper: Wrapper, ...renderOptions});
}

// eslint-disable-next-line no-duplicate-imports
export * from '@testing-library/react-native';