Skip to content

Commit b1c1f0b

Browse files
Balijepalli Vamshi KrishnaBalijepalli Vamshi Krishna
authored andcommitted
handle missing action error
1 parent c62caaf commit b1c1f0b

File tree

1 file changed

+18
-8
lines changed

1 file changed

+18
-8
lines changed

remediation/workflow/secureworkflow_test.go

Lines changed: 18 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -214,15 +214,17 @@ func TestSecureWorkflow(t *testing.T) {
214214
wantAddedHardenRunner bool
215215
wantAddedPermissions bool
216216
wantAddedMaintainedActions bool
217+
wantError bool
217218
}{
218-
{fileName: "oneJob.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: false, wantAddedMaintainedActions: true},
219-
{fileName: "allscenarios.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: true},
220-
{fileName: "nohardenrunner.yml", wantPinnedActions: true, wantAddedHardenRunner: false, wantAddedPermissions: true},
221-
{fileName: "noperms.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: false},
222-
{fileName: "nopin.yml", wantPinnedActions: false, wantAddedHardenRunner: true, wantAddedPermissions: true},
223-
{fileName: "allperms.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: true},
224-
{fileName: "multiplejobperms.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: true},
225-
{fileName: "error.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: false},
219+
{fileName: "replaceactions.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: false, wantAddedMaintainedActions: true, wantError: false},
220+
{fileName: "allscenarios.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: true, wantError: false},
221+
{fileName: "nohardenrunner.yml", wantPinnedActions: true, wantAddedHardenRunner: false, wantAddedPermissions: true, wantError: false},
222+
{fileName: "noperms.yml", wantPinnedActions: true, wantAddedHardenRunner: true, wantAddedPermissions: false, wantError: false},
223+
{fileName: "nopin.yml", wantPinnedActions: false, wantAddedHardenRunner: true, wantAddedPermissions: true, wantError: false},
224+
{fileName: "allperms.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: true, wantError: false},
225+
{fileName: "multiplejobperms.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: true, wantError: false},
226+
{fileName: "error.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: false, wantError: false},
227+
{fileName: "missingaction.yml", wantPinnedActions: false, wantAddedHardenRunner: false, wantAddedPermissions: false, wantError: true},
226228
}
227229
for _, test := range tests {
228230
var err error
@@ -269,6 +271,14 @@ func TestSecureWorkflow(t *testing.T) {
269271
output, err = SecureWorkflow(queryParams, string(input), &mockDynamoDBClient{})
270272
}
271273

274+
if test.wantError {
275+
if err == nil {
276+
t.Errorf("test failed %s expected an error but got none", test.fileName)
277+
}
278+
// Skip further validation if we expected an error
279+
continue
280+
}
281+
272282
if err != nil {
273283
t.Log(err)
274284
t.Errorf("Error not expected")

0 commit comments

Comments
 (0)