import { CreateApplicationLoadBalancerComponent } from './loadBalancer/configure/application/CreateApplicationLoadBalancer';
import { CreateClassicLoadBalancerComponent } from './loadBalancer/configure/classic/CreateClassicLoadBalancer';
import { CreateNetworkLoadBalancerComponent } from './loadBalancer/configure/network/CreateNetworkLoadBalancer';
import { CreateLambdaFunctionComponent } from './function/CreateLambdaFunction';
import { AmazonSecurityGroupModalComponent } from './securityGroup/configure/AmazonSecurityGroupModal';
import { AmazonCloneServerGroupModalComponent } from './serverGroup/configure/wizard/AmazonCloneServerGroupModal';

describe('Amazon routed modal consumers', () => {
  function stateService(includedState: string) {
    return {
      go: vi.fn(),
      includes: vi.fn().mockImplementation((state: string) => state === includedState),
    };
  }

  [
    ['application', CreateApplicationLoadBalancerComponent],
    ['classic', CreateClassicLoadBalancerComponent],
    ['network', CreateNetworkLoadBalancerComponent],
  ].forEach(([type, Component]: [string, any]) => {
    it(`navigates from the ${type} load balancer modal through its injected state service`, () => {
      const state = stateService('**.loadBalancerDetails');
      const modal = Object.create(Component.prototype) as any;
      modal.props = { dismissModal: vi.fn(), stateService: state };
      modal.state = {};
      modal.setState = vi.fn();

      modal.onApplicationRefresh({
        credentials: 'test-account',
        name: 'fnord-main',
        region: 'eu-west-1',
        vpcId: 'vpc-1',
      });

      expect(state.go).toHaveBeenCalledWith('^.loadBalancerDetails', {
        accountId: 'test-account',
        name: 'fnord-main',
        provider: 'aws',
        region: 'eu-west-1',
        vpcId: 'vpc-1',
      });
    });
  });

  [
    ['application load balancer', CreateApplicationLoadBalancerComponent, 'loadBalancers'],
    ['classic load balancer', CreateClassicLoadBalancerComponent, 'loadBalancers'],
    ['network load balancer', CreateNetworkLoadBalancerComponent, 'loadBalancers'],
    ['Lambda function', CreateLambdaFunctionComponent, 'functions'],
  ].forEach(([type, Component, dataSourceName]: [string, any, string]) => {
    it(`owns the ${type} refresh subscription across replacement and unmount`, () => {
      const firstUnsubscribe = vi.fn();
      const secondUnsubscribe = vi.fn();
      const callbacks: Array<() => void> = [];
      const onNextRefresh = vi.fn().mockImplementation((callback: () => void) => {
        callbacks.push(callback);
        return callbacks.length === 1 ? firstUnsubscribe : secondUnsubscribe;
      });
      const refresh = vi.fn();
      const state = stateService('');
      const modal = Object.create(Component.prototype) as any;
      modal.props = {
        app: { [dataSourceName]: { onNextRefresh, refresh } },
        dismissModal: vi.fn(),
        stateService: state,
      };
      modal.state = {};
      modal.setState = vi.fn();

      modal.onTaskComplete({});

      expect(onNextRefresh.mock.invocationCallOrder[0]).toBeLessThan(refresh.mock.invocationCallOrder[0]);

      modal.onTaskComplete({});

      expect(firstUnsubscribe).toHaveBeenCalledTimes(1);

      modal.componentWillUnmount();
      callbacks[1]();

      expect(secondUnsubscribe).toHaveBeenCalledTimes(1);
      expect(modal.refreshUnsubscribe).toBeUndefined();
      expect(state.go).not.toHaveBeenCalled();
    });
  });

  it('navigates from the function modal through its injected state service', () => {
    const state = stateService('');
    const modal = Object.create(CreateLambdaFunctionComponent.prototype) as any;
    modal.props = { dismissModal: vi.fn(), stateService: state };
    modal.state = {};
    modal.setState = vi.fn();

    modal.onApplicationRefresh({ credentials: 'test-account', name: 'fnord', region: 'eu-west-1', vpcId: 'vpc-1' });

    expect(state.go).toHaveBeenCalledWith('.functionDetails', {
      accountId: 'test-account',
      name: 'fnord',
      provider: 'aws',
      region: 'eu-west-1',
      vpcId: 'vpc-1',
    });
  });

  it('navigates from the security group modal through its injected state service', () => {
    const state = stateService('**.firewallDetails');
    const refresh = vi.fn();
    const closeModal = vi.fn();
    const modal = new AmazonSecurityGroupModalComponent({
      app: { securityGroups: { refresh } },
      closeModal,
      dismissModal: vi.fn(),
      mode: 'clone',
      stateService: state,
    } as any) as any;
    modal.state = {
      securityGroup: {
        credentials: 'test-account',
        name: 'fnord-firewall',
        region: 'eu-west-1',
        vpcId: 'vpc-1',
      },
    };

    modal.onTaskComplete();

    expect(refresh).toHaveBeenCalled();
    expect(closeModal).toHaveBeenCalled();
    expect(state.go).toHaveBeenCalledWith('^.firewallDetails', {
      accountId: 'test-account',
      name: 'fnord-firewall',
      provider: 'aws',
      region: 'eu-west-1',
      vpcId: 'vpc-1',
    });
  });

  it('navigates from the clone server group modal through its injected state service', () => {
    const state = stateService('**.clusters');
    const modal = new AmazonCloneServerGroupModalComponent({
      application: { name: 'fnord' },
      command: {
        credentials: 'test-account',
        region: 'eu-west-1',
        viewState: { requiresTemplateSelection: true },
      },
      dismissModal: vi.fn(),
      stateService: state,
    } as any) as any;
    modal.state = {
      taskMonitor: {
        task: {
          execution: {
            stages: [
              {
                context: { 'deploy.server.groups': { 'eu-west-1': 'fnord-main-v042' } },
                type: 'cloneServerGroup',
              },
            ],
          },
        },
      },
    };

    modal.onApplicationRefresh();

    expect(state.go).toHaveBeenCalledWith('.serverGroup', {
      accountId: 'test-account',
      provider: 'aws',
      region: 'eu-west-1',
      serverGroup: 'fnord-main-v042',
    });
  });

  it('owns the clone server group refresh subscription across replacement and unmount', () => {
    const firstUnsubscribe = vi.fn();
    const secondUnsubscribe = vi.fn();
    const callbacks: Array<() => void> = [];
    const onNextRefresh = vi.fn().mockImplementation((callback: () => void) => {
      callbacks.push(callback);
      return callbacks.length === 1 ? firstUnsubscribe : secondUnsubscribe;
    });
    const refresh = vi.fn();
    const state = stateService('**.clusters');
    const modal = new AmazonCloneServerGroupModalComponent({
      application: { name: 'fnord', serverGroups: { onNextRefresh, refresh } },
      command: { credentials: 'test-account', region: 'eu-west-1', viewState: { requiresTemplateSelection: true } },
      dismissModal: vi.fn(),
      stateService: state,
    } as any) as any;

    modal.onTaskComplete();

    expect(onNextRefresh.mock.invocationCallOrder[0]).toBeLessThan(refresh.mock.invocationCallOrder[0]);

    modal.onTaskComplete();

    expect(firstUnsubscribe).toHaveBeenCalledTimes(1);

    modal.componentWillUnmount();
    callbacks[1]();

    expect(secondUnsubscribe).toHaveBeenCalledTimes(1);
    expect(modal.refreshUnsubscribe).toBeUndefined();
    expect(state.go).not.toHaveBeenCalled();
  });
});