def main()

in ftl/integration_tests/ftl_python_integration_tests_yaml.py [0:0]


def main():

    cloudbuild_yaml = util.INITIAL_CLOUDBUILD_YAML
    cloudbuild_yaml['steps'].append(
        # Build the FTL image from source and load it into the daemon.
        {
            'name': 'gcr.io/cloud-builders/bazel@sha256:7360c36bded15db68a35cfb1740a994f0a09ad5ce378a97f96d698bc223e442a',
            'args': ['run', '//ftl:python_builder_image', '--', '--norun'],
            'id': 'build-builder',
        }, )

    # Generate a set of steps for each test and add them.
    test_map = {}
    for test in TEST_DIRS:
        test_map[test] = [
            '--base', _PYTHON_BASE, '--name',
            'gcr.io/ftl-node-test/%s-image:latest' % test, '--directory',
            os.path.join(_TEST_DIR, test), '--no-cache'
        ]
    test_map['metadata_test'].extend(['--entrypoint', '/bin/echo'])
    test_map['metadata_test'].extend(['--exposed-ports', '8090,8091'])
    test_map['python3_test'].extend(['--python-cmd', 'python3.6'])
    test_map['python3_test'].extend(['--pip-cmd', 'python3.6 -m pip'])
    test_map['virtualenv_dir_test'].extend(['--virtualenv-dir',
                                            '/alternate-env'])
    test_map['additional_directory'].extend([
        '--additional-directory',
        '/workspace/ftl/python/testdata/additional_directory'
    ])
    test_map['setup_cfg_test'].extend(['--python-cmd', 'python3.6'])
    test_map['setup_cfg_test'].extend(['--pip-cmd', 'python3.6 -m pip'])
    test_map['venv_test'].extend(['--venv-cmd', 'python3.6 -m venv /env'])

    for test, args in test_map.iteritems():
        cloudbuild_yaml['steps'] += util.run_test_steps(
            'python_builder_image',
            'gcr.io/ftl-node-test/%s-image:latest' % test,
            os.path.join(_TEST_DIR, test), args)

    print yaml.dump(cloudbuild_yaml)