def test_scheduling_checkpoint_when_it_is_enabled(self):
# mocking
q = get_queue()
myreader = Reader(q)
myreader.schedule_checkpoint_writing = MagicMock()
conf = {'checkpoint_enabled' : True,
'checkpoint_path' : '/tmp/checkpoint'}
myreader.__init__(q, conf=conf)
# after __init__, methods, should be called
myreader.schedule_checkpoint_writing.assert_called_with()
def test_initialization_calls(self):
# mocking
q = get_queue()
myreader = Reader(q)
myreader.set_conf = MagicMock()
myreader.setup = MagicMock()
myreader.schedule_tasks = MagicMock()
conf = {'foo' : 'bar'}
myreader.__init__(q, conf=conf)
# after __init__, methods should be called
myreader.set_conf.assert_called_with(conf)
myreader.setup.assert_called_with()
myreader.schedule_tasks.assert_called_with()
