def test_add_list_info_and_remove(self):
count_before = len(utils.torrent_list())
self.assertEqual(utils.torrent_add(TORRENT_MAGNET), TRANSMISSION_MSG_SUCCESS)
self.assertEqual(utils.torrent_add(TORRENT_MAGNET), 'Error: duplicate torrent')
torrents = utils.torrent_list()
count_after = len(torrents)
added_torrent = torrents[-1]
self.assertTrue(utils.torrent_info(added_torrent.id))
self.assertEqual(count_after, count_before + 1)
self.assertEqual(utils.torrent_remove(added_torrent.id), TRANSMISSION_MSG_SUCCESS)
def torrent_add(message):
log.info("Handling command add torrent - %s" % message.text)
args = message.text.split()
if len(args) < 2:
log.warning("Missing torrent URL")
bot.reply_to(message, "Missing torrent URL. Usage: /add [URL]")
return
url = args[1]
response = utils.torrent_add(url)
if "success" in response:
log.info("Torrent added")
bot.reply_to(message, "Torrent added")
else:
log.info("Error adding torrent")
bot.reply_to(message, response)
def torrent_add(message):
log.info("Handling command add torrent - %s" % message.text)
args = message.text.split()
if len(args) < 2:
log.warn("Missing torrent URL")
bot.reply_to(message, "Missing torrent URL. Usage: /torrent_add [URL]")
return
url = args[1]
response = utils.torrent_add(url)
if "success" in response:
log.info("Torrent added")
bot.reply_to(message, "Torrent added")
else:
log.info("Error adding torrent")
bot.reply_to(message, response)
def test_torrent_add_invalid(self):
self.assertEqual(utils.torrent_add("magnet:?xt=urn:btih:2013fdfff"),
TRANSMISSION_MSG_ADD_INVALID_TORRENT)
def test_torrent_add_empty(self):
self.assertEqual(utils.torrent_add(""),
TRANSMISSION_MSG_ADD_INVALID_TORRENT)
