@@ -22,6 +22,7 @@ def get_reportsizedeltas_object(
2222 repository_name : str = "FooOwner/BarRepository" ,
2323 sketches_reports_source : str = "foo-artifact-pattern" ,
2424 token : str = "foo token" ,
25+ update_comment : bool = False ,
2526) -> reportsizedeltas .ReportSizeDeltas :
2627 """Return a reportsizedeltas.ReportSizeDeltas object to use in tests.
2728
@@ -32,7 +33,10 @@ def get_reportsizedeltas_object(
3233 token -- GitHub access token
3334 """
3435 return reportsizedeltas .ReportSizeDeltas (
35- repository_name = repository_name , sketches_reports_source = sketches_reports_source , token = token
36+ repository_name = repository_name ,
37+ sketches_reports_source = sketches_reports_source ,
38+ token = token ,
39+ update_comment = update_comment ,
3640 )
3741
3842
@@ -106,10 +110,12 @@ class ActionInputs:
106110 repository_name = "GoldenOwner/GoldenRepository"
107111 sketches_reports_source = "golden-source-pattern"
108112 token = "golden-github-token"
113+ update_comment = "false"
109114
110115 monkeypatch .setenv ("GITHUB_REPOSITORY" , ActionInputs .repository_name )
111116 monkeypatch .setenv ("INPUT_SKETCHES-REPORTS-SOURCE" , ActionInputs .sketches_reports_source )
112117 monkeypatch .setenv ("INPUT_GITHUB-TOKEN" , ActionInputs .token )
118+ monkeypatch .setenv ("INPUT_UPDATE-COMMENT" , ActionInputs .update_comment )
113119
114120 return ActionInputs ()
115121
@@ -132,6 +138,7 @@ def report_size_deltas(self):
132138 repository_name = setup_environment_variables .repository_name ,
133139 sketches_reports_source = setup_environment_variables .sketches_reports_source ,
134140 token = setup_environment_variables .token ,
141+ update_comment = setup_environment_variables .update_comment == "true" ,
135142 )
136143 ReportSizeDeltas .report_size_deltas .assert_called_once ()
137144
@@ -768,7 +775,6 @@ def test_comment_report():
768775 report_size_deltas = get_reportsizedeltas_object (repository_name = repository_name )
769776
770777 report_size_deltas .http_request = unittest .mock .MagicMock ()
771- report_size_deltas .get_previous_comment = unittest .mock .Mock (return_value = None )
772778
773779 report_size_deltas .comment_report (pr_number = pr_number , report_markdown = report_markdown )
774780
0 commit comments