diesel_cli 2.3.7

#![allow(clippy::expect_fun_call)]
use chrono::Utc;
use std::thread::sleep;
use std::time::Duration;

use crate::support::{database, project};
pub static TIMESTAMP_FORMAT: &str = "%Y-%m-%d-%H%M%S";

#[test]
fn migration_list_lists_pending_applied_migrations() {
    let p = project("migration_list_pending_applied")
        .folder("migrations")
        .build();
    let db = database(&p.database_url());

    p.command("setup").run();

    p.create_migration(
        "12345_create_users_table",
        "CREATE TABLE users (id INTEGER PRIMARY KEY)",
        Some("DROP TABLE users"),
        None,
    );

    assert!(!db.table_exists("users"));

    // finds unapplied migration
    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    assert!(result.stdout().contains("[ ] 12345_create_users_table"));

    let result = p.command("migration").arg("run").run();
    assert!(result.is_success());
    assert!(db.table_exists("users"));

    // finds applied migration
    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    assert!(result.stdout().contains("[X] 12345_create_users_table"));
}

fn assert_tags_in_order(output: &str, tags: &[&str]) {
    let matches: Vec<_> = tags
        .iter()
        .map(|s| {
            let index = output
                .find(s)
                .expect(&format!("tag {:?} not found in output: {:?}", s, output));
            (index, s)
        })
        .collect();
    for window in matches.as_slice().windows(2) {
        assert!(
            window[0].0 < window[1].0,
            "expected {:?} before {:?}",
            window[0].1,
            window[1].1
        );
    }
}

#[test]
fn migration_list_lists_migrations_ordered_by_timestamp() {
    let p = project("migration_list_ordered")
        .folder("migrations")
        .build();

    p.command("setup").run();

    let tag1 = format!("{}_initial", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag1, "", Some(""), None);

    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    assert!(result.stdout().contains(&format!("[ ] {}", &tag1)));

    sleep(Duration::from_millis(1100));

    let tag2 = format!("{}_alter", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag2, "", Some(""), None);

    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    let output = result.stdout();
    assert_tags_in_order(output, &[&tag1, &tag2]);
}

#[test]
fn migration_list_orders_unknown_timestamps_last() {
    let p = project("migration_list_custom")
        .folder("migrations")
        .build();

    p.command("setup").run();

    let tag1 = format!("{}_migration1", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag1, "", Some(""), None);

    let tag4 = "abc_migration4";
    p.create_migration(tag4, "", Some(""), None);

    let tag5 = "zzz_migration5";
    p.create_migration(tag5, "", Some(""), None);

    sleep(Duration::from_millis(1100));

    let tag2 = format!("{}_migration2", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag2, "", Some(""), None);

    sleep(Duration::from_millis(1100));

    let tag3 = format!("{}_migration3", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag3, "", Some(""), None);

    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    let output = result.stdout();
    assert_tags_in_order(output, &[&tag1, &tag2, &tag3, tag4, tag5]);
}

#[test]
fn migration_list_orders_nontimestamp_versions_alphabetically() {
    let p = project("migration_list_nontimestamp_versions")
        .folder("migrations")
        .build();

    p.command("setup").run();

    let tag4 = "a_migration";
    p.create_migration(tag4, "", Some(""), None);

    let tag6 = "bc_migration";
    p.create_migration(tag6, "", Some(""), None);

    let tag5 = "aa_migration";
    p.create_migration(tag5, "", Some(""), None);

    let tag1 = "!wow_migration";
    p.create_migration(tag1, "", Some(""), None);

    let tag3 = "7letters";
    p.create_migration(tag3, "", Some(""), None);

    let tag2 = format!("{}_stamped_migration", Utc::now().format(TIMESTAMP_FORMAT));
    p.create_migration(&tag2, "", Some(""), None);

    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    let output = result.stdout();
    assert_tags_in_order(output, &[tag1, &tag2, tag3, tag4, tag5, tag6]);
}

#[test]
fn migration_list_orders_old_and_new_timestamp_forms_mixed_correctly() {
    let p = project("migration_list_mixed_timestamps")
        .folder("migrations")
        .build();

    p.command("setup").run();

    let tag1 = "20170505070309_migration";
    p.create_migration(tag1, "", Some(""), None);

    let tag2 = "2017-11-23-064836_migration";
    p.create_migration(tag2, "", Some(""), None);

    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    let output = result.stdout();
    assert_tags_in_order(output, &[tag1, tag2]);
}

#[test]
fn migration_list_respects_migrations_dir_from_diesel_toml() {
    let p = project("migration_list_respects_migrations_dir_from_diesel_toml")
        .folder("custom_migrations")
        .file(
            "diesel.toml",
            r#"
            [migrations_directory]
            dir = "custom_migrations"
            "#,
        )
        .build();
    let db = database(&p.database_url());

    p.command("setup").run();

    p.create_migration_in_directory(
        "custom_migrations",
        "12345_create_users_table",
        "CREATE TABLE users (id INTEGER PRIMARY KEY)",
        Some("DROP TABLE users"),
        None,
    );

    assert!(!db.table_exists("users"));

    // finds unapplied migration
    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    assert!(result.stdout().contains("[ ] 12345_create_users_table"));

    let result = p.command("migration").arg("run").run();
    assert!(result.is_success());
    assert!(db.table_exists("users"));

    // finds applied migration
    let result = p.command("migration").arg("list").run();
    assert!(result.is_success(), "Result was unsuccessful {:?}", result);
    assert!(result.stdout().contains("[X] 12345_create_users_table"));
}