random-image-server 0.2.1

A simple image server that serves random images from a preconfigured list of paths and URLs.
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

use std::fs;

use pretty_assertions::assert_eq;
use random_image_server::{
    ImageServer,
    config::{Config, ImageSource},
};
use tempfile::TempDir;

#[tokio::test]
async fn test_image_server_populate_cache_no_sources() {
    let config = Config::default();
    let server = ImageServer::with_config(config);

    // Should not panic even with no sources
    server.populate_cache().await;

    assert_eq!(server.state.read().await.cache.size(), 0);
}

#[tokio::test]
async fn test_image_server_populate_cache_with_file() {
    let temp_dir = TempDir::new().unwrap();
    let image_path = temp_dir.path().join("test.jpg");
    let test_data = vec![0xFF, 0xD8, 0xFF]; // JPEG header
    fs::write(&image_path, &test_data).unwrap();

    let mut config = Config::default();
    config.server.sources = vec![ImageSource::Path(image_path)];

    let server = ImageServer::with_config(config);
    server.populate_cache().await;

    assert_eq!(server.state.read().await.cache.size(), 1);
}

#[tokio::test]
async fn test_image_server_populate_cache_with_directory() {
    let temp_dir = TempDir::new().unwrap();
    let image1_path = temp_dir.path().join("test1.jpg");
    let image2_path = temp_dir.path().join("test2.png");
    let text_file_path = temp_dir.path().join("readme.txt");

    fs::write(&image1_path, vec![0xFF, 0xD8, 0xFF]).unwrap();
    fs::write(&image2_path, vec![0x89, 0x50, 0x4E, 0x47]).unwrap();
    fs::write(&text_file_path, "not an image").unwrap();

    let mut config = Config::default();
    config.server.sources = vec![ImageSource::Path(temp_dir.path().to_path_buf())];

    let server = ImageServer::with_config(config);
    server.populate_cache().await;

    // Should only load image files, not text files
    assert_eq!(server.state.read().await.cache.size(), 2);
}

#[tokio::test]
async fn test_image_server_populate_cache_invalid_file() {
    let temp_dir = TempDir::new().unwrap();
    let text_path = temp_dir.path().join("test.txt");
    fs::write(&text_path, "not an image").unwrap();

    let mut config = Config::default();
    config.server.sources = vec![ImageSource::Path(text_path)];

    let server = ImageServer::with_config(config);
    server.populate_cache().await;

    // Should not load non-image files
    assert_eq!(server.state.read().await.cache.size(), 0);
}