1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import unittest
import tempfile
import json5
from pathlib import Path
from conventionalrp.extractors.rule_extractor import RuleExtractor
class TestRuleExtractor(unittest.TestCase):
def setUp(self):
self.temp_rules = tempfile.NamedTemporaryFile(
mode='w',
suffix='.json5',
delete=False,
encoding='utf-8'
)
self.temp_rules.write('''{
test_rule: "test_value",
metadata: [{type: "test"}],
content: [{type: "test_content"}]
}''')
self.temp_rules.close()
def tearDown(self):
Path(self.temp_rules.name).unlink(missing_ok=True)
def test_init_without_file(self):
extractor = RuleExtractor()
self.assertEqual(extractor.rules, {})
self.assertIsNone(extractor.config_file)
def test_init_with_file(self):
extractor = RuleExtractor(self.temp_rules.name)
self.assertIsNotNone(extractor.rules)
self.assertIn("test_rule", extractor.rules)
def test_load_rules_from_file_success(self):
extractor = RuleExtractor()
rules = extractor.load_rules_from_file(self.temp_rules.name)
self.assertIsInstance(rules, dict)
self.assertIn("test_rule", rules)
self.assertEqual(rules["test_rule"], "test_value")
def test_load_rules_from_file_not_found(self):
extractor = RuleExtractor()
with self.assertRaises(FileNotFoundError):
extractor.load_rules_from_file("nonexistent.json5")
def test_load_rules_empty_file(self):
empty_file = tempfile.NamedTemporaryFile(
mode='w',
suffix='.json5',
delete=False,
encoding='utf-8'
)
empty_file.write('')
empty_file.close()
try:
extractor = RuleExtractor()
with self.assertRaises(ValueError):
extractor.load_rules_from_file(empty_file.name)
finally:
Path(empty_file.name).unlink(missing_ok=True)
def test_load_rules_method(self):
extractor = RuleExtractor()
rules = extractor.load_rules(self.temp_rules.name)
self.assertIsInstance(rules, dict)
self.assertEqual(extractor.rules, rules)
def test_extract_method(self):
extractor = RuleExtractor(self.temp_rules.name)
extracted = extractor.extract()
self.assertIsInstance(extracted, dict)
self.assertEqual(extracted, extractor.rules)
if __name__ == "__main__":
unittest.main()
|