deluge/tests/test_maketorrent.py

from twisted.trial import unittest
from twisted.python.failure import Failure

import os
import tempfile

from deluge import maketorrent


def check_torrent(filename):
    # Test loading with libtorrent to make sure it's valid
    from deluge._libtorrent import lt
    lt.torrent_info(filename)

    # Test loading with our internal TorrentInfo class
    from deluge.ui.common import TorrentInfo
    ti = TorrentInfo(filename)

class MakeTorrentTestCase(unittest.TestCase):
    def test_save_multifile(self):
        # Create a temporary folder for torrent creation
        tmp_path = tempfile.mkdtemp()
        open(os.path.join(tmp_path, "file_A"), "wb").write("a" * (312 * 1024))
        open(os.path.join(tmp_path, "file_B"), "wb").write("b" * (2354 * 1024))
        open(os.path.join(tmp_path, "file_C"), "wb").write("c" * (11 * 1024))

        t = maketorrent.TorrentMetadata()
        t.data_path = tmp_path
        tmp_file = tempfile.mkstemp(".torrent")[1]
        t.save(tmp_file)

        check_torrent(tmp_file)

        os.remove(os.path.join(tmp_path, "file_A"))
        os.remove(os.path.join(tmp_path, "file_B"))
        os.remove(os.path.join(tmp_path, "file_C"))
        os.rmdir(tmp_path)
        os.remove(tmp_file)

    def test_save_singlefile(self):
        tmp_data = tempfile.mkstemp("testdata")[1]
        open(tmp_data, "wb").write("a"*(2314*1024))
        t = maketorrent.TorrentMetadata()
        t.data_path = tmp_data
        tmp_file = tempfile.mkstemp(".torrent")[1]
        t.save(tmp_file)

        check_torrent(tmp_file)

        os.remove(tmp_data)
        os.remove(tmp_file)

    def test_save_multifile_padded(self):
        # Create a temporary folder for torrent creation
        tmp_path = tempfile.mkdtemp()
        open(os.path.join(tmp_path, "file_A"), "wb").write("a" * (312 * 1024))
        open(os.path.join(tmp_path, "file_B"), "wb").write("b" * (2354 * 1024))
        open(os.path.join(tmp_path, "file_C"), "wb").write("c" * (11 * 1024))

        t = maketorrent.TorrentMetadata()
        t.data_path = tmp_path
        t.pad_files = True
        tmp_file = tempfile.mkstemp(".torrent")[1]
        t.save(tmp_file)

        check_torrent(tmp_file)

        os.remove(os.path.join(tmp_path, "file_A"))
        os.remove(os.path.join(tmp_path, "file_B"))
        os.remove(os.path.join(tmp_path, "file_C"))
        os.rmdir(tmp_path)
        os.remove(tmp_file)