class EngineTest(TestCase):
def setUp(self):
super(EngineTest, self).setUp()
self.log_mock = Logger()
self.log_info_mock = MagicMock()
self.log_downloaded_mock = MagicMock()
self.log_failed_mock = MagicMock()
self.log_mock.info = self.log_info_mock
self.log_mock.downloaded = self.log_downloaded_mock
self.log_mock.failed = self.log_failed_mock
self.clients_manager = ClientsManager()
self.engine = Engine(self.log_mock, self.clients_manager)
def test_engine_find_torrent(self):
finded_torrent = {'date_added': datetime.now(pytz.utc)}
self.clients_manager.find_torrent = MagicMock(return_value=finded_torrent)
result = self.engine.find_torrent('hash')
self.assertEqual(finded_torrent, result)
@data(True, False)
def test_engine_remove_torrent(self, value):
self.clients_manager.remove_torrent = MagicMock(return_value=value)
self.assertEqual(value, self.engine.remove_torrent('hash'))
class EngineTest(TestCase):
def setUp(self):
super(EngineTest, self).setUp()
self.log_mock = Logger()
self.log_info_mock = MagicMock()
self.log_downloaded_mock = MagicMock()
self.log_failed_mock = MagicMock()
self.log_mock.info = self.log_info_mock
self.log_mock.downloaded = self.log_downloaded_mock
self.log_mock.failed = self.log_failed_mock
self.clients_manager = ClientsManager()
self.engine = Engine(self.log_mock, self.clients_manager)
def test_engine_find_torrent(self):
finded_torrent = {'date_added': datetime.now(pytz.utc)}
self.clients_manager.find_torrent = MagicMock(
return_value=finded_torrent)
result = self.engine.find_torrent('hash')
self.assertEqual(finded_torrent, result)
@data(True, False)
def test_engine_remove_torrent(self, value):
self.clients_manager.remove_torrent = MagicMock(return_value=value)
self.assertEqual(value, self.engine.remove_torrent('hash'))
def setUp(self):
super(EngineTest, self).setUp()
self.log_mock = Logger()
self.log_info_mock = MagicMock()
self.log_downloaded_mock = MagicMock()
self.log_failed_mock = MagicMock()
self.log_mock.info = self.log_info_mock
self.log_mock.downloaded = self.log_downloaded_mock
self.log_mock.failed = self.log_failed_mock
self.clients_manager = ClientsManager()
self.engine = Engine(self.log_mock, self.clients_manager)
def setUp(self):
super(EngineTest, self).setUp()
self.log_mock = Logger()
self.log_info_mock = MagicMock()
self.log_downloaded_mock = MagicMock()
self.log_failed_mock = MagicMock()
self.log_mock.info = self.log_info_mock
self.log_mock.downloaded = self.log_downloaded_mock
self.log_mock.failed = self.log_failed_mock
self.clients_manager = ClientsManager()
self.engine = Engine(self.log_mock, self.clients_manager)
def setUp(self):
super(EngineTest, self).setUp()
self.log_mock = Logger()
self.log_info_mock = MagicMock()
self.log_downloaded_mock = MagicMock()
self.log_failed_mock = MagicMock()
self.log_mock.info = self.log_info_mock
self.log_mock.downloaded = self.log_downloaded_mock
self.log_mock.failed = self.log_failed_mock
self.clients_manager = ClientsManager()
self.settings_manager = Mock()
self.trackers_manager = TrackersManager(self.settings_manager, {})
self.notifier_manager = NotifierManager({})
self.engine = Engine(self.log_mock, self.settings_manager,
self.trackers_manager, self.clients_manager,
self.notifier_manager)
