serverforge 0.1.3

ServerForge - A robust server setup and maintenance tool
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

use server_forge::rollback::RollbackManager;
use std::fs;

#[test]
fn test_create_snapshot() {
    let rollback_manager = RollbackManager::new();
    let snapshot_id = rollback_manager.create_snapshot().unwrap();
    assert!(snapshot_id > 0);
}

#[test]
fn test_add_file_change() {
    let rollback_manager = RollbackManager::new();
    let snapshot_id = rollback_manager.create_snapshot().unwrap();

    let test_file = "/tmp/test_rollback.txt";
    fs::write(test_file, "original content").unwrap();

    assert!(rollback_manager
        .add_file_change(snapshot_id, test_file)
        .is_ok());

    // Modify the file
    fs::write(test_file, "modified content").unwrap();

    // Rollback
    assert!(rollback_manager.rollback_to(snapshot_id).is_ok());

    // Verify the file content is back to original
    let content = fs::read_to_string(test_file).unwrap();
    assert_eq!(content, "original content");
}

#[test]
fn test_add_package_installed() {
    let rollback_manager = RollbackManager::new();
    let snapshot_id = rollback_manager.create_snapshot().unwrap();

    let test_package = "htop";

    // Install the package
    std::process::Command::new("apt-get")
        .args(&["install", "-y", test_package])
        .status()
        .unwrap();

    assert!(rollback_manager
        .add_package_installed(snapshot_id, test_package)
        .is_ok());

    // Uninstall the package
    std::process::Command::new("apt-get")
        .args(&["remove", "-y", test_package])
        .status()
        .unwrap();

    // Rollback
    assert!(rollback_manager.rollback_to(snapshot_id).is_ok());

    // Verify the package is installed
    let status = std::process::Command::new("which")
        .arg(test_package)
        .status()
        .unwrap();
    assert!(status.success());
}