Skip to content

Commit

Permalink
added test for colony upgrades / builds
Browse files Browse the repository at this point in the history
  • Loading branch information
@ametel01
ametel01 committed Feb 7, 2024
1 parent ab0a30d commit 39c6619
Showing 1 changed file with 208 additions and 11 deletions.
219 changes: 208 additions & 11 deletions src/colony/actions.cairo
View file
Original file line number Diff line number Diff line change
Expand Up @@ -52,8 +52,9 @@ mod colonyactions {
let current_count = get!(world, constants::GAME_ID, ColonyCount).count;
assert!(
current_planet_colonies < max_colonies.into(),
"Colony: max colonies {} reached, upgrade Exocraft tech to increase max colonies",
max_colonies
"Colony: max colonies {} for planet {} reached, upgrade Exocraft tech to increase max colonies",
max_colonies,
planet_id
);
let price: u256 = 0;
if !price.is_zero() {
Expand All @@ -72,13 +73,6 @@ mod colonyactions {
);
set!(world, PlanetColoniesCount { planet_id, count: colony_id });
set!(world, ColonyCount { game_id: constants::GAME_ID, count: current_count + 1 });
let current_number_of_planets = get!(world, constants::GAME_ID, GamePlanetCount).count;
set!(
world,
GamePlanetCount {
game_id: constants::GAME_ID, count: current_number_of_planets + 1
}
);
}

fn process_colony_compound_upgrade(
Expand Down Expand Up @@ -636,10 +630,10 @@ mod test {
use nogame::colony::actions::{IColonyActionsDispatcher, IColonyActionsDispatcherTrait};
use nogame::colony::models::{
ColonyOwner, ColonyPosition, ColonyCount, ColonyResourceTimer, PlanetColoniesCount,
ColonyResource, ColonyShips, ColonyDefences,
ColonyResource, ColonyShips, ColonyDefences, ColonyCompounds
};
use nogame::libraries::{constants};
use nogame::data::types::{Position, ShipBuildType};
use nogame::data::types::{Position, ShipBuildType, CompoundUpgradeType, DefenceBuildType};
use nogame::libraries::names::Names;
use nogame::compound::models::{PlanetCompounds};
use nogame::game::models::{GameSetup, GamePlanetCount};
Expand Down Expand Up @@ -698,4 +692,207 @@ mod test {
"Colony: position not set correctly"
);
}

#[test]
fn test_generate_colony_different_planets() {
let (world, actions, nft, eth) = setup_world();
actions.game.spawn(OWNER(), nft, eth, constants::MIN_PRICE_UNSCALED, GAME_SPEED,);

set_contract_address(ACCOUNT_1());
actions.planet.generate_planet();
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::EXOCRAFT, level: 1 });
actions.colony.generate_colony();

let colony_owner = get!(world, 1001, ColonyOwner).planet_id;
assert!(colony_owner == 1, "Colony: owner not set correctly");
let total_colonies_count = get!(world, constants::GAME_ID, ColonyCount).count;
assert!(total_colonies_count == 1, "Colony: count not set correctly");
let planet_colonies_count = get!(world, 1, PlanetColoniesCount).count;
assert!(planet_colonies_count == 1, "Colony: count not set correctly");
let colony_position = get!(world, (1, 1), ColonyPosition).position;
assert!(
colony_position.system == 188 && colony_position.orbit == 10,
"Colony: position not set correctly"
);

set_contract_address(ACCOUNT_2());
actions.planet.generate_planet();
set!(world, PlanetTechs { planet_id: 2, name: Names::Tech::EXOCRAFT, level: 1 });
actions.colony.generate_colony();

let colony_owner = get!(world, 2001, ColonyOwner).planet_id;
assert!(colony_owner == 2, "Colony: owner not set correctly");
let total_colonies_count = get!(world, constants::GAME_ID, ColonyCount).count;
assert!(total_colonies_count == 2, "Colony: count not set correctly");
let planet_colonies_count = get!(world, 2, PlanetColoniesCount).count;
assert!(planet_colonies_count == 1, "Colony: count not set correctly");
let colony_position = get!(world, (2, 1), ColonyPosition).position;
assert!(
colony_position.system == 182 && colony_position.orbit == 2,
"Colony: position not set correctly"
);
}

#[test]
fn test_upgrade_colony_compound() {
let (world, actions, nft, eth) = setup_world();
actions.game.spawn(OWNER(), nft, eth, constants::MIN_PRICE_UNSCALED, GAME_SPEED,);

set_contract_address(ACCOUNT_1());
actions.planet.generate_planet();
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::EXOCRAFT, level: 1 });
actions.colony.generate_colony();
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::STEEL, amount: 10000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::QUARTZ, amount: 10000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::TRITIUM, amount: 10000
}
);

actions.colony.process_colony_compound_upgrade(1, CompoundUpgradeType::SteelMine, 1);
actions.colony.process_colony_compound_upgrade(1, CompoundUpgradeType::QuartzMine, 1);
actions.colony.process_colony_compound_upgrade(1, CompoundUpgradeType::TritiumMine, 1);
actions.colony.process_colony_compound_upgrade(1, CompoundUpgradeType::EnergyPlant, 1);
actions.colony.process_colony_compound_upgrade(1, CompoundUpgradeType::Dockyard, 1);

