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