import {ApiV3} from '../lib/ApiV3';
import {MockInstance, vi} from 'vitest';
import {CustomerPayload} from '../Types';
import {customer} from './customer';
import {InternalConfig} from '../config/configure';

describe('customer', () => {
  const mockConfiguration: InternalConfig = {
    apiBasePath: 'https://api.zaius.com/v3/',
    apiKey: 'api-key',
  };
  const apiV3: ApiV3.API = new ApiV3.API(mockConfiguration);
  let postMock!: MockInstance;
  beforeEach(() => {
    postMock = vi.spyOn(apiV3, 'post').mockReturnValue(Promise.resolve({} as any));
  });

  it('sends a transformed post with one customer to /profiles', async () => {
    const payload: CustomerPayload = {identifiers: {email: 'test@optimizely.com'}, attributes: {name: 'Jim Bob'}};
    const transformedPayload = {attributes: {name: 'Jim Bob', email: 'test@optimizely.com'}};
    await customer(apiV3, payload);
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('sends a transformed post with several customers to /profiles', async () => {
    const payload: CustomerPayload[] = [
      {identifiers: {email: 'test1@optimizely.com'}, attributes: {name: 'Jim Bob'}},
      {identifiers: {email: 'test2@optimizely.com'}, attributes: {name: 'Bob Joe'}},
      {identifiers: {email: 'test3@optimizely.com'}, attributes: {name: 'Joe Jim'}},
    ];
    const transformedPayload = [
      {attributes: {name: 'Jim Bob', email: 'test1@optimizely.com'}},
      {attributes: {name: 'Bob Joe', email: 'test2@optimizely.com'}},
      {attributes: {name: 'Joe Jim', email: 'test3@optimizely.com'}},
    ];
    await customer(apiV3, payload);
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('sanitizes the payload', async () => {
    const payload: CustomerPayload = {
      identifiers: {
        email: 'test@optimizely.com',
      },
      attributes: {
        name: 'Jim Bob',
        blank: ' ',
        nullValue: null,
      },
    };
    const transformedPayload = {attributes: {name: 'Jim Bob', email: 'test@optimizely.com'}};
    await customer(apiV3, payload);
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('sanitizes the payload with several customers', async () => {
    const payload: CustomerPayload[] = [
      {identifiers: {email: 'test1@optimizely.com'}, attributes: {name: 'Jim Bob', blank: ' ', nullValue: null}},
      {identifiers: {email: 'test2@optimizely.com'}, attributes: {name: 'Bob Joe', blank: ' ', nullValue: null}},
      {identifiers: {email: 'test3@optimizely.com'}, attributes: {name: 'Joe Jim', blank: ' ', nullValue: null}},
    ];
    const transformedPayload = [
      {attributes: {name: 'Jim Bob', email: 'test1@optimizely.com'}},
      {attributes: {name: 'Bob Joe', email: 'test2@optimizely.com'}},
      {attributes: {name: 'Joe Jim', email: 'test3@optimizely.com'}},
    ];
    await customer(apiV3, payload);
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('applies PayloadOptions', async () => {
    const payload: CustomerPayload = {
      identifiers: {
        email: 'test@optimizely.com',
      },
      attributes: {
        name: 'Jim Bob',
        blank: ' ',
        nullValue: null,
      },
    };
    const transformedPayload = {
      attributes: {name: 'Jim Bob', email: 'test@optimizely.com', blank: null, nullValue: null},
    };
    await customer(apiV3, payload, {excludeNulls: false});
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('applies PayloadOptions with several customers', async () => {
    const payload: CustomerPayload[] = [
      {identifiers: {email: 'test1@optimizely.com'}, attributes: {name: 'Jim Bob', blank: ' ', nullValue: null}},
      {identifiers: {email: 'test2@optimizely.com'}, attributes: {name: 'Bob Joe', blank: ' ', nullValue: null}},
      {identifiers: {email: 'test3@optimizely.com'}, attributes: {name: 'Joe Jim', blank: ' ', nullValue: null}},
    ];
    const transformedPayload = [
      {attributes: {name: 'Jim Bob', email: 'test1@optimizely.com', blank: ' '}},
      {attributes: {name: 'Bob Joe', email: 'test2@optimizely.com', blank: ' '}},
      {attributes: {name: 'Joe Jim', email: 'test3@optimizely.com', blank: ' '}},
    ];
    await customer(apiV3, payload, {trimToNull: false});
    expect(postMock).toHaveBeenCalledWith('/profiles', transformedPayload);
  });

  it('throws an error if too many customers are sent in one call', async () => {
    const payload: CustomerPayload[] = [];
    for (let i = 0; i < ApiV3.BATCH_LIMIT + 1; i++) {
      payload.push({identifiers: {email: 'test@optimizely.com'}, attributes: {name: 'Jim Bob'}});
    }

    expect.assertions(2);
    try {
      await customer(apiV3, payload);
    } catch (error: any) {
      expect(error.message).toMatch(/maximum batch size/);
      expect(error.code).toEqual(ApiV3.ErrorCode.BatchLimitExceeded);
    }
  });
});