53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
import unittest
|
|
from saucebrush import Recipe, run_recipe, SaucebrushError, OvercookedError
|
|
from saucebrush.filters import Filter
|
|
|
|
|
|
class Raiser(Filter):
|
|
def process_record(self, record):
|
|
raise Exception("bad record")
|
|
|
|
|
|
class Saver(Filter):
|
|
def __init__(self):
|
|
self.saved = []
|
|
|
|
def process_record(self, record):
|
|
self.saved.append(record)
|
|
return record
|
|
|
|
|
|
class RecipeTestCase(unittest.TestCase):
|
|
def test_error_stream(self):
|
|
saver = Saver()
|
|
recipe = Recipe(Raiser(), error_stream=saver)
|
|
recipe.run([{"a": 1}, {"b": 2}])
|
|
recipe.done()
|
|
|
|
self.assertEqual(saver.saved[0]["record"], {"a": 1})
|
|
self.assertEqual(saver.saved[1]["record"], {"b": 2})
|
|
|
|
# Must pass either a Recipe, a Filter or an iterable of Filters
|
|
# as the error_stream argument
|
|
self.assertRaises(SaucebrushError, Recipe, error_stream=5)
|
|
|
|
def test_run_recipe(self):
|
|
saver = Saver()
|
|
run_recipe([1, 2], saver)
|
|
|
|
self.assertEqual(saver.saved, [1, 2])
|
|
|
|
def test_done(self):
|
|
saver = Saver()
|
|
recipe = Recipe(saver)
|
|
recipe.run([1])
|
|
recipe.done()
|
|
|
|
self.assertRaises(OvercookedError, recipe.run, [2])
|
|
self.assertRaises(OvercookedError, recipe.done)
|
|
self.assertEqual(saver.saved, [1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|