Lunalytics/test/server/middleware/user/transferOwnership.test.js

import { beforeEach, describe, expect, it, vi } from 'vitest';
import { createRequest, createResponse } from 'node-mocks-http';
import {
  emailIsOwner,
  emailIsOwner,
  transferOwnership,
  getUserByEmail,
} from '../../../../server/database/queries/user';
import transferOwnershipMiddleware from '../../../../server/middleware/user/transferOwnership';

vi.mock('../../../../server/database/queries/user');

describe('transferOwnershipMiddleware - Middleware', () => {
  const session_token = 'test_token';

  const user = {
    email: 'KSJaay@lunalytics.xyz',
    permission: 1,
  };

  let fakeRequest;
  let fakeResponse;

  beforeEach(() => {
    transferOwnership = vi.fn();
    emailIsOwner = vi.fn().mockReturnValue({ isOwner: true });
    getUserByEmail = vi.fn().mockReturnValue(true);

    fakeRequest = createRequest();
    fakeResponse = createResponse();

    fakeRequest.cookies = { session_token };
    fakeRequest.body = { email: '123@lunalytics.xyz' };
    fakeResponse.locals = { user };
  });

  afterEach(() => {
    vi.restoreAllMocks();
  });

  it('should return 400 when email is not provided', async () => {
    fakeRequest.body = {};

    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(fakeResponse.statusCode).toEqual(400);
  });

  it('should return 401 when user does has ownership', async () => {
    emailIsOwner = vi.fn().mockReturnValue({ permission: 2 });

    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(fakeResponse.statusCode).toEqual(401);
  });

  it('should call emailIsOwner with locals email', async () => {
    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(emailIsOwner).toHaveBeenCalledWith('KSJaay@lunalytics.xyz');
  });

  it('should return 400 when email does not exist', async () => {
    emailIsOwner = vi.fn().mockReturnValue(false);

    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(fakeResponse.statusCode).toEqual(401);

    expect(transferOwnership).not.toHaveBeenCalled();
  });

  it('should call transferOwnership with user.email and body.email', async () => {
    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(transferOwnership).toHaveBeenCalledWith(
      'KSJaay@lunalytics.xyz',
      '123@lunalytics.xyz'
    );
  });

  it('should return 200', async () => {
    await transferOwnershipMiddleware(fakeRequest, fakeResponse);

    expect(fakeResponse.statusCode).toEqual(200);
  });
});