bytestashy 0.2.2

A command-line interface for ByteStash for the easy posting of snippets without opening a browser.
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
121
122
123
124
125
126
127

use assert_cmd::Command;
use predicates::prelude::*;
use std::fs;
use tempfile::TempDir;

#[test]
fn test_help_command() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg("--help")
        .assert()
        .success()
        .stdout(predicate::str::contains(
            "CLI to push snippets to ByteStash",
        ));
}

#[test]
fn test_version_command() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg("--version")
        .assert()
        .success()
        .stdout(predicate::str::contains("bytestashy"));
}

#[test]
fn test_list_help() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.args(&["list", "--help"])
        .assert()
        .success()
        .stdout(predicate::str::contains(
            "Show a paginated list of snippets",
        ));
}

#[test]
fn test_get_help() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.args(&["get", "--help"])
        .assert()
        .success()
        .stdout(predicate::str::contains("Retrieve a snippet by ID"));
}

#[test]
fn test_login_help() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.args(&["login", "--help"])
        .assert()
        .success()
        .stdout(predicate::str::contains("Fetches an API token"));
}

#[test]
fn test_no_files_provided() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.assert().failure().stderr(predicate::str::contains(
        "Provide at least one file to upload",
    ));
}

#[test]
fn test_nonexistent_file() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg("/nonexistent/file.txt")
        .assert()
        .failure()
        .stderr(predicate::str::contains("File does not exist"));
}

#[test]
fn test_path_traversal_protection() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg("../../../etc/passwd")
        .assert()
        .failure()
        .stderr(predicate::str::contains(".."));
}

#[test]
fn test_invalid_url_scheme() {
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.args(&["login", "ftp://example.com"])
        .assert()
        .failure()
        .stderr(predicate::str::contains(
            "URL must use http or https scheme",
        ));
}

#[test]
fn test_file_upload_validation() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.txt");
    fs::write(&file_path, "test content").unwrap();

    // This will fail because it tries to prompt user, but it should validate the file first
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg(file_path.to_str().unwrap())
        .assert()
        .failure()
        .stderr(
            predicate::str::contains("Dialog interaction failed")
                .or(predicate::str::contains("Failed to initialize API client")),
        );
}

#[test]
fn test_list_command_runs() {
    // This test just checks that the list command can be executed
    // It might succeed if there's a valid config, or fail if not logged in
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.arg("list")
        .assert()
        .code(predicate::in_iter(vec![0, 1])); // Either success or failure is acceptable
}

#[test]
fn test_get_invalid_snippet() {
    // This test checks that get command handles invalid snippet IDs properly
    let mut cmd = Command::cargo_bin("bytestashy").unwrap();
    cmd.args(&["get", "999999"]).assert().failure().stderr(
        predicate::str::contains("Snippet not found")
            .or(predicate::str::contains("Failed to initialize API client")),
    );
}