import { createComponent } from './test-utils';

import {
  mockCreateDistribution,
  mockUpdateDistribution,
  mockCreateDistributionPromise,
  mockGetDistributionConfigPromise,
  mockUpdateDistributionPromise,
} from 'aws-sdk';
import CloudFrontComponent from '../serverless';

jest.mock('aws-sdk', () => require('../__mocks__/aws-sdk.mock'));

describe('General options propagation', () => {
  let component: CloudFrontComponent;

  // sample origins
  const origins = ['https://exampleorigin.com'];

  beforeEach(async () => {
    mockCreateDistributionPromise.mockResolvedValueOnce({
      Distribution: {
        Id: 'distribution123',
      },
    });

    mockGetDistributionConfigPromise.mockResolvedValueOnce({
      ETag: 'etag',
      DistributionConfig: {
        Origins: {
          Items: [],
        },
      },
    });
    mockUpdateDistributionPromise.mockResolvedValueOnce({
      Distribution: {
        Id: 'xyz',
      },
    });

    component = await createComponent();
  });

  it('create distribution with comment and update it', async () => {
    await component.default({
      comment: 'test comment',
      origins,
    });

    expect(mockCreateDistribution).toBeCalledWith(
      expect.objectContaining({
        DistributionConfig: expect.objectContaining({
          Comment: 'test comment',
        }),
      })
    );

    await component.default({
      comment: 'updated comment',
      origins,
    });

    expect(mockUpdateDistribution).toBeCalledWith(
      expect.objectContaining({
        DistributionConfig: expect.objectContaining({
          Comment: 'updated comment',
        }),
      })
    );
  });

  it('create disabled distribution and update it', async () => {
    await component.default({
      enabled: false,
      origins,
    });

    expect(mockCreateDistribution).toBeCalledWith(
      expect.objectContaining({
        DistributionConfig: expect.objectContaining({
          Enabled: false,
        }),
      })
    );

    await component.default({
      enabled: true,
      origins,
    });

    expect(mockUpdateDistribution).toBeCalledWith(
      expect.objectContaining({
        DistributionConfig: expect.objectContaining({
          Enabled: true,
        }),
      })
    );
  });
});