howler-core 0.1.0

Core library for the Howler wolf-tracking platform: database, data ingestion, ML, web server, and real-time streaming
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

use crate::models::SpeciesStatus;
use crate::{Database, Sighting, Source};
use chrono::Utc;
use tempfile::NamedTempFile;

#[test]
fn test_database_creation() {
    let temp_file = NamedTempFile::new().unwrap();
    let db = Database::new(temp_file.path().to_str().unwrap()).unwrap();
    // Test that database was created successfully by inserting a sighting
    let sighting = Sighting {
        id: None,
        species: "Test".to_string(),
        scientific_name: None,
        latitude: 0.0,
        longitude: 0.0,
        observed_on: Utc::now(),
        source: Source::GBIF,
        source_id: "test".to_string(),
        details: None,
    };
    assert!(db.insert_sighting(&sighting).is_ok());
}

#[test]
fn test_insert_and_get_sighting() {
    let temp_file = NamedTempFile::new().unwrap();
    let db = Database::new(temp_file.path().to_str().unwrap()).unwrap();

    let sighting = Sighting {
        id: None,
        species: "Canis lupus".to_string(),
        scientific_name: Some("Canis lupus".to_string()),
        latitude: 45.5,
        longitude: -122.5,
        observed_on: Utc::now(),
        source: Source::GBIF,
        source_id: "test_123".to_string(),
        details: Some("Test sighting".to_string()),
    };

    let id = db.insert_sighting(&sighting).unwrap();
    assert!(id > 0);

    let sightings = db.get_all_sightings().unwrap();
    assert_eq!(sightings.len(), 1);
    assert_eq!(sightings[0].species, "Canis lupus");
    assert_eq!(sightings[0].source_id, "test_123");
}

#[test]
fn test_insert_duplicate_sighting() {
    let temp_file = NamedTempFile::new().unwrap();
    let db = Database::new(temp_file.path().to_str().unwrap()).unwrap();

    let sighting = Sighting {
        id: None,
        species: "Canis lupus".to_string(),
        scientific_name: Some("Canis lupus".to_string()),
        latitude: 45.5,
        longitude: -122.5,
        observed_on: Utc::now(),
        source: Source::GBIF,
        source_id: "test_123".to_string(),
        details: Some("Test sighting".to_string()),
    };

    db.insert_sighting(&sighting).unwrap();
    db.insert_sighting(&sighting).unwrap(); // Should replace due to UNIQUE constraint

    let sightings = db.get_all_sightings().unwrap();
    assert_eq!(sightings.len(), 1);
}

#[test]
fn test_insert_and_get_species_status() {
    let temp_file = NamedTempFile::new().unwrap();
    let db = Database::new(temp_file.path().to_str().unwrap()).unwrap();

    let status = SpeciesStatus {
        id: None,
        scientific_name: "Canis lupus".to_string(),
        common_name: Some("Gray Wolf".to_string()),
        red_list_category: Some("LC".to_string()),
        population_trend: Some("Stable".to_string()),
        threats: Some("Habitat loss".to_string()),
    };

    let id = db.insert_species_status(&status).unwrap();
    assert!(id > 0);

    let retrieved = db.get_species_status("Canis lupus").unwrap();
    assert!(retrieved.is_some());
    let retrieved = retrieved.unwrap();
    assert_eq!(retrieved.scientific_name, "Canis lupus");
    assert_eq!(retrieved.common_name, Some("Gray Wolf".to_string()));
}

#[test]
fn test_get_nonexistent_species_status() {
    let temp_file = NamedTempFile::new().unwrap();
    let db = Database::new(temp_file.path().to_str().unwrap()).unwrap();

    let retrieved = db.get_species_status("Nonexistent species").unwrap();
    assert!(retrieved.is_none());
}