aboutsummaryrefslogtreecommitdiff
path: root/azalea-client/tests/simulation/close_open_container.rs
blob: 033451f1fc6e5a532b532626d1424d40f4a1d142 (plain)
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
use azalea_chat::FormattedText;
use azalea_client::test_utils::prelude::*;
use azalea_core::position::ChunkPos;
use azalea_entity::inventory::Inventory;
use azalea_protocol::packets::{
    ConnectionProtocol,
    game::{ClientboundContainerClose, ClientboundOpenScreen, ClientboundSetChunkCacheCenter},
};
use azalea_registry::builtin::MenuKind;

#[test]
fn test_close_open_container() {
    let _lock = init();

    let mut simulation = Simulation::new(ConnectionProtocol::Game);

    simulation.receive_packet(default_login_packet());
    simulation.tick();
    // receive a chunk so the player is "loaded" now
    simulation.receive_packet(ClientboundSetChunkCacheCenter { x: 1, z: 23 });
    simulation.receive_packet(make_basic_empty_chunk(
        ChunkPos::new(1, 23),
        (384 + 64) / 16,
    ));
    simulation.tick();

    // ensure no container is open
    simulation.with_component(|inventory: &Inventory| {
        assert!(inventory.container_menu.is_none());
        assert_eq!(inventory.id, 0);
    });

    // open a container
    simulation.receive_packet(ClientboundOpenScreen {
        container_id: 1,
        menu_type: MenuKind::Generic9x3,
        title: FormattedText::default(),
    });
    simulation.tick();

    simulation.with_component(|inventory: &Inventory| {
        assert!(inventory.container_menu.is_some());
        assert_eq!(inventory.id, 1);
    });

    // close and open
    simulation.receive_packet(ClientboundContainerClose { container_id: 1 });
    simulation.receive_packet(ClientboundOpenScreen {
        container_id: 2,
        menu_type: MenuKind::Generic9x3,
        title: FormattedText::default(),
    });
    simulation.tick();
    simulation.with_component(|inventory: &Inventory| {
        // ensure that the new container was opened
        assert!(inventory.container_menu.is_some());
        assert_eq!(inventory.id, 2);
    });

    // close with the wrong container id should still close
    simulation.receive_packet(ClientboundContainerClose { container_id: 123 });
    simulation.tick();
    simulation.with_component(|inventory: &Inventory| {
        assert!(inventory.container_menu.is_none());
        assert_eq!(inventory.id, 0);
    });
}