def test_should_execute_build_dags_command(self):
# given
create_dags_leftovers()
clear_image_leftovers(IMAGE_DIR_PATH)
clear_package_leftovers(DIST_DIR_PATH, EGGS_DIR_PATH, BUILD_PATH)
# when
self.test_project.run_build(
'python project_setup.py build_project --build-dags')
# then
self.assertTrue(dags_built(TEST_PROJECT_PATH, 2))
self.assertFalse(dags_leftovers_exist(TEST_PROJECT_PATH))
self.assertTrue(
dags_contain(TEST_PROJECT_PATH, now(template="%Y-%m-%d")))
# when
self.test_project.run_build(
"python project_setup.py build_project --build-dags --start-time '2020-01-02 00:00:00'"
)
# then
self.assertTrue(dags_contain(TEST_PROJECT_PATH, '2020-01-02'))
# when
self.test_project.run_build(
'python project_setup.py build_project --build-dags --workflow workflow1'
)
# then
self.assertTrue(self.single_dag_for_workflow_exists('workflow1'))
def test_should_execute_build_dags_command(self):
# given
create_dags_leftovers()
clear_image_leftovers(IMAGE_DIR_PATH)
clear_package_leftovers(DIST_DIR_PATH, EGGS_DIR_PATH, BUILD_PATH)
# when
self.test_project.run_build('python project_setup.py build_project --build-dags')
# then
self.assertTrue(dags_built(TEST_PROJECT_PATH, 2))
self.assertFalse(dags_leftovers_exist(TEST_PROJECT_PATH))
self.assertTrue(dags_contain(TEST_PROJECT_PATH, repr(datetime.datetime.now().replace(hour=0, minute=0, second=0, microsecond=0) - timedelta(hours=24))))
# when
self.test_project.run_build("python project_setup.py build_project --build-dags --start-time '2020-01-02 00:00:00'")
# then
self.assertTrue(dags_contain(TEST_PROJECT_PATH, 'datetime.datetime(2020, 1, 1, 0, 0)'))
# when
self.test_project.run_build('python project_setup.py build_project --build-dags --workflow workflow1')
# then
self.assertTrue(self.single_dag_for_workflow_exists('workflow1'))
def run(self):
clear_package_leftovers(
Path(__file__).parent / 'dist',
Path(__file__).parent / 'bigflow.egg-info',
Path(__file__).parent / 'build')
self.run_command('bdist_wheel')
print(
subprocess.getoutput(
'source env/bin/activate;pip install bigflow --find-links dist/'
))
def test_should_execute_build_image_command(self):
# given
create_image_leftovers()
clear_dags_leftovers(DAGS_DIR_PATH)
clear_package_leftovers(DIST_DIR_PATH, EGGS_DIR_PATH, BUILD_PATH)
self.test_project.run_build('python project_setup.py build_project --build-package')
# when
self.test_project.run_build('python project_setup.py build_project --build-image')
# then
self.assertFalse(image_leftovers_exist())
self.assertFalse(docker_image_built_in_registry(DOCKER_REPOSITORY, '0.1.0'))
self.assertTrue(docker_image_as_file_built())
self.assertTrue(deployment_config_copied())