def test_if_cached_valid_when_build_single_function_definition(
self, dir_checksum_mock, exists_mock, copytree_mock):
pass
with osutils.mkdir_temp() as temp_base_dir:
build_dir = Path(temp_base_dir, ".aws-sam", "build")
build_dir.mkdir(parents=True)
cache_dir = Path(temp_base_dir, ".aws-sam", "cache")
cache_dir.mkdir(parents=True)
exists_mock.return_value = True
dir_checksum_mock.return_value = CachedBuildStrategyTest.SOURCE_MD5
build_graph_path = Path(build_dir.parent, "build.toml")
build_graph_path.write_text(
CachedBuildStrategyTest.BUILD_GRAPH_CONTENTS)
build_graph = BuildGraph(str(build_dir))
cached_build_strategy = CachedBuildStrategy(
build_graph, DefaultBuildStrategy, temp_base_dir, build_dir,
cache_dir, True)
func1 = Mock()
func1.name = "func1_name"
func2 = Mock()
func2.name = "func2_name"
build_definition = build_graph.get_function_build_definitions()[0]
layer_definition = build_graph.get_layer_build_definitions()[0]
build_graph.put_function_build_definition(build_definition, func1)
build_graph.put_function_build_definition(build_definition, func2)
layer = Mock()
layer.name = "layer_name"
build_graph.put_layer_build_definition(layer_definition, layer)
cached_build_strategy.build_single_function_definition(
build_definition)
cached_build_strategy.build_single_layer_definition(
layer_definition)
self.assertEqual(copytree_mock.call_count, 3)
def test_if_cached_invalid_with_no_cached_folder(self, build_layer_mock,
build_function_mock,
copytree_mock):
with osutils.mkdir_temp() as temp_base_dir:
build_dir = Path(temp_base_dir, ".aws-sam", "build")
build_dir.mkdir(parents=True)
cache_dir = Path(temp_base_dir, ".aws-sam", "cache")
cache_dir.mkdir(parents=True)
build_function_mock.return_value = {
"HelloWorldPython": "artifact1",
"HelloWorldPython2": "artifact2"
}
build_layer_mock.return_value = {"SumLayer": "artifact3"}
build_graph_path = Path(build_dir.parent, "build.toml")
build_graph_path.write_text(
CachedBuildStrategyTest.BUILD_GRAPH_CONTENTS)
build_graph = BuildGraph(str(build_dir))
cached_build_strategy = CachedBuildStrategy(
build_graph, DefaultBuildStrategy, temp_base_dir, build_dir,
cache_dir, True)
cached_build_strategy.build_single_function_definition(
build_graph.get_function_build_definitions()[0])
cached_build_strategy.build_single_layer_definition(
build_graph.get_layer_build_definitions()[0])
build_function_mock.assert_called_once()
build_layer_mock.assert_called_once()
self.assertEqual(copytree_mock.call_count, 2)
def _build_layers(self, build_graph: BuildGraph) -> Dict[str, str]:
"""
Iterates through build graph and runs each unique build and copies outcome to the corresponding layer folder
"""
layer_build_results = {}
for layer_definition in build_graph.get_layer_build_definitions():
layer_build_results.update(self.build_single_layer_definition(layer_definition))
return layer_build_results