Skip to content

Commit ebb4bbf

Browse files
committed
convert test_cases and failure_test_cases to pytest.param
1 parent 8e648b3 commit ebb4bbf

File tree

12 files changed

+416
-844
lines changed

12 files changed

+416
-844
lines changed

tests/unit/ng/processor/calculator/test_calculator.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,21 +32,21 @@ class TestCalculator(BaseProcessorTestCase):
3232
"rules": ["tests/testdata/unit/calculator/rules"],
3333
}
3434

35-
@pytest.mark.parametrize("testcase, rule, event, expected", test_cases)
36-
def test_testcases(self, testcase, rule, event, expected): # pylint: disable=unused-argument
35+
@pytest.mark.parametrize("rule, event, expected", test_cases)
36+
def test_testcases(self, rule, event, expected):
3737
self._load_rule(rule)
3838
event = LogEvent(event, original=b"")
3939
self.object.process(event)
40-
assert event.data == expected, testcase
40+
assert event.data == expected
4141

42-
@pytest.mark.parametrize("testcase, rule, event, expected, error_message", failure_test_cases)
43-
def test_testcases_failure_handling(self, testcase, rule, event, expected, error_message):
42+
@pytest.mark.parametrize("rule, event, expected, error_message", failure_test_cases)
43+
def test_testcases_failure_handling(self, rule, event, expected, error_message):
4444
self._load_rule(rule)
4545
event = LogEvent(event, original=b"")
4646
result = self.object.process(event)
4747
assert len(result.warnings) == 1
4848
assert re.match(rf".*{error_message}", str(result.warnings[0]))
49-
assert event.data == expected, testcase
49+
assert event.data == expected
5050

5151
@pytest.mark.parametrize(
5252
"expression, expected",

tests/unit/ng/processor/dissector/test_dissector.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,18 +24,18 @@ class TestDissector(BaseProcessorTestCase):
2424
"rules": ["tests/testdata/unit/dissector/rules"],
2525
}
2626

27-
@pytest.mark.parametrize("testcase, rule, event, expected", test_cases)
28-
def test_testcases(self, testcase, rule, event, expected): # pylint: disable=unused-argument
27+
@pytest.mark.parametrize("rule, event, expected", test_cases)
28+
def test_testcases(self, rule, event, expected):
2929
self._load_rule(rule)
3030
log_event = LogEvent(event, original=b"test_message")
3131
self.object.process(log_event)
3232
assert log_event.data == expected
3333

34-
@pytest.mark.parametrize("testcase, rule, event, expected", failure_test_cases)
35-
def test_testcases_failure_handling(self, testcase, rule, event, expected):
34+
@pytest.mark.parametrize("rule, event, expected", failure_test_cases)
35+
def test_testcases_failure_handling(self, rule, event, expected):
3636
self._load_rule(rule)
3737
log_event = LogEvent(event, original=b"test_message")
3838
result = self.object.process(log_event)
3939
assert len(result.warnings) == 1
4040
assert isinstance(result.warnings[0], ProcessingWarning)
41-
assert log_event.data == expected, testcase
41+
assert log_event.data == expected

tests/unit/ng/processor/generic_adder/test_generic_adder.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,25 +37,21 @@ class TestGenericAdder(BaseProcessorTestCase):
3737
"rules": ["tests/testdata/unit/generic_adder/rules"],
3838
}
3939

40-
@pytest.mark.parametrize("testcase, rule, event, expected", test_cases)
41-
def test_generic_adder_testcases(
42-
self, testcase, rule, event, expected
43-
): # pylint: disable=unused-argument
40+
@pytest.mark.parametrize("rule, event, expected", test_cases)
41+
def test_generic_adder_testcases(self, rule, event, expected):
4442
self._load_rule(rule)
4543
log_event = LogEvent(event, original=b"")
4644
self.object.process(log_event)
4745
assert event == expected
4846

49-
@pytest.mark.parametrize("testcase, rule, event, expected, error_message", failure_test_cases)
50-
def test_generic_adder_testcases_failure_handling(
51-
self, testcase, rule, event, expected, error_message
52-
):
47+
@pytest.mark.parametrize("rule, event, expected, error_message", failure_test_cases)
48+
def test_generic_adder_testcases_failure_handling(self, rule, event, expected, error_message):
5349
self._load_rule(rule)
5450
log_event = LogEvent(event, original=b"")
5551
result = self.object.process(log_event)
5652
assert len(result.warnings) == 1
5753
assert re.match(rf".*FieldExistsWarning.*{error_message}", str(result.warnings[0]))
58-
assert event == expected, testcase
54+
assert event == expected
5955

6056
def test_add_generic_fields_from_file_missing_and_existing_with_all_required(self):
6157
with pytest.raises(InvalidRuleDefinitionError, match=r"files do not exist"):

0 commit comments

Comments
 (0)