sql-gen 0.2.3

A CLI tool for generating models based on a SQL Database using SQLx
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

use crate::core::{
    intepreters::get_file_structs::get_file_structs,
    models::rust::{dbset_attribute_with_table_name, RustDbSetStruct},
};

#[test]
fn should_parse_simple_struct() {
    let res = get_file_structs(
        r#"
    pub struct Customer {
        id: String,
        first_name: String,
        last_name: String
    }
    "#,
    );

    let expected = vec![RustDbSetStruct {
        name: "Customer".to_string(),
        ..Default::default()
    }];

    pretty_assertions::assert_eq!(res, expected);
}

#[test]
fn should_parse_struct_with_db_set_macro() {
    let res = get_file_structs(
        r#"
        #[dbset(table_name = "users")]
        pub struct Customer {
        id: String,
        first_name: String,
        last_name: String
    }
    "#,
    );

    let expected = vec![RustDbSetStruct {
        name: "Customer".to_string(),
        attributes: vec![dbset_attribute_with_table_name("users")],
        ..Default::default()
    }];

    pretty_assertions::assert_eq!(res, expected);
}