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()