polite 0.1.3

rusqlite × Polars bridge library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use polars::prelude::*;
use polite::prelude::*;
use tempfile::NamedTempFile;

#[test]
fn test_connect_error() {
    let err = connect_sqlite(Some("/nonexistent/dir/db.sqlite")).unwrap_err();
    assert!(matches!(err, PoliteError::Connect { .. }));
}

#[test]
fn test_query_error() {
    let db = NamedTempFile::new().unwrap();
    let db_path = db.path().to_str().unwrap();
    let err = to_dataframe(db_path, "SELECT * FROM nope").unwrap_err();
    assert!(matches!(err, PoliteError::Query { .. }));
}

#[test]
fn test_load_wrapper_error() {
    let db = NamedTempFile::new().unwrap();
    let db_path = db.path().to_str().unwrap();
    let err = load_dataframe(db_path, "SELECT * FROM nope").unwrap_err();
    eprintln!("Got error variant: {err:?}");
    assert!(matches!(err, PoliteError::Load { .. }));
}

#[test]
fn test_save_wrapper_error() {
    let df = df! { "id" => [1] }.unwrap();
    let err = save_dataframe("/dev/null", "friends", &df).unwrap_err();
    assert!(matches!(err, PoliteError::Save { .. }));
}

#[test]
fn test_load_dataframe_error_variant() {
    let db = NamedTempFile::new().unwrap();
    let db_path = db.path().to_str().unwrap();

    // table does not exist -> should trigger PoliteError::Load
    let err = load_dataframe(db_path, "SELECT * FROM imaginary").unwrap_err();
    eprintln!("Got error variant: {err:?}");
    assert!(matches!(err, PoliteError::Load { .. }));
}

#[test]
fn test_save_dataframe_error_variant() {
    let df = df! { "id" => [1] }.unwrap();

    // invalid path -> should trigger PoliteError::Connect or Save depending on failure mode
    let err = save_dataframe("/definitely/not/a/real/path/xxx", "friends", &df).unwrap_err();
    assert!(matches!(
        err,
        PoliteError::Connect { .. } | PoliteError::Save { .. }
    ));
}