dlin-core 0.2.2

Core library for dbt model lineage analysis
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use super::*;

#[test]
fn test_resolve_ref_model() {
    let mut node_map = HashMap::new();
    let graph = &mut LineageGraph::new();
    let idx = graph.add_node(NodeData {
        unique_id: "model.orders".to_string(),
        label: "orders".to_string(),
        node_type: NodeType::Model,
        file_path: None,
        description: None,
        materialization: None,
        tags: vec![],
        columns: vec![],
        exposure: None,
        aliases: vec![],
    });
    node_map.insert("model.orders".to_string(), idx);

    assert_eq!(resolve_ref("orders", &node_map), "model.orders");
}

#[test]
fn test_resolve_ref_seed() {
    let mut node_map = HashMap::new();
    let graph = &mut LineageGraph::new();
    let idx = graph.add_node(NodeData {
        unique_id: "seed.countries".to_string(),
        label: "countries".to_string(),
        node_type: NodeType::Seed,
        file_path: None,
        description: None,
        materialization: None,
        tags: vec![],
        columns: vec![],
        exposure: None,
        aliases: vec![],
    });
    node_map.insert("seed.countries".to_string(), idx);

    assert_eq!(resolve_ref("countries", &node_map), "seed.countries");
}

#[test]
fn test_resolve_ref_snapshot() {
    let mut node_map = HashMap::new();
    let graph = &mut LineageGraph::new();
    let idx = graph.add_node(NodeData {
        unique_id: "snapshot.snap_orders".to_string(),
        label: "snap_orders".to_string(),
        node_type: NodeType::Snapshot,
        file_path: None,
        description: None,
        materialization: None,
        tags: vec![],
        columns: vec![],
        exposure: None,
        aliases: vec![],
    });
    node_map.insert("snapshot.snap_orders".to_string(), idx);

    assert_eq!(
        resolve_ref("snap_orders", &node_map),
        "snapshot.snap_orders"
    );
}

#[test]
fn test_resolve_ref_unknown_defaults_to_model() {
    let node_map = HashMap::new();
    assert_eq!(resolve_ref("unknown_ref", &node_map), "model.unknown_ref");
}

#[test]
fn test_parse_exposure_ref() {
    assert_eq!(
        parse_exposure_ref("ref('orders')"),
        Some(("orders".to_string(), None))
    );
    assert_eq!(
        parse_exposure_ref("ref(\"orders\")"),
        Some(("orders".to_string(), None))
    );
    assert_eq!(
        parse_exposure_ref("ref('my_model', version=2)"),
        Some(("my_model".to_string(), Some("2".to_string())))
    );
    assert_eq!(
        parse_exposure_ref("ref('my_model', v=3)"),
        Some(("my_model".to_string(), Some("3".to_string())))
    );
    // Package-qualified refs are not supported and return None
    assert_eq!(parse_exposure_ref("ref('other_pkg', 'orders')"), None);
    assert_eq!(parse_exposure_ref("source('raw', 'orders')"), None);
    assert_eq!(parse_exposure_ref("something_else"), None);
}