def test_generate_files(app_config, should_find, test_string): result = gen_dockerfile.generate_files(app_config) assert set(result.keys()) == EXPECTED_OUTPUT_FILES dockerfile = result['Dockerfile'] if should_find: assert test_string in dockerfile else: assert test_string not in dockerfile