about summary refs log tree commit diff
path: root/users/tazjin/finito/finito-postgres/src/tests.rs
blob: dd270c38759fab3c6cbbbb2b888534fe4eda1304 (plain) (blame)
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
use super::*;

use finito_door::*;
use postgres::{Connection, TlsMode};

// TODO: read config from environment
fn open_test_connection() -> Connection {
    Connection::connect("postgres://finito:finito@localhost/finito", TlsMode::None)
        .expect("Failed to connect to test database")
}

#[test]
fn test_insert_machine() {
    let conn = open_test_connection();
    let initial = DoorState::Opened;
    let door = insert_machine(&conn, initial).expect("Failed to insert door");
    let result = get_machine(&conn, &door, false).expect("Failed to fetch door");

    assert_eq!(
        result,
        DoorState::Opened,
        "Inserted door state should match"
    );
}

#[test]
fn test_advance() {
    let conn = open_test_connection();

    let initial = DoorState::Opened;
    let events = vec![
        DoorEvent::Close,
        DoorEvent::Open,
        DoorEvent::Close,
        DoorEvent::Lock(1234),
        DoorEvent::Unlock(1234),
        DoorEvent::Lock(4567),
        DoorEvent::Unlock(1234),
    ];

    let door = insert_machine(&conn, initial).expect("Failed to insert door");

    for event in events {
        advance(&conn, &door, event).expect("Failed to advance door FSM");
    }

    let result = get_machine(&conn, &door, false).expect("Failed to fetch door");
    let expected = DoorState::Locked {
        code: 4567,
        attempts: 2,
    };

    assert_eq!(result, expected, "Advanced door state should match");
}