sql-splitter 1.13.1

High-performance CLI tool for splitting large SQL dump files into individual table files
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

//! Unit tests for writer module, extracted from src/writer/mod.rs

use sql_splitter::writer::{TableWriter, WriterPool};
use tempfile::TempDir;

#[test]
fn test_table_writer() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.sql");

    let mut writer = TableWriter::new(&file_path).unwrap();
    writer
        .write_statement(b"CREATE TABLE t1 (id INT);")
        .unwrap();
    writer
        .write_statement(b"INSERT INTO t1 VALUES (1);")
        .unwrap();
    writer.flush().unwrap();

    let content = std::fs::read_to_string(&file_path).unwrap();
    assert!(content.contains("CREATE TABLE t1"));
    assert!(content.contains("INSERT INTO t1"));
}

#[test]
fn test_writer_pool() {
    let temp_dir = TempDir::new().unwrap();
    let mut pool = WriterPool::new(temp_dir.path().to_path_buf());
    pool.ensure_output_dir().unwrap();

    pool.write_statement("users", b"CREATE TABLE users (id INT);")
        .unwrap();
    pool.write_statement("posts", b"CREATE TABLE posts (id INT);")
        .unwrap();
    pool.write_statement("users", b"INSERT INTO users VALUES (1);")
        .unwrap();

    pool.close_all().unwrap();

    // Verify both table files were created
    let users_content = std::fs::read_to_string(temp_dir.path().join("users.sql")).unwrap();
    assert!(users_content.contains("CREATE TABLE users"));
    assert!(users_content.contains("INSERT INTO users"));

    let posts_content = std::fs::read_to_string(temp_dir.path().join("posts.sql")).unwrap();
    assert!(posts_content.contains("CREATE TABLE posts"));
}

#[test]
fn test_table_writer_flush_after_buffer_count() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.sql");
    let mut writer = TableWriter::new(&file_path).unwrap();
    for i in 0..150 {
        writer
            .write_statement(format!("INSERT INTO t VALUES ({});", i).as_bytes())
            .unwrap();
    }
    writer.flush().unwrap();
    let content = std::fs::read_to_string(&file_path).unwrap();
    let lines: Vec<&str> = content.lines().collect();
    assert_eq!(lines.len(), 150);
}

#[test]
fn test_table_writer_write_statement_with_suffix() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.sql");
    let mut writer = TableWriter::new(&file_path).unwrap();
    writer
        .write_statement_with_suffix(b"INSERT INTO t VALUES (1)", b";")
        .unwrap();
    writer.flush().unwrap();
    let content = std::fs::read_to_string(&file_path).unwrap();
    assert!(content.contains("INSERT INTO t VALUES (1);"));
}

#[test]
fn test_writer_pool_creates_output_dir() {
    let temp_dir = TempDir::new().unwrap();
    let output_dir = temp_dir.path().join("nested").join("output");
    let pool = WriterPool::new(output_dir.clone());
    pool.ensure_output_dir().unwrap();
    assert!(output_dir.exists());
}

#[test]
fn test_writer_pool_multiple_tables() {
    let temp_dir = TempDir::new().unwrap();
    let mut pool = WriterPool::new(temp_dir.path().to_path_buf());
    pool.ensure_output_dir().unwrap();

    for table in &["users", "posts", "comments", "tags", "categories"] {
        pool.write_statement(
            table,
            format!("CREATE TABLE {} (id INT);", table).as_bytes(),
        )
        .unwrap();
    }
    pool.close_all().unwrap();

    for table in &["users", "posts", "comments", "tags", "categories"] {
        let path = temp_dir.path().join(format!("{}.sql", table));
        assert!(path.exists(), "File for table {} should exist", table);
        let content = std::fs::read_to_string(&path).unwrap();
        assert!(content.contains(&format!("CREATE TABLE {}", table)));
    }
}

#[test]
fn test_writer_pool_write_statement_with_suffix() {
    let temp_dir = TempDir::new().unwrap();
    let mut pool = WriterPool::new(temp_dir.path().to_path_buf());
    pool.ensure_output_dir().unwrap();
    pool.write_statement_with_suffix("users", b"INSERT INTO users VALUES (1)", b";")
        .unwrap();
    pool.close_all().unwrap();
    let content = std::fs::read_to_string(temp_dir.path().join("users.sql")).unwrap();
    assert!(content.contains("INSERT INTO users VALUES (1);"));
}