Compass/backend/api/tests/unit/update-notif-setting.unit.test.ts

jest.mock('shared/supabase/init');
jest.mock('shared/supabase/users');
jest.mock('shared/websockets/helpers');

import { AuthedUser } from "api/helpers/endpoint";
import { updateNotifSettings } from "api/update-notif-setting";
import * as supabaseInit from "shared/supabase/init";
import * as supabaseUsers from "shared/supabase/users";
import * as websocketHelpers from "shared/websockets/helpers";

describe('updateNotifSettings', () => {
    let mockPg = {} as any;
    beforeEach(() => {
        jest.resetAllMocks();
        mockPg = {
            none: jest.fn()
        };

        (supabaseInit.createSupabaseDirectClient as jest.Mock)
            .mockReturnValue(mockPg);
    });
    afterEach(() => {
        jest.restoreAllMocks();
    });

    describe('when given valid input', () => {
        it('should update notification settings', async () => {
            const mockProps = {
                type: "new_match" as const,
                medium: "email" as const,
                enabled: false
            };
            const mockAuth = { uid: '321' } as AuthedUser;
            const mockReq = {} as any;

            (mockPg.none as jest.Mock).mockResolvedValue(null);
            (websocketHelpers.broadcastUpdatedPrivateUser as jest.Mock).mockReturnValue(null);

            await updateNotifSettings(mockProps, mockAuth, mockReq);

            expect(mockPg.none).toBeCalledTimes(1);
            expect(mockPg.none).toBeCalledWith(
                expect.stringContaining('update private_users'),
                [mockProps.type, mockProps.medium, mockAuth.uid]
            );
            expect(websocketHelpers.broadcastUpdatedPrivateUser).toBeCalledTimes(1);
            expect(websocketHelpers.broadcastUpdatedPrivateUser).toBeCalledWith(mockAuth.uid);
        });    

        it('should turn off notifications', async () => {
            const mockProps = {
                type: "opt_out_all" as const,
                medium: "mobile" as const,
                enabled: true
            };
            const mockAuth = { uid: '321' } as AuthedUser;
            const mockReq = {} as any;

            (supabaseUsers.updatePrivateUser as jest.Mock).mockResolvedValue(null);

            await updateNotifSettings(mockProps, mockAuth, mockReq);

            expect(supabaseUsers.updatePrivateUser).toBeCalledTimes(1);
            expect(supabaseUsers.updatePrivateUser).toBeCalledWith(
                expect.any(Object),
                mockAuth.uid,
                {interestedInPushNotifications: !mockProps.enabled}
            );
        });    
    });
});