Skip to content

Commit 4156b9b

Browse files
committed
fix: update python tests
Signed-off-by: Felipe Zipitria <felipe.zipitria@owasp.org>
1 parent 3789453 commit 4156b9b

2 files changed

Lines changed: 19 additions & 1 deletion

File tree

src/seclang_parser/listener.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def enterRemove_rule_by_msg(self, ctx: SecLangParser.Remove_rule_by_msgContext):
6464
def enterRemove_rule_by_tag(self, ctx: SecLangParser.Remove_rule_by_tagContext):
6565
self.results.directive_list.append(ctx.getText())
6666

67-
def enterValues(self, ctx: SecLangParser.ValuesContext):
67+
def enterString_remove_rules_values(self, ctx: SecLangParser.String_remove_rules_valuesContext):
6868
self.results.directive_values.append(ctx.getText())
6969

7070
def enterUpdate_target_by_id(self, ctx: SecLangParser.Update_target_by_idContext):
@@ -119,3 +119,9 @@ def enterAssignment(self, ctx: SecLangParser.AssignmentContext):
119119

120120
def enterVar_assignment(self, ctx: SecLangParser.Var_assignmentContext):
121121
self.results.directive_values.append(ctx.getText())
122+
123+
def enterCtl_action(self, ctx: SecLangParser.Ctl_actionContext):
124+
self.results.directive_values.append(ctx.getText())
125+
126+
def enterCtl_id(self, ctx: SecLangParser.Ctl_idContext):
127+
self.results.directive_values.append(ctx.getText())

tests/test_parser.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -358,6 +358,18 @@
358358
directive_values=["bar", "0", "2", "1", "/%{tx.0}/", "%{tx.critical_anomaly_score}"],
359359
),
360360
},
361+
"testdata/test_43_colon.conf": {
362+
"error_count": 0,
363+
"comment": "",
364+
"expected_result": ParserResult(
365+
collections=["TX", "TX"],
366+
operator_list=["unconditionalMatch"],
367+
directive_list=["SecRule"],
368+
collection_args=["paramcounter_ARGS_NAMES:folders.folders", "paramcounter_ARGS_NAMES:folders.folders"],
369+
directive_values=["ruleRemoveTargetById", "921180"],
370+
setvar_operations=["="],
371+
),
372+
},
361373
}
362374

363375

0 commit comments

Comments
 (0)