faction_camp_test.cpp

#include <cstddef>
#include <iosfwd>
#include <utility>
#include <vector>

#include "basecamp.h"
#include "cata_catch.h"
#include "clzones.h"
#include "coordinates.h"
#include "faction.h"
#include "map.h"
#include "map_helpers.h"
#include "overmap.h"
#include "overmapbuffer.h"
#include "player_helpers.h"

static const vitamin_id vitamin_mutagen( "mutagen" );
static const vitamin_id vitamin_mutant_toxin( "mutant_toxin" );

static const zone_type_id zone_type_CAMP_FOOD( "CAMP_FOOD" );
static const zone_type_id zone_type_CAMP_STORAGE( "CAMP_STORAGE" );

TEST_CASE( "camp_calorie_counting", "[camp]" )
{
    clear_avatar();
    clear_map();
    map &m = get_map();
    const tripoint_abs_ms zone_loc = m.getglobal( tripoint{ 5, 5, 0 } );
    mapgen_place_zone( zone_loc.raw(), zone_loc.raw(), zone_type_CAMP_FOOD, your_fac, {},
                       "food" );
    mapgen_place_zone( zone_loc.raw(), zone_loc.raw(), zone_type_CAMP_STORAGE, your_fac, {},
                       "storage" );
    faction *camp_faction = get_player_character().get_faction();
    const tripoint_abs_omt this_omt = project_to<coords::omt>( zone_loc );
    m.add_camp( this_omt, "faction_camp" );
    std::optional<basecamp *> bcp = overmap_buffer.find_camp( this_omt.xy() );
    basecamp *test_camp = *bcp;
    test_camp->set_owner( your_fac );
    nutrients &food_supply = camp_faction->food_supply;
    WHEN( "a base item is added to larder" ) {
        food_supply *= 0;
        item test_100_kcal( "test_100_kcal" );
        tripoint_bub_ms zone_local = m.bub_from_abs( zone_loc );
        m.i_clear( zone_local.raw() );
        m.add_item_or_charges( zone_local, test_100_kcal );
        REQUIRE( m.has_items( zone_local ) );
        test_camp->distribute_food();
        CHECK( food_supply.kcal() == 100 );
    }

    WHEN( "an item with inherited components is added to larder" ) {
        food_supply *= 0;
        item test_100_kcal( "test_100_kcal" );
        item test_200_kcal( "test_200_kcal" );
        item_components made_of;
        made_of.add( test_100_kcal );
        made_of.add( test_100_kcal );
        // Setting the actual components. This will return 185 unless it's actually made up of two 100kcal components!
        test_200_kcal.components = made_of;
        tripoint_bub_ms zone_local = m.bub_from_abs( zone_loc );
        m.i_clear( zone_local.raw() );
        m.add_item_or_charges( zone_local, test_200_kcal );
        test_camp->distribute_food();
        CHECK( food_supply.kcal() == 200 );
    }

    WHEN( "an item with vitamins is added to larder" ) {
        food_supply *= 0;
        item test_500_kcal( "test_500_kcal" );
        tripoint_bub_ms zone_local = m.bub_from_abs( zone_loc );
        m.i_clear( zone_local.raw() );
        m.add_item_or_charges( zone_local, test_500_kcal );
        test_camp->distribute_food();
        REQUIRE( food_supply.kcal() == 500 );
        REQUIRE( food_supply.get_vitamin( vitamin_mutant_toxin ) == 100 );
        REQUIRE( food_supply.get_vitamin( vitamin_mutagen ) == 200 );
    }

    WHEN( "a larder with stored calories and vitamins has food withdrawn" ) {
        food_supply *= 0;
        food_supply.calories += 100000;
        food_supply.set_vitamin( vitamin_mutant_toxin, 100 );
        food_supply.set_vitamin( vitamin_mutagen, 200 );
        CHECK( food_supply.kcal() == 100 );
        CHECK( food_supply.get_vitamin( vitamin_mutant_toxin ) == 100 );
        CHECK( food_supply.get_vitamin( vitamin_mutagen ) == 200 );
        // Now withdraw 15% of the total calories, this should also draw out 15% of the stored vitamins.
        test_camp->camp_food_supply( -15 );
        REQUIRE( food_supply.kcal() == 85 );
        REQUIRE( food_supply.get_vitamin( vitamin_mutant_toxin ) == 85 );
        REQUIRE( food_supply.get_vitamin( vitamin_mutagen ) == 170 );
    }
}
// TODO: Tests for: Check calorie display at various activity levels, camp crafting works as expected (consumes inputs, returns outputs+byproducts, costs calories)