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
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 crate::{
    core::models::db::{CustomEnum, CustomEnumVariant},
    mysql::{queries::get_enums::get_mysql_enums, test_helper::setup_mysql_db},
};
use pretty_assertions::assert_eq;
use std::error::Error;

#[tokio::test]
async fn test_get_mysql_enums() -> Result<(), Box<dyn Error>> {
    let (pool, _) = setup_mysql_db().await;

    sqlx::query("DROP TABLE IF EXISTS test_table;")
        .execute(&pool)
        .await?;

    sqlx::query(
        "
        CREATE TABLE test_table (
          id INT AUTO_INCREMENT PRIMARY KEY,
          mood ENUM('sad', 'ok', 'happy') NOT NULL
        );
    ",
    )
    .execute(&pool)
    .await?;
    let enums = get_mysql_enums(&pool).await?;

    let expected = vec![CustomEnum {
        name: "mood".to_string(),
        child_of_table: Some("test_table".to_string()),
        schema: None,
        variants: vec![
            CustomEnumVariant {
                name: "happy".to_string(),
            },
            CustomEnumVariant {
                name: "ok".to_string(),
            },
            CustomEnumVariant {
                name: "sad".to_string(),
            },
        ],
        comments: None,
        ..Default::default()
    }];

    assert_eq!(enums, expected);

    Ok(())
}

#[tokio::test]
async fn test_get_mysql_enums_with_comments() -> Result<(), Box<dyn Error>> {
    let (pool, _) = setup_mysql_db().await;
    // Clean up any existing type
    // Drop any existing table that uses the enum.
    sqlx::query("DROP TABLE IF EXISTS test_weather;")
        .execute(&pool)
        .await?;

    // Create a table with a 'weather' column defined as an ENUM with a comment.
    sqlx::query("
    CREATE TABLE test_weather (
      id INT AUTO_INCREMENT PRIMARY KEY,
      weather ENUM('rainy', 'cloudy', 'sunny') NOT NULL COMMENT 'This enum represents different weather'
    );
")
    .execute(&pool)
    .await?;

    // Retrieve the enum definitions including comments
    let enums = get_mysql_enums(&pool).await?;

    // Define the expected result including the comment on the type
    let expected = vec![CustomEnum {
        name: "weather".to_string(),
        child_of_table: Some("test_weather".to_string()),
        schema: None,
        variants: vec![
            CustomEnumVariant {
                name: "cloudy".to_string(),
            },
            CustomEnumVariant {
                name: "rainy".to_string(),
            },
            CustomEnumVariant {
                name: "sunny".to_string(),
            },
        ],
        comments: Some("This enum represents different weather".to_string()),
        ..Default::default()
    }];

    pretty_assertions::assert_eq!(enums, expected);

    Ok(())
}