def load_rules(rule_dir, extra_stepset, context):
base_rule = Rule()
base_rule.context = context
if extra_stepset:
base_rule.add_step_set(load_mod_var(extra_stepset))
base_rule.add_step_set(basic_step_set)
base_rule.add_step_set(html_step_set)
return base_rule.load_directory(rule_dir)
def load_rules(rule_dir, extra_stepset, context):
base_rule = Rule()
base_rule.context = context
if extra_stepset:
base_rule.add_step_set(load_mod_var(extra_stepset))
base_rule.add_step_set(basic_step_set)
base_rule.add_step_set(html_step_set)
return base_rule.load_directory(rule_dir)
def setUp(self):
self.rule = Rule()
def step_fn(context, args):
context.calls = context.get('calls', 0) + 1
return args[1] + str(context.calls)
self.step = RegexFuncStep('(.+)', step_fn)
self.rule.step_set.add_step(self.step)
self.rule_text = dedent(u"""\
title: Check yo self
category: test rule
points: 42
...
This bit: the step text.
This bit: the step text.
""")
def setUp(self):
self.rule = Rule()
def step_fn(context, args):
context.calls = context.get('calls', 0) + 1
return args[1] + str(context.calls)
self.step = RegexFuncStep('(.+)', step_fn)
self.rule.step_set.add_step(self.step)
self.rule_text = dedent(u"""\
title: Check yo self
category: test rule
points: 42
...
This bit: the step text.
This bit: the step text.
""")
class RuleTest(TestCase):
def setUp(self):
self.rule = Rule()
def step_fn(context, args):
context.calls = context.get('calls', 0) + 1
return args[1] + str(context.calls)
self.step = RegexFuncStep('(.+)', step_fn)
self.rule.step_set.add_step(self.step)
self.rule_text = dedent(u"""\
title: Check yo self
category: test rule
points: 42
...
This bit: the step text.
This bit: the step text.
""")
def test_add_step(self):
self.rule.step_set.add_step = Mock(self.rule.step_set.add_step)
self.rule.add_step('notastep')
self.rule.step_set.add_step.assert_called_with('notastep')
def test_add_step_set(self):
ss = self.rule.step_set
replacement_ss = StepSet()
ss.concat = Mock(return_value=replacement_ss)
new_step_set = StepSet()
self.rule.add_step_set(new_step_set)
ss.concat.assert_called_with(new_step_set)
self.assertIs(self.rule.step_set, replacement_ss)
def test_parse_metadata(self):
metadata, i = self.rule.parse_metadata(self.rule_text, 0)
expected = {
'title': 'Check yo self',
'category': 'test rule',
'points': 42}
self.assertEqual(metadata, expected)
self.assertEqual(i, 55)
def test_parse_steps(self):
steps = self.rule.parse_steps(self.rule_text, 55)
expected = [({1: 'This bit: the step text.'}, self.step),
({1: 'This bit: the step text.'}, self.step)]
self.assertEqual(steps, expected)
def test_parse(self):
self.rule.parse(self.rule_text)
expected_meta = {
'title': 'Check yo self',
'category': 'test rule',
'points': 42}
self.assertEqual(self.rule.metadata, expected_meta)
expected_steps = [({1: 'This bit: the step text.'}, self.step),
({1: 'This bit: the step text.'}, self.step)]
self.assertEqual(self.rule.steps, expected_steps)
def test_parse_no_metadata(self):
self.rule.parse('No YAML here')
self.assertEqual(self.rule.metadata, {})
self.assertEqual(self.rule.steps, [({1: 'No YAML here'}, self.step)])
def test_run(self):
self.rule.parse(self.rule_text)
self.rule.run(extra_context_arg=42)
expected_ctx = {
'calls': 2,
'last_return': 'This bit: the step text.2',
'extra_context_arg': 42
}
self.assertEqual(self.rule.context, expected_ctx)
def test_load(self):
file_path = join(dirname(__file__), 'rules', 'simple.rule')
self.rule.load(file_path)
expected_steps = [({1: 'This bit: the step text.'}, self.step),
({1: 'This bit: the step text.'}, self.step)]
self.assertEqual(self.rule.steps, expected_steps)
self.assertEqual(self.rule.file_path, file_path)
def test_load_directory(self):
rules = self.rule.load_directory(join(dirname(__file__), 'rules'))
expected_rules = [
({'title': 'Another test rule',
'category': 'test rule',
'points': -12},
[({1: 'This bit: the step text.'}, self.step)]),
({'title': 'Check yo self',
'category': 'test rule',
'points': 42},
[({1: 'This bit: the step text.'}, self.step),
({1: 'This bit: the step text.'}, self.step)])
]
self.assertEqual([(r.metadata, r.steps)
for r in rules], expected_rules)
class RuleTest(TestCase):
def setUp(self):
self.rule = Rule()
def step_fn(context, args):
context.calls = context.get('calls', 0) + 1
return args[1] + str(context.calls)
self.step = RegexFuncStep('(.+)', step_fn)
self.rule.step_set.add_step(self.step)
self.rule_text = dedent(u"""\
title: Check yo self
category: test rule
points: 42
...
This bit: the step text.
This bit: the step text.
""")
def test_add_step(self):
self.rule.step_set.add_step = Mock(self.rule.step_set.add_step)
self.rule.add_step('notastep')
self.rule.step_set.add_step.assert_called_with('notastep')
def test_add_step_set(self):
ss = self.rule.step_set
replacement_ss = StepSet()
ss.concat = Mock(return_value=replacement_ss)
new_step_set = StepSet()
self.rule.add_step_set(new_step_set)
ss.concat.assert_called_with(new_step_set)
self.assertIs(self.rule.step_set, replacement_ss)
def test_parse_metadata(self):
metadata, i = self.rule.parse_metadata(self.rule_text, 0)
expected = {
'title': 'Check yo self',
'category': 'test rule',
'points': 42
}
self.assertEqual(metadata, expected)
self.assertEqual(i, 55)
def test_parse_steps(self):
steps = self.rule.parse_steps(self.rule_text, 55)
expected = [({
1: 'This bit: the step text.'
}, self.step), ({
1: 'This bit: the step text.'
}, self.step)]
self.assertEqual(steps, expected)
def test_parse(self):
self.rule.parse(self.rule_text)
expected_meta = {
'title': 'Check yo self',
'category': 'test rule',
'points': 42
}
self.assertEqual(self.rule.metadata, expected_meta)
expected_steps = [({
1: 'This bit: the step text.'
}, self.step), ({
1: 'This bit: the step text.'
}, self.step)]
self.assertEqual(self.rule.steps, expected_steps)
def test_parse_no_metadata(self):
self.rule.parse('No YAML here')
self.assertEqual(self.rule.metadata, {})
self.assertEqual(self.rule.steps, [({1: 'No YAML here'}, self.step)])
def test_run(self):
self.rule.parse(self.rule_text)
self.rule.run(extra_context_arg=42)
expected_ctx = {
'calls': 2,
'last_return': 'This bit: the step text.2',
'extra_context_arg': 42
}
self.assertEqual(self.rule.context, expected_ctx)
def test_load(self):
file_path = join(dirname(__file__), 'rules', 'simple.rule')
self.rule.load(file_path)
expected_steps = [({
1: 'This bit: the step text.'
}, self.step), ({
1: 'This bit: the step text.'
}, self.step)]
self.assertEqual(self.rule.steps, expected_steps)
self.assertEqual(self.rule.file_path, file_path)
def test_load_directory(self):
rules = self.rule.load_directory(join(dirname(__file__), 'rules'))
expected_rules = [({
'title': 'Another test rule',
'category': 'test rule',
'points': -12
}, [({
1: 'This bit: the step text.'
}, self.step)]),
({
'title': 'Check yo self',
'category': 'test rule',
'points': 42
}, [({
1: 'This bit: the step text.'
}, self.step),
({
1: 'This bit: the step text.'
}, self.step)])]
self.assertEqual([(r.metadata, r.steps) for r in rules],
expected_rules)
