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_periodic_scheduling_calling_read_method(self):
q = get_queue()
conf = {'period' : 1}
myreader = Reader(q, conf=conf)
myreader.read = MagicMock(return_value=True)
myreader.start()
# waits reader period
time.sleep(1)
myreader.read.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()
