def test_pickup_data_for_pb_expansion_unlocked(echoes_item_database, echoes_resource_database):
# Setup
pickup = pickup_creator.create_ammo_expansion(
echoes_item_database.ammo["Power Bomb Expansion"],
[2],
False,
echoes_resource_database,
)
creator = pickup_exporter.PickupExporterSolo(patch_data_factory._simplified_memo_data())
# Run
details = creator.export(PickupIndex(0), PickupTarget(pickup, 0), pickup, PickupModelStyle.ALL_VISIBLE)
result = patch_data_factory.echoes_pickup_details_to_patcher(details, MagicMock())
# Assert
assert result == {
"pickup_index": 0,
"scan": "Power Bomb Expansion. Provides 2 Power Bombs and 1 Item Percentage",
"model": {"game": "prime2", "name": "PowerBombExpansion"},
"hud_text": ["Power Bomb Expansion acquired!"],
'resources': [{'amount': 2, 'index': 43},
{'amount': 1, 'index': 47}],
"conditional_resources": [],
"convert": [],
}
def test_get_single_hud_text_locked_pbs():
# Run
result = pickup_exporter._get_single_hud_text("Locked Power Bomb Expansion",
patch_data_factory._simplified_memo_data(),
tuple())
# Assert
assert result == "Power Bomb Expansion acquired, but the main Power Bomb is required to use it."
def test_create_pickup_all_from_pool(echoes_resource_database,
default_echoes_configuration,
disable_hud_popup: bool
):
item_pool = pool_creator.calculate_pool_results(default_echoes_configuration,
echoes_resource_database)
index = PickupIndex(0)
if disable_hud_popup:
memo_data = patch_data_factory._simplified_memo_data()
else:
memo_data = default_prime2_memo_data()
creator = pickup_exporter.PickupExporterSolo(memo_data)
for item in item_pool.pickups:
data = creator.export(index, PickupTarget(item, 0), item, PickupModelStyle.ALL_VISIBLE)
for hud_text in data.hud_text:
assert not hud_text.startswith("Locked")
def test_pickup_data_for_pb_expansion_locked(simplified, echoes_item_database, echoes_resource_database):
# Setup
pickup = pickup_creator.create_ammo_expansion(
echoes_item_database.ammo["Power Bomb Expansion"],
[2],
True,
echoes_resource_database,
)
if simplified:
memo = patch_data_factory._simplified_memo_data()
hud_text = [
"Power Bomb Expansion acquired, but the main Power Bomb is required to use it.",
"Power Bomb Expansion acquired!",
]
else:
memo = default_database.default_prime2_memo_data()
hud_text = [
"Power Bomb Expansion acquired! \n"
"Without the main Power Bomb item, you are still unable to release Power Bombs.",
"Power Bomb Expansion acquired! \nMaximum Power Bomb carrying capacity increased by 2.",
]
creator = pickup_exporter.PickupExporterSolo(memo)
# Run
details = creator.export(PickupIndex(0), PickupTarget(pickup, 0), pickup, PickupModelStyle.ALL_VISIBLE)
result = patch_data_factory.echoes_pickup_details_to_patcher(details, MagicMock())
# Assert
assert result == {
"pickup_index": 0,
"scan": "Power Bomb Expansion. Provides 2 Power Bombs and 1 Item Percentage",
"model": {"game": "prime2", "name": "PowerBombExpansion"},
"hud_text": hud_text,
'resources': [{'amount': 2, 'index': 72},
{'amount': 1, 'index': 47}],
"conditional_resources": [
{'item': 43,
'resources': [{'amount': 2, 'index': 43},
{'amount': 1, 'index': 47}]}
],
"convert": [],
}
def test_pickup_data_for_seeker_launcher(echoes_item_database, echoes_resource_database):
# Setup
state = MajorItemState(
include_copy_in_original_location=False,
num_shuffled_pickups=0,
num_included_in_starting_items=0,
included_ammo=(5,),
)
pickup = pickup_creator.create_major_item(
echoes_item_database.major_items["Seeker Launcher"],
state,
True,
echoes_resource_database,
echoes_item_database.ammo["Missile Expansion"],
True
)
creator = pickup_exporter.PickupExporterSolo(patch_data_factory._simplified_memo_data())
# Run
details = creator.export(PickupIndex(0), PickupTarget(pickup, 0), pickup, PickupModelStyle.ALL_VISIBLE)
result = patch_data_factory.echoes_pickup_details_to_patcher(details, MagicMock())
# Assert
assert result == {
"pickup_index": 0,
"scan": "Seeker Launcher",
"model": {"game": "prime2", "name": "SeekerLauncher"},
"hud_text": ["Seeker Launcher acquired, but the Missile Launcher is required to use it.",
"Seeker Launcher acquired!"],
'resources': [{'amount': 5, 'index': 71},
{'amount': 1, 'index': 47},
{'amount': 1, 'index': 26}],
"conditional_resources": [
{'item': 73,
'resources': [{'amount': 5, 'index': 44},
{'amount': 1, 'index': 47},
{'amount': 1, 'index': 26}]}
],
"convert": [],
}
