lava_torrent 0.1.0

extern crate lava_torrent;
extern crate rand;

use std::path::PathBuf;
use rand::Rng;
use lava_torrent::ErrorKind;
use lava_torrent::bencode::BencodeElem;
use lava_torrent::torrent::v1::{Torrent, TorrentBuilder};

const OUTPUT_ROOT: &str = "tests/tmp/";
const PIECE_LENGTH: i64 = 16 * 1024; // 1024 KiB

fn rand_file_name() -> String {
    OUTPUT_ROOT.to_string() + &rand::thread_rng().gen::<u16>().to_string()
}

#[test]
fn build_single_file_ok() {
    let output_name = rand_file_name() + ".torrent";

    let builder = TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        PathBuf::from("tests/files/tails-amd64-3.6.1.torrent")
            .canonicalize()
            .unwrap(),
        PIECE_LENGTH,
    );
    // create a new chain because rustfmt does
    // weird things when `builder` is not declared separately
    builder
        .add_extra_field(
            "created by".to_string(),
            BencodeElem::String("qBittorrent v4.0.4".to_string()),
        )
        .add_extra_field(
            "creation date".to_string(),
            BencodeElem::Integer(1522837988),
        )
        .build()
        .unwrap()
        .write_into_file(&output_name)
        .unwrap();
}

#[test]
fn build_multi_file_ok() {
    let output_name = rand_file_name() + ".torrent";

    let builder = TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        PathBuf::from("tests/files").canonicalize().unwrap(),
        PIECE_LENGTH,
    );
    // create a new chain because rustfmt does
    // weird things when `builder` is not declared separately
    builder
        .add_extra_field(
            "created by".to_string(),
            BencodeElem::String("qBittorrent v4.0.4".to_string()),
        )
        .add_extra_field(
            "creation date".to_string(),
            BencodeElem::Integer(1522892260),
        )
        .build()
        .unwrap()
        .write_into_file(&output_name)
        .unwrap();
}

#[test]
fn build_with_name() {
    let output_name = rand_file_name() + ".torrent";

    TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        PathBuf::from("tests/files/tails-amd64-3.6.1.torrent")
            .canonicalize()
            .unwrap(),
        PIECE_LENGTH,
    ).set_name("file".to_string())
        .build()
        .unwrap()
        .write_into_file(&output_name)
        .unwrap();

    assert_eq!(
        Torrent::read_from_file(output_name).unwrap().name,
        "file".to_string(),
    );
}

#[test]
fn build_private() {
    let output_name = rand_file_name() + ".torrent";

    let builder = TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        PathBuf::from("tests/files").canonicalize().unwrap(),
        PIECE_LENGTH,
    );
    // create a new chain because rustfmt does
    // weird things when `builder` is not declared separately
    builder
        .add_extra_field(
            "created by".to_string(),
            BencodeElem::String("qBittorrent v4.0.4".to_string()),
        )
        .add_extra_field(
            "creation date".to_string(),
            BencodeElem::Integer(1522892270),
        )
        .set_privacy(true)
        .build()
        .unwrap()
        .write_into_file(&output_name)
        .unwrap();

    assert!(Torrent::read_from_file(output_name).unwrap().is_private());
}

#[test]
fn build_symbolic_link() {
    // `canonicalize()` follows symbolic links, so we have to do a
    // `push()` separately to avoid that resolution
    let mut path = PathBuf::from("tests/files").canonicalize().unwrap();
    path.push("symlink");

    let result = TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        path,
        PIECE_LENGTH,
    ).build();

    match result {
        Ok(_) => assert!(false),
        Err(e) => assert_eq!(e.kind(), ErrorKind::TorrentBuilderFailure),
    }
}

#[test]
fn build_hidden_file() {
    let result = TorrentBuilder::new(
        "udp://tracker.coppersurfer.tk:6969/announce".to_string(),
        PathBuf::from("tests/files/.hidden").canonicalize().unwrap(),
        PIECE_LENGTH,
    ).build();

    match result {
        Ok(_) => assert!(false),
        Err(e) => assert_eq!(e.kind(), ErrorKind::TorrentBuilderFailure),
    }
}