def test_commit(self, unlink):
sn = '123'
path = '/tmp/123'
p = Pending('')
p.journal = {sn: path}
p.commit(sn)
unlink.assert_called_once_with(path)
self.assertEqual(p.journal, {})
def test_commit_not_found(self, unlink):
sn = '123'
path = '/tmp/123'
p = Pending('')
p.journal = {sn: path}
p.commit('invalid')
self.assertFalse(unlink.called)
self.assertEqual(p.journal, {sn: path})
def __init__(self, plugin):
"""
:param plugin: A plugin.
:type plugin: gofer.agent.plugin.Plugin
"""
Thread.__init__(self, name='scheduler:%s' % plugin.name)
self.plugin = plugin
self.pending = Pending(plugin.name)
self.setDaemon(True)