let steel_level = get!(world, (1, 1, Names::Compound::STEEL), ColonyCompounds).level;
assert!(steel_level == 1, "Colony: steel mine not upgraded correctly");

let quartz_level = get!(world, (1, 1, Names::Compound::QUARTZ), ColonyCompounds).level;
assert!(quartz_level == 1, "Colony: quartz mine not upgraded correctly");

let tritium_level = get!(world, (1, 1, Names::Compound::TRITIUM), ColonyCompounds).level;
assert!(tritium_level == 1, "Colony: tritium mine not upgraded correctly");

let energy_level = get!(world, (1, 1, Names::Compound::ENERGY), ColonyCompounds).level;
assert!(energy_level == 1, "Colony: energy plant not upgraded correctly");

let dockyard_level = get!(world, (1, 1, Names::Compound::DOCKYARD), ColonyCompounds).level;
assert!(dockyard_level == 1, "Colony: dockyard not upgraded correctly");
}

#[test]
fn test_build_colony_ships() {
let (world, actions, nft, eth) = setup_world();
actions.game.spawn(OWNER(), nft, eth, constants::MIN_PRICE_UNSCALED, GAME_SPEED,);

set_contract_address(ACCOUNT_1());
actions.planet.generate_planet();
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::EXOCRAFT, level: 1 });
actions.colony.generate_colony();
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::STEEL, amount: 100000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::QUARTZ, amount: 100000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::TRITIUM, amount: 100000
}
);
set!(
world,
ColonyCompounds {
planet_id: 1, colony_id: 1, name: Names::Compound::DOCKYARD, level: 7
}
);
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::COMBUSTION, level: 6 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::SHIELD, level: 2 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::ION, level: 2 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::THRUST, level: 4 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::WARP, level: 4 });
actions.colony.process_ship_build(1, ShipBuildType::Carrier, 1);
actions.colony.process_ship_build(1, ShipBuildType::Scraper, 1);
actions.colony.process_ship_build(1, ShipBuildType::Sparrow, 1);
actions.colony.process_ship_build(1, ShipBuildType::Frigate, 1);
actions.colony.process_ship_build(1, ShipBuildType::Armade, 1);

let carrier_count = get!(world, (1, 1, Names::Fleet::CARRIER), ColonyShips).count;
assert!(carrier_count == 1, "Colony: carrier not built correctly");
let scraper_count = get!(world, (1, 1, Names::Fleet::SCRAPER), ColonyShips).count;
assert!(scraper_count == 1, "Colony: scraper not built correctly");
let sparrow_count = get!(world, (1, 1, Names::Fleet::SPARROW), ColonyShips).count;
assert!(sparrow_count == 1, "Colony: sparrow not built correctly");
let frigate_count = get!(world, (1, 1, Names::Fleet::FRIGATE), ColonyShips).count;
assert!(frigate_count == 1, "Colony: frigate not built correctly");
let armade_count = get!(world, (1, 1, Names::Fleet::ARMADE), ColonyShips).count;
assert!(armade_count == 1, "Colony: armade not built correctly");
}

#[test]
fn test_build_colony_defence() {
let (world, actions, nft, eth) = setup_world();
actions.game.spawn(OWNER(), nft, eth, constants::MIN_PRICE_UNSCALED, GAME_SPEED,);

set_contract_address(ACCOUNT_1());
actions.planet.generate_planet();
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::EXOCRAFT, level: 1 });
actions.colony.generate_colony();
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::STEEL, amount: 100_000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::QUARTZ, amount: 100_000
}
);
set!(
world,
ColonyResource {
planet_id: 1, colony_id: 1, name: Names::Resource::TRITIUM, amount: 100_000
}
);
set!(
world,
ColonyCompounds {
planet_id: 1, colony_id: 1, name: Names::Compound::DOCKYARD, level: 8
}
);
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::COMBUSTION, level: 1 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::SHIELD, level: 1 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::ENERGY, level: 6 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::BEAM, level: 6 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::WEAPONS, level: 3 });
set!(world, PlanetTechs { planet_id: 1, name: Names::Tech::PLASMA, level: 7 });
actions.colony.process_defence_build(1, DefenceBuildType::Celestia, 1);
actions.colony.process_defence_build(1, DefenceBuildType::Blaster, 1);
actions.colony.process_defence_build(1, DefenceBuildType::Beam, 1);
actions.colony.process_defence_build(1, DefenceBuildType::Astral, 1);
actions.colony.process_defence_build(1, DefenceBuildType::Plasma, 1);

let celestia_count = get!(world, (1, 1, Names::Defence::CELESTIA), ColonyDefences).count;
assert!(celestia_count == 1, "Colony: celestia not built correctly");
let blaster_count = get!(world, (1, 1, Names::Defence::BLASTER), ColonyDefences).count;
assert!(blaster_count == 1, "Colony: blaster not built correctly");
let beam_count = get!(world, (1, 1, Names::Defence::BEAM), ColonyDefences).count;
assert!(beam_count == 1, "Colony: beam not built correctly");
let astral_count = get!(world, (1, 1, Names::Defence::ASTRAL), ColonyDefences).count;
assert!(astral_count == 1, "Colony: astral not built correctly");
let plasma_count = get!(world, (1, 1, Names::Defence::PLASMA), ColonyDefences).count;
assert!(plasma_count == 1, "Colony: plasma not built correctly");
}
}

0 comments on commit 39c6619

Please sign in to comment.