Skip to content

Commit 3440064

Browse files
author
Taniya Mathur
committed
updated tests and added main lambda functions for validation
1 parent 192fae3 commit 3440064

File tree

2 files changed

+210
-100
lines changed

2 files changed

+210
-100
lines changed

lib/idp_common_pkg/tests/unit/test_publish.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1048,13 +1048,17 @@ def test_run_minimal_success_flow(self):
10481048
patch.object(publisher, "setup_artifacts_bucket") as mock_setup_bucket,
10491049
patch.object(publisher, "clean_temp_files"),
10501050
patch.object(publisher, "clean_lib"),
1051+
patch.object(
1052+
publisher, "ensure_idp_common_library_ready"
1053+
) as mock_ensure_lib,
10511054
patch.object(publisher, "needs_rebuild", return_value=False),
10521055
patch.object(
10531056
publisher, "build_patterns_concurrently", return_value=True
10541057
) as mock_build_patterns,
10551058
patch.object(
10561059
publisher, "build_options_concurrently", return_value=True
10571060
) as mock_build_options,
1061+
patch.object(publisher, "validate_lambda_builds") as mock_validate_builds,
10581062
patch.object(publisher, "upload_config_library") as mock_upload_config,
10591063
patch.object(
10601064
publisher, "package_ui", return_value="test-ui.zip"
@@ -1075,8 +1079,10 @@ def test_run_minimal_success_flow(self):
10751079
mock_check_prereq.assert_called_once()
10761080
mock_ensure_sam.assert_called_once()
10771081
mock_setup_bucket.assert_called_once()
1082+
mock_ensure_lib.assert_called_once()
10781083
mock_build_patterns.assert_called_once()
10791084
mock_build_options.assert_called_once()
1085+
mock_validate_builds.assert_called_once()
10801086
mock_upload_config.assert_called_once()
10811087
mock_package_ui.assert_called_once()
10821088
mock_build_main.assert_called_once_with("test-ui.zip")
@@ -1127,6 +1133,7 @@ def test_run_pattern_build_failure(self):
11271133
patch.object(publisher, "setup_artifacts_bucket"),
11281134
patch.object(publisher, "clean_temp_files"),
11291135
patch.object(publisher, "clean_lib"),
1136+
patch.object(publisher, "ensure_idp_common_library_ready"),
11301137
patch.object(publisher, "needs_rebuild", return_value=False),
11311138
patch.object(publisher, "build_patterns_concurrently", return_value=False),
11321139
patch.object(publisher.console, "print") as mock_print,

0 commit comments

Comments
 (0)