def test_adding_buff(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
# Add the buff
add_buff(buffable, buff, CompleteBuildingEvent())
# Since it has no triggers or conditions, it was added automatically
assert buffable.attributes[Attributes.DEF] == 30
assert buff.buff_id in buffable.active_buffs
def test_removing_buff(self):
buffable = Buffable()
buffable.attributes[Attributes.ATK] = 100
buff = BuffSpec()
buff.modifiers = [Modifier("%", 0.5, Attributes.ATK)]
add_buff(buffable, buff, CompleteBuildingEvent())
assert buffable.attributes[Attributes.ATK] == 150
remove_buff(buffable, buff.buff_id)
assert buffable.attributes[Attributes.ATK] == 100
assert buff.buff_id not in buffable.active_buffs
def test_buff_modification_history(self):
buffable = Buffable()
buff = BuffSpec()
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
# Faking an event that would result in adding a buff
event_to_get_the_buff = CompleteBuildingEvent()
# Add the buff
event_result = add_buff(buffable, buff, event_to_get_the_buff)
added_modifications = event_result.added_modifications
# Check modification history to debug/backtrack
assert added_modifications[0].modifier.operator == "+"
assert added_modifications[0].modifier.attribute_id == Attributes.DEF
assert added_modifications[
0].source_event.trigger_event == event_to_get_the_buff
def test_expiring_stacks(self):
buffable = Buffable()
buff = BuffSpec(1)
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.duration_seconds = 10 # Only lasts for 10 seconds
buff.max_stack = 2
buff.modifiers = [Modifier("+", 50, Attributes.DEF)]
# Add the buff 2 times, one in the future
add_buff(buffable, buff, CompleteBuildingEvent())
with FixedTime(get_timestamp() + 5):
add_buff(buffable, buff, CompleteBuildingEvent())
expiry_time_1 = get_timestamp() + buff.duration_seconds
expiry_time_2 = expiry_time_1 + 5
assert buffable.attributes[Attributes.DEF] == 100
assert buffable.active_buffs[buff.buff_id].stack == 2
assert len(buffable.expiry_times) == 2
with FixedTime(expiry_time_1):
assert buffable.attributes[Attributes.DEF] == 50
assert buffable.active_buffs[buff.buff_id].stack == 1
assert len(buffable.expiry_times) == 1
with FixedTime(expiry_time_2):
assert buffable.attributes[Attributes.DEF] == 0
assert buff.buff_id not in buffable.active_buffs
assert len(buffable.expiry_times) == 0
def test_next_to_expire(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.duration_seconds = 10 # Only lasts for 10 seconds
buff.max_stack = 3
buff.modifiers = [Modifier("+", 50, Attributes.DEF)]
with FixedTime(get_timestamp()):
expiry_time = get_timestamp() + buff.duration_seconds
# Add the buff
add_buff(buffable, buff, CompleteBuildingEvent())
add_buff(buffable, buff, CompleteBuildingEvent())
add_buff(buffable, buff, CompleteBuildingEvent())
assert _get_next_expiry_time(buffable) == expiry_time
expired_buffs = list(get_expired_buffs(buffable))
assert len(expired_buffs) == 0
with FixedTime(expiry_time):
expired_buffs = list(get_expired_buffs(buffable))
assert len(expired_buffs) == 3
# Calling expired buffs should have removed them from expiry list
assert len(buffable.expiry_times) == 0
def test_condition_parameters(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = ["DamageEvent"]
buff.deactivation_triggers = ["DamageEvent"]
buff.conditions = ["is_damage_higher_then 10"
] # condition parameters after condition name
buff.buff_id = 5
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
buffspecs.register_buff(buff)
@buffspecs.AddConditionFor([DamageEvent])
def is_damage_higher_then(event, param):
return event.damage > param
# Add the buff
add_buff(buffable, buff, CompleteBuildingEvent())
# Damage lower then condition threshhold
call_event(DamageEvent(buffable, 8))
# Buff should not have been activated
assert buff.buff_id not in buffable.active_buffs
assert buffable.attributes[Attributes.DEF] == 0
# Now a damage higher
call_event(DamageEvent(buffable, 12))
# Buff should have been activated
assert buff.buff_id in buffable.active_buffs
assert buffable.attributes[Attributes.DEF] == 30
def test_adding_multiple_modifiers(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.modifiers = [
Modifier("+", 50, Attributes.DEF),
Modifier("+", 50, Attributes.DEF),
Modifier("%", 0.5, Attributes.DEF)
]
buffspecs.register_buff(buff) # Add buff to registry
# Adding a possible buff
add_buff(buffable, buff, CompleteBuildingEvent())
# +50 +50 + 50% is 100 + 50% = 150
asd = buffable.attributes.get_data(Attributes.DEF)
assert buffable.attributes[Attributes.DEF] == 150
def test_registering_1_expiry_per_stack(self):
buffable = Buffable()
buff = BuffSpec(1)
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.duration_seconds = 10 # Only lasts for 10 seconds
buff.max_stack = 2
buff.modifiers = [Modifier("+", 50, Attributes.DEF)]
# Add the buff 2 times
add_buff(buffable, buff, CompleteBuildingEvent())
add_buff(buffable, buff, CompleteBuildingEvent())
# Should have registered one expiry time per stack
assert buffable.attributes[Attributes.DEF] == 100
assert buffable.active_buffs[buff.buff_id].stack == 2
assert len(buffable.expiry_times) == 2
def test_negating_condition(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = {} # no triggers
buff.conditions = ["not cond_is_blue_yellow"]
buff.buff_id = 5
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
buffspecs.register_buff(buff)
@buffspecs.AddConditionFor([CompleteBuildingEvent])
def cond_is_blue_yellow(event):
return False
add_buff(buffable, buff, CompleteBuildingEvent())
# This buff should be be applied as the condition was negated
assert buff.buff_id in buffable.active_buffs
def test_basic_condition_failing(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = {} # no triggers
buff.conditions = ["cond_is_blue_yellow"]
buff.buff_id = 5
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
buffspecs.register_buff(buff)
# A condition that can be triggered by a MockEvent
@buffspecs.AddConditionFor([CompleteBuildingEvent])
def cond_is_blue_yellow(event):
return False
add_buff(buffable, buff, CompleteBuildingEvent())
# This buff should not be applied
assert buff.buff_id not in buffable.active_buffs
def test_buff_expiring(self):
buffable = Buffable()
buff = BuffSpec(1)
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.duration_seconds = 10 # Only lasts for 10 seconds
buff.modifiers = [Modifier("+", 50, Attributes.DEF)]
expiry_time = get_timestamp() + buff.duration_seconds
# Add the buff
add_buff(buffable, buff, CompleteBuildingEvent())
assert buffable.attributes[Attributes.DEF] == 50
with FixedTime(expiry_time):
# Simply by reading the attribute we will expire that buff
assert buffable.attributes[Attributes.DEF] == 0
# Should not be an active buff anymore
assert buff.buff_id not in buffable.active_buffs
# Expiry time
assert len(buffable.expiry_times) == 0
def test_adding_buff(self):
buffable = Buffable()
buff = BuffSpec(1)
buff.activation_triggers = [] # no triggers
buff.conditions = [] # no conditions
buff.duration_seconds = 10 # Only lasts for 10 seconds
buff.modifiers = [Modifier("+", 50, Attributes.DEF)]
with FixedTime(get_timestamp()):
# Add the buff
expiry_time = get_timestamp() + 10
add_buff(buffable, buff, CompleteBuildingEvent())
# Since it has no triggers or conditions, it was added automatically
assert buffable.attributes[Attributes.DEF] == 50
assert buff.buff_id in buffable.active_buffs
# Check the expiry time was registered
registered_expiry_time, buff_id = buffable.expiry_times[0]
assert registered_expiry_time == expiry_time
assert buff_id == buff.buff_id
def test_buff_trigger(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = ["FartEvent"]
buff.conditions = []
buff.buff_id = 5
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
buffspecs.register_buff(buff)
add_buff(buffable, buff, CompleteBuildingEvent())
# The buff was not triggered yet as "FartEvent" was not called
assert buffable.attributes[Attributes.DEF] == 0
# Instead, we did not added a buff, we just added a trigger for a buff
assert "FartEvent" in buffable.activation_triggers
# Now we call the event
call_event(FartEvent(buffable))
# Now the buff should be added
assert buffable.attributes[Attributes.DEF] == 30
# And the trigger removed
assert "FartEvent" not in buffable.activation_triggers
def test_condition_switching_buff(self):
buffable = Buffable()
buff = BuffSpec()
buff.activation_triggers = ["DamageEvent"]
buff.deactivation_triggers = ["DamageEvent"]
buff.conditions = ["is_burning"]
buff.buff_id = 5
buff.modifiers = [Modifier("+", 30, Attributes.DEF)]
buffspecs.register_buff(buff)
buffable.attributes[
"Burning"] = 0 # example to set a state, not burning
@buffspecs.AddConditionFor([DamageEvent])
def is_burning(event):
return event.buffable.attributes["Burning"] == 1
# Add the buff
add_buff(buffable, buff, CompleteBuildingEvent())
# The buff should not be applied because the buffable is not burning
assert buff.buff_id not in buffable.active_buffs
assert buffable.attributes[Attributes.DEF] == 0
# Now make it burn
damage = 10
buffable.attributes["Burning"] = 1
call_event(DamageEvent(buffable, damage))
# Now the buff should have applied
assert buff.buff_id in buffable.active_buffs
assert buffable.attributes[Attributes.DEF] == 30
# And we should have added the remove trigger
assert buff.buff_id in buffable.deactivation_triggers["DamageEvent"]
# Buff history contains this modification
assert len(buffable.attributes.get_data(Attributes.DEF).history) == 1
# Now After calling the other event the buff should be removed
buffable.attributes["Burning"] = 0
call_event(DamageEvent(buffable, damage))
# Now the buff should be removed
assert buff.buff_id not in buffable.active_buffs
assert buff.buff_id not in buffable.deactivation_triggers
assert buffable.attributes[Attributes.DEF] == 0
# And the trigger should be added again and the remove trigger should be removed
assert buff.buff_id not in buffable.deactivation_triggers[
"DamageEvent"]
assert buff.buff_id in buffable.activation_triggers["DamageEvent"]
# Also the modification history is removed because this buff is inactive and not modifyng anything
assert len(buffable.attributes.get_data(Attributes.DEF).history) == 0
# In case we burn again...
buffable.attributes["Burning"] = 1
call_event(DamageEvent(buffable, damage))
# Buff is activated again
assert buff.buff_id in buffable.active_buffs
assert buffable.attributes[Attributes.DEF] == 30