team-provider-migrate.test.ts

import { Context } from '../domain/context';

const teamProviderInfoSecrets = {
  dev: {
    awscloudformation: {
      AuthRoleName: 'amplify-teamprovider-dev-134909-authRole',
      UnauthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-dev-134909-unauthRole',
      AuthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-dev-134909-authRole',
      Region: 'us-east-1',
      DeploymentBucketName: 'amplify-teamprovider-dev-134909-deployment',
      UnauthRoleName: 'amplify-teamprovider-dev-134909-unauthRole',
      StackName: 'amplify-teamprovider-dev-134909',
      StackId: 'arn:aws:cloudformation:us-east-1:1234567891011:stack/amplify-teamprovider-dev-134909/df33f4d0-1895-11eb-a8b4-0e706f74ed45',
      AmplifyAppId: 'd1gmlw7l76gj9',
    },
    categories: {
      auth: {
        teamprovider1819bdce: {
          hostedUIProviderCreds: '[{"ProviderName":"Facebook","client_id":"asdasdasdasd","client_secret":"asdasdasd"}]',
        },
      },
    },
  },
  prod: {
    awscloudformation: {
      AuthRoleName: 'amplify-teamprovider-prod-164239-authRole',
      UnauthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-prod-164239-unauthRole',
      AuthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-prod-164239-authRole',
      Region: 'us-east-1',
      DeploymentBucketName: 'amplify-teamprovider-prod-164239-deployment',
      UnauthRoleName: 'amplify-teamprovider-prod-164239-unauthRole',
      StackName: 'amplify-teamprovider-prod-164239',
      StackId: 'arn:aws:cloudformation:us-east-1:1234567891011:stack/amplify-teamprovider-prod-164239/1b625f60-18ae-11eb-9e65-0ab042f700a7',
      AmplifyAppId: 'd1gmlw7l76gj9',
    },
    categories: {
      auth: {
        teamprovider1819bdce: {
          hostedUIProviderCreds: '[{"ProviderName":"Facebook","client_id":"abc","client_secret":"adb"}]',
        },
      },
    },
  },
};

const teamProviderInfoWithoutSecrets = {
  dev: {
    awscloudformation: {
      AuthRoleName: 'amplify-teamprovider-dev-134909-authRole',
      UnauthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-dev-134909-unauthRole',
      AuthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-dev-134909-authRole',
      Region: 'us-east-1',
      DeploymentBucketName: 'amplify-teamprovider-dev-134909-deployment',
      UnauthRoleName: 'amplify-teamprovider-dev-134909-unauthRole',
      StackName: 'amplify-teamprovider-dev-134909',
      StackId: 'arn:aws:cloudformation:us-east-1:1234567891011:stack/amplify-teamprovider-dev-134909/df33f4d0-1895-11eb-a8b4-0e706f74ed45',
      AmplifyAppId: 'd1gmlw7l76gj9',
    },
    categories: {
      auth: {
        teamprovider1819bdce: {},
      },
    },
  },
  prod: {
    awscloudformation: {
      AuthRoleName: 'amplify-teamprovider-prod-164239-authRole',
      UnauthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-prod-164239-unauthRole',
      AuthRoleArn: 'arn:aws:iam::1234567891011:role/amplify-teamprovider-prod-164239-authRole',
      Region: 'us-east-1',
      DeploymentBucketName: 'amplify-teamprovider-prod-164239-deployment',
      UnauthRoleName: 'amplify-teamprovider-prod-164239-unauthRole',
      StackName: 'amplify-teamprovider-prod-164239',
      StackId: 'arn:aws:cloudformation:us-east-1:1234567891011:stack/amplify-teamprovider-prod-164239/1b625f60-18ae-11eb-9e65-0ab042f700a7',
      AmplifyAppId: 'd1gmlw7l76gj9',
    },
    categories: {
      auth: {
        teamprovider1819bdce: {
          hostedUIProviderCreds: '[{"ProviderName":"Facebook","client_id":"abc","client_secret":"adb"}]',
        },
      },
    },
  },
};
const secrets = {
  d1gmlw7l76gj9: {
    dev: {
      auth: {
        teamprovider1819bdce: {
          hostedUIProviderCreds: '[{"ProviderName":"Facebook","client_id":"asdasdasdasd","client_secret":"asdasdasd"}]',
        },
      },
    },
  },
};

describe('test migration code', () => {
  it('case: test for teamprovide with secrets', async () => {
    const promptConfirm = jest.fn().mockReturnValue(true);
    const mockMoveSecrets = jest.fn();
    const mockteamProviderInfoExists = jest.fn().mockReturnValue(true);
    const mockGetTeamProviderInfo = jest.fn().mockReturnValue(teamProviderInfoSecrets);
    const mockGetLocalEnvInfo = jest.fn().mockReturnValue({ envName: 'dev' });
    const mockGetResourceParameterJson = jest.fn().mockReturnValue({
      authSelections: 'identityPoolAndUserPool',
    });
    jest.setMock('amplify-category-auth', {
      externalAuthEnable: jest.fn(),
    });

    jest.setMock('../utils/move-secrets-to-deployment', {
      moveSecretsFromTeamProviderToDeployment: mockMoveSecrets,
    });
    jest.setMock('amplify-cli-core', {
      stateManager: {
        teamProviderInfoExists: mockteamProviderInfoExists,
        getTeamProviderInfo: mockGetTeamProviderInfo,
        getLocalEnvInfo: mockGetLocalEnvInfo,
        getResourceParametersJson: mockGetResourceParameterJson,
      },
      pathManager: {
        findProjectRoot: jest.fn().mockReturnValue(true),
      },
      PathConstants: {
        TeamProviderFileName: 'team-provider-info.json',
        DeploymentSecretsFileName: 'deployment-secrets.json',
      },
    });
    const mockContext: Context = jest.genMockFromModule('../domain/context');
    mockContext.exeInfo = {
      inputParams: {},
    };
    mockContext.prompt = {
      confirm: promptConfirm,
    };
    mockContext.input = {
      argv: [],
      command: 'status',
    };
    const migrated = await require('../utils/team-provider-migrate').migrateTeamProviderInfo(mockContext);
    expect(migrated).toEqual(true);
    expect(mockMoveSecrets).toBeCalled();
    expect(mockteamProviderInfoExists).toBeCalled();
  });
});