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
|
#include "avatar.h"
#include "cata_catch.h"
#include "creature_tracker.h"
#include "game.h"
#include "map.h"
#include "map_helpers.h"
#include "mapgen_helpers.h"
#include "npc.h"
#include "overmapbuffer.h"
#include "player_helpers.h"
static const string_id<npc_template> npc_template_test_npc_trader( "test_npc_trader" );
static const string_id<npc_template> npc_template_thug( "thug" );
static const update_mapgen_id update_mapgen_test_update_place_npc( "test_update_place_npc" );
static const update_mapgen_id
update_mapgen_test_update_place_npc_thug( "test_update_place_npc_thug" );
static const update_mapgen_id update_mapgen_test_update_remove_npc( "test_update_remove_npc" );
namespace
{
void check_creature( tripoint const &loc, string_id<npc_template> const &id, bool state )
{
Creature *critter = get_creature_tracker().creature_at( loc, true );
if( state ) {
REQUIRE( critter != nullptr );
REQUIRE( critter->as_npc()->idz == id );
} else {
REQUIRE( critter == nullptr );
}
}
void place_npc_and_check( map &m, tripoint const &loc, update_mapgen_id const &id,
string_id<npc_template> const &nid )
{
check_creature( loc, nid, false );
tripoint_abs_omt const omt = project_to<coords::omt>( m.getglobal( loc ) );
manual_mapgen( omt, manual_update_mapgen, id );
check_creature( loc, nid, true );
g->mon_info_update();
}
void remove_npc_and_check( map &m, tripoint const &loc, update_mapgen_id const &id,
string_id<npc_template> const &nid )
{
check_creature( loc, nid, true );
tripoint_abs_omt const omt = project_to<coords::omt>( m.getglobal( loc ) );
manual_mapgen( omt, manual_update_mapgen, id );
check_creature( loc, nid, false );
for( auto const &t : get_avatar().get_mon_visible().unique_types ) {
for( npc *n : t ) {
REQUIRE( !n->is_dead() ); // REQUIRE( dont-segfault-bro )
}
}
}
} // namespace
TEST_CASE( "mapgen_remove_npcs" )
{
GIVEN( "in bounds of main map" ) {
map &here = get_map();
clear_map();
clear_avatar();
tripoint const start_loc( HALF_MAPSIZE_X + SEEX - 2, HALF_MAPSIZE_Y + SEEY - 1, 0 );
get_avatar().setpos( start_loc );
clear_npcs();
set_time( calendar::turn_zero + 12_hours );
tripoint_abs_omt const omt = project_to<coords::omt>( get_avatar().get_location() );
tripoint_abs_omt const omt2 = omt + tripoint_east;
tripoint const loc = here.getlocal( project_to<coords::ms>( omt ) );
tripoint const loc2 = here.getlocal( project_to<coords::ms>( omt2 ) );
tripoint const loc3 = loc2 + tripoint_east;
REQUIRE( get_map().inbounds( loc ) );
place_npc_and_check( here, loc, update_mapgen_test_update_place_npc, npc_template_test_npc_trader );
REQUIRE( overmap_buffer.get_npcs_near_omt( omt, 0 ).size() == 1 );
place_npc_and_check( here, loc2, update_mapgen_test_update_place_npc,
npc_template_test_npc_trader );
REQUIRE( overmap_buffer.get_npcs_near_omt( omt2, 0 ).size() == 1 );
REQUIRE( get_avatar().sees( loc, true ) );
WHEN( "removing NPC" ) {
remove_npc_and_check( here, loc, update_mapgen_test_update_remove_npc,
npc_template_test_npc_trader );
THEN( "NPC of same class on different submap not affected" ) {
REQUIRE( overmap_buffer.get_npcs_near_omt( omt, 0 ).empty() );
REQUIRE( overmap_buffer.get_npcs_near_omt( omt2, 0 ).size() == 1 );
}
THEN( "NPC of different class on same submap not affected" ) {
place_npc_and_check( here, loc3, update_mapgen_test_update_place_npc_thug, npc_template_thug );
REQUIRE( overmap_buffer.get_npcs_near_omt( omt2, 0 ).size() == 2 );
remove_npc_and_check( here, loc2, update_mapgen_test_update_remove_npc,
npc_template_test_npc_trader );
REQUIRE( overmap_buffer.get_npcs_near_omt( omt2, 0 ).size() == 1 );
check_creature( loc3, npc_template_thug, true );
}
}
}
}
|
