def test_prepare_activity()

in aws/hhvm1/lambdas/test.py [0:0]
74 lines of code
1 McCabe index (conditional complexity)

  def test_prepare_activity(self):
    self.assertEqual(
      prepare_activity.lambda_handler({
        'activity': 'MakeBinaryPackage',
        'version': '4.26.12345',
        'platform': 'ubuntu-18.04-bionic',
      }),
      {
        'skip': False,
        'taskInput': {
          'name': 'MakeBinaryPackage-4.26.12345-ubuntu-18.04-bionic',
          'env': (
            'VERSION="4.26.12345"\n'
            'DISTRO="ubuntu-18.04-bionic"\n'
            'IS_NIGHTLY="false"\n'
            'S3_BUCKET="hhvm-scratch"\n'
            'S3_PATH="hhvm-4.26.12345.tar.gz"\n'
            'S3_SOURCE="s3://hhvm-scratch/hhvm-4.26.12345.tar.gz"\n'
            'PACKAGING_BRANCH="HHVM-4.26"'
          ),
          'fail_args': '',
        },
      }
    )

    self.assertEqual(
      prepare_activity.lambda_handler({
        'buildInput': {'debug': 'test_build'},
        'activity': 'MakeBinaryPackage',
        'version': '4.26.12345',
        'platform': 'ubuntu-18.04-bionic',
      }),
      {
        'skip': False,
        'taskInput': {
          'name': 'MakeBinaryPackage-4.26.12345-ubuntu-18.04-bionic',
          'env': (
            'VERSION="4.26.12345"\n'
            'DISTRO="ubuntu-18.04-bionic"\n'
            'IS_NIGHTLY="false"\n'
            'S3_BUCKET="hhvm-scratch"\n'
            'S3_PATH="hhvm-4.26.12345.tar.gz"\n'
            'S3_SOURCE="s3://hhvm-scratch/hhvm-4.26.12345.tar.gz"\n'
            'PACKAGING_BRANCH="HHVM-4.26"\n'
            'SKIP_PUBLISH="1"'
          ),
          'fail_args': '--error TestBuildNoRetry',
        },
      }
    )

    future = (date.today() + timedelta(days=2)).strftime('%Y.%m.%d')
    self.assertEqual(
      prepare_activity.lambda_handler({
        'buildInput': {'debug': 'skip_ec2'},
        'activity': 'PublishDockerImages',
        'version': future,
      }),
      {
        # skip_ec2 causes this to be True, but the taskInput is still included
        # since it's useful for debugging
        'skip': True,
        'taskInput': {
          'name': f'PublishDockerImages-{future}',
          'env': (
            f'VERSION="{future}"\n'
            'IS_NIGHTLY="true"\n'
            'S3_BUCKET="hhvm-downloads"\n'
            f'S3_PATH="source/nightlies/hhvm-nightly-{future}.tar.gz"\n'
            'S3_SOURCE="s3://hhvm-downloads/source/nightlies/'
              f'hhvm-nightly-{future}.tar.gz"\n'
            'PACKAGING_BRANCH="master"\n'
            'DOCKER_ONLY="1"'
          ),
          'fail_args': '',
        },
      }
    )