def test_upgrade_building():
game_state = initialise_gamestate()
game = Game(game_state)
upgrade_success = game.action_upgrade_building(
"max", PlanetLocation(1, 1, 3), "metal_mine"
)
assert not upgrade_success
game.update(3600)
upgrade_success = game.action_upgrade_building(
"max", PlanetLocation(1, 1, 3), "metal_mine"
)
planet_component = game_state.world.planets[PlanetLocation(1, 1, 3)].components[
PlanetComponent
]
assert upgrade_success
assert planet_component.resources[Resources.Metal] < 30 * game_state.world.speed
assert (
int(planet_component.resources[Resources.Cristal]) < 20 * game_state.world.speed
)
with pytest.raises(AssertionError) as excinfo:
game.action_upgrade_building("max", "ploup", "metal_mine")
assert "unkwnown planet" in str(excinfo.value).lower()
with pytest.raises(AssertionError) as excinfo:
game.action_upgrade_building("ploup", PlanetLocation(1, 1, 3), "metal_mine")
assert "unknown player" in str(excinfo.value).lower()
def test_universe_speed():
game_state = initialise_gamestate()
game = Game(game_state)
game_state.world.speed = 10000000
earth = game_state.world.planets[PlanetLocation(1, 1, 3)]
game.update(1)
assert earth.components[PlanetComponent].resources[Resources.Metal] == 10000
def test_fleet_transport():
game_state = init_state_complex()
game = Game(game_state)
game.update(100)
fleet = game_state.players["max"].fleets[0]
fleet_comp = fleet.components[FleetComponent]
MissionSystem.order_mission(
fleet,
"TRANSPORT",
PlanetLocation(1, 1, 4),
{Resources.Metal: 2, Resources.Cristal: 0.5},
)
game.update(10)
assert fleet_comp.travelling_to == PlanetLocation(1, 1, 4)
assert fleet_comp.cargo[Resources.Metal] == 2
game.update(100),
assert fleet_comp.mission == "RETURN"
assert fleet_comp.travelling_to == PlanetLocation(1, 1, 3)
assert fleet_comp.cargo[Resources.Metal] == 0
def test_create_new_player():
game_state = initialise_empty_universe()
game = Game(game_state)
success = game.create_new_player("john", "John")
assert success
john_planets = PositionSystem.get_player_planets("john")
assert (
game_state.world.planets[john_planets[0]].components[PlanetComponent].owner_id
== "john"
)
success = game.create_new_player("john", "Michel")
assert not success
def test_game_update():
game_state = initialise_gamestate()
game = Game(game_state)
earth = game.game_state.world.planets[PlanetLocation(1, 1, 3)]
game.update(3600)
assert (
earth.components[PlanetComponent].resources[Resources.Metal]
== 30 * game_state.world.speed
)
earth.buildings["metal_mine"].components[BuildingComponent].level += 1
earth.buildings["solar_plant"].components[BuildingComponent].level += 1
# assert earth.components[PlanetComponent].production_per_second == 800
game.update(3600)
assert (
earth.components[PlanetComponent].resources[Resources.Metal]
> 60 * game_state.world.speed
)
def test_combat():
game_state = init_state_complex()
game = Game(game_state)
earth = PlanetLocation(1, 1, 3)
jupiter = PlanetLocation(1, 1, 5)
game_state.world.planets[jupiter].buildings["metal_mine"].components[
BuildingComponent
].level = 0
game.update(360)
game.action_send_mission("max", earth, "ATTACK", jupiter)
attacking_fleet = game_state.players["max"].fleets[0]
assert attacking_fleet.components[FleetComponent].mission == "ATTACK"
assert attacking_fleet.ships[0].components[ShipComponent].number == 10
assert attacking_fleet.components[FleetComponent].cargo == {
Resources.Metal: 0,
Resources.Cristal: 0,
Resources.Deuterium: 0,
}
travelling_time = attacking_fleet.components[FleetComponent].travel_time_left
game.update(travelling_time - 1)
assert (
game_state.world.planets[jupiter]
.components[PlanetComponent]
.resources[Resources.Metal]
> 0
)
game.update(1)
assert game_state.world.planets[jupiter].components[PlanetComponent].resources == {
Resources.Metal: 0,
Resources.Cristal: 0,
Resources.Deuterium: 0,
}
assert len(game_state.players["bob"].fleets) == 0
assert attacking_fleet.ships[0].components[ShipComponent].number == 7
assert attacking_fleet.components[FleetComponent].cargo[Resources.Metal] > 0
assert attacking_fleet.components[FleetComponent].cargo[Resources.Cristal] > 0
assert attacking_fleet.components[FleetComponent].cargo[Resources.Deuterium] > 0
game.update(20)
def test_colonize_planet():
game_state = init_state_complex()
game = Game(game_state)
earth = PlanetLocation(1, 1, 3)
game.update(100)
game.action_upgrade_building("max", earth, "shipyard")
game.update(100)
game.action_build_ship("max", earth, "colony_ship")
game.action_send_mission(
"max", earth, "COLONIZE", PlanetLocation(1, 1, 6), cargo={Resources.Metal: 5}
)
game.update(25)
game.update(25)
assert len(game_state.world.planets) == 4
new_planet = game_state.world.planets[PlanetLocation(1, 1, 6)]
new_planet_comp = new_planet.components[PlanetComponent]
assert new_planet.buildings["metal_mine"].components[BuildingComponent].level == 0
def test_fleet_transport_from_game():
game_state = init_state_complex()
game = Game(game_state)
game.update(3600)
earth = PlanetLocation(1, 1, 3)
assert (
game_state.world.planets[earth]
.components[PlanetComponent]
.resources[Resources.Metal]
== 30 * game_state.world.speed
)
game.action_send_mission(
"max",
PlanetLocation(1, 1, 3),
"TRANSPORT",
PlanetLocation(1, 1, 4),
{Resources.Metal: 2, Resources.Cristal: 2},
)
assert (
game_state.world.planets[earth]
.components[PlanetComponent]
.resources[Resources.Metal]
== 30 * game_state.world.speed - 2
)
game.update(5)
fleets = game.get_player_fleets("max")
fleet_comp = fleets[0].components[FleetComponent]
assert fleet_comp.travelling_to == PlanetLocation(1, 1, 4)
assert fleet_comp.cargo[Resources.Metal] == 2
game.update(100)
assert fleet_comp.mission == "RETURN"
assert fleet_comp.travelling_to == PlanetLocation(1, 1, 3)
assert fleet_comp.cargo[Resources.Metal] == 0
def test_build_ship():
game_state = initialise_gamestate()
game = Game(game_state)
game_state.world.speed = 1000000
earth = PlanetLocation(1, 1, 3)
outcome = game.action_build_ship("max", earth, "light_fighter")
assert not outcome.success
game.update(24 * 3600)
success = game.action_upgrade_building("max", earth, "shipyard")
assert success
outcome = game.action_build_ship("max", earth, "light_fighter")
assert not outcome.success
assert game.action_upgrade_building("max", earth, "research_lab")
assert game.action_upgrade_research("max", earth, "energy")
assert game.action_upgrade_research("max", earth, "combustion_drive")
outcome = game.action_build_ship("max", earth, "light_fighter")
assert outcome.success
assert len(game_state.players["max"].fleets) == 1
assert (
game_state.players["max"]
.fleets[0]
.light_fighter.components[ShipComponent]
.number
== 1
)
def test_upgrade_research():
game_state = initialise_gamestate()
game = Game(game_state)
game.update(24 * 3600)
earth = PlanetLocation(1, 1, 3)
upgrade_success = game.action_upgrade_research("max", earth, "energy")
assert not upgrade_success
upgrade_success = game.action_upgrade_building("max", earth, "research_lab")
assert upgrade_success
upgrade_success = game.action_upgrade_research("max", earth, "laser")
assert not upgrade_success
upgrade_success = game.action_upgrade_research("max", earth, "energy")
assert upgrade_success
upgrade_success = game.action_upgrade_research("max", earth, "laser")
assert not upgrade_success
upgrade_success = game.action_upgrade_building("max", earth, "research_lab")
upgrade_sucess = game.action_upgrade_research("max", earth, "energy")
upgrade_sucess = game.action_upgrade_research("max", earth, "laser")
assert upgrade_success
def send_mission(player_id: str, planet_id: str, mission: str):
content = request.json
assert "destination_id" in content
destination_id = content["destination_id"]
if "cargo" in content:
assert isinstance(content["cargo"], dict)
cargo = {
Resources(res): quantity
for res, quantity in content["cargo"].items()
}
else:
cargo = None
action_outcome = game_thread.action_send_mission(
player_id,
PlanetLocation.from_str(planet_id),
mission,
PlanetLocation.from_str(destination_id),
cargo,
)
return action_outcome.to_json()
if __name__ == "__main__":
game_state = init_fast_empty()
game_thread = Game(game_state)
game_thread.start()
app.run()