rbook 0.7.6

A fast, format-agnostic, ergonomic ebook library for reading, building, and editing EPUB 2 and 3.
Documentation
use crate::epub::util::TestEpub::Epub3File;
use rbook::epub::reader::{EpubReaderContent, LinearBehavior};
use rbook::reader::errors::ReaderResult;
use wasm_bindgen_test::wasm_bindgen_test;

#[test]
#[wasm_bindgen_test]
fn test_reader_linear_behavior() {
    let epub = Epub3File.open_strict();
    #[rustfmt::skip]
    let linear_behaviors = [
        (
            LinearBehavior::Original,                // Spine entry positioning
            vec!["cover", "toc", "c1", "c1a", "c2"], // Expected spine entry order
        ),
        (LinearBehavior::LinearOnly, vec!["toc", "c1", "c2"]),
        (LinearBehavior::NonLinearOnly, vec!["cover", "c1a"]),
        (LinearBehavior::PrependNonLinear, vec!["cover", "c1a", "toc", "c1", "c2"]),
        (LinearBehavior::AppendNonLinear, vec!["toc", "c1", "c2", "cover", "c1a"]),
    ];

    for (linear_behavior, cases) in linear_behaviors {
        let mut reader = epub
            .reader_builder()
            .linear_behavior(linear_behavior)
            .create();

        assert_eq!(cases.len(), reader.len());

        for case in cases {
            assert_eq!(case, reader.next().unwrap().unwrap().spine_entry().idref());
        }
        assert!(reader.next().is_none());

        reader.next();
    }
}

#[test]
#[wasm_bindgen_test]
fn test_reader_cursor() -> ReaderResult<()> {
    let epub = Epub3File.open_strict();
    let mut reader = epub.reader();

    fn idref(content: EpubReaderContent<'_>) -> &str {
        content.spine_entry().idref()
    }

    assert!(!reader.is_empty());
    assert_eq!(5, reader.len());
    assert_eq!(reader.len(), reader.remaining());
    assert_eq!(None, reader.current_position());

    // Jump
    assert_eq!("toc", idref(reader.read(1)?));
    assert_eq!(Some(1), reader.current_position());
    assert_eq!(3, reader.remaining());

    // Move Backward
    assert_eq!("cover", idref(reader.read_prev().unwrap()?));
    assert_eq!(Some(0), reader.current_position());
    assert_eq!(4, reader.remaining());

    // Move backward
    assert!(reader.read_prev().is_none());
    assert!(reader.read_prev().is_none());

    // Move forward
    assert_eq!("toc", idref(reader.next().unwrap()?));
    assert_eq!(Some(1), reader.current_position());
    assert_eq!(3, reader.remaining());

    // Jump
    assert_eq!("c2", idref(reader.read("c2")?));
    assert_eq!(Some(4), reader.current_position());
    assert_eq!(0, reader.remaining());
    assert!(reader.next().is_none());
    assert!(reader.next().is_none());
    assert_eq!(Some(4), reader.current_position());

    // Move backward
    assert_eq!("c1a", idref(reader.read_prev().unwrap()?));
    assert_eq!(Some(3), reader.current_position());
    assert_eq!(1, reader.remaining());

    // Jump
    assert_eq!("c1", idref(reader.read(2)?));
    assert_eq!(Some(2), reader.current_position());
    assert_eq!(2, reader.remaining());

    assert_eq!(5, reader.len());

    Ok(())
}

#[test]
#[wasm_bindgen_test]
fn test_reader_into_content_integrity() -> ReaderResult<()> {
    let epub = Epub3File.open_strict();

    // Ensure integrity
    for content_result in epub.reader() {
        let entry_a = content_result?;
        let entry_b = entry_a.clone();

        let href = entry_a.manifest_entry().href();
        let resource = epub.read_resource_str(href)?;

        assert_eq!(resource, entry_a.into_string());
        assert_eq!(resource.as_bytes(), entry_b.into_bytes());
    }

    Ok(())
}