def test_no_mana_to_steal(self):
madonna = User("madonna")
madonna.set_value("mana", 0)
bowie = User("bowie")
handbag = Command("handbag").save().allow_user("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
result = subject.steal()
assert (
result.metadata["stealing_result"]
== "@madonna has no Mana to steal from @bowie"
)
def test_stole_from_a_rich_person(self):
random.seed(1)
madonna = User("madonna")
madonna.set_value("cool_points", 1)
bowie = User("bowie")
bowie.set_value("cool_points", 10)
handbag = Command("handbag").save().allow_user("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
result = subject.steal()
assert (
result.metadata["stealing_result"]
== "@madonna stole from @bowie. Chance of Success: 39%"
)
def test_trying_to_steal_sound_you_do_not_own(self):
madonna = User("madonna")
bowie = User("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
assert "handbag" not in madonna.commands()
assert "handbag" not in bowie.commands()
result = subject.steal()
assert isinstance(result, Result)
assert "handbag" not in madonna.commands()
assert "handbag" not in bowie.commands()
result.metadata[
"stealing_result"
] == "@madonna failed to steal !handbag from @bowie"
def test_stealing(self):
madonna = User("madonna")
bowie = User("bowie")
handbag = Command("handbag").save().allow_user("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
assert "handbag" not in madonna.commands()
assert "handbag" in bowie.commands()
result = subject.steal()
assert isinstance(result, Result)
assert "handbag" in madonna.commands()
assert "handbag" not in bowie.commands()
assert madonna.mana() == 1
result.metadata["stealing_result"] == "@madonna stole from @bowie"
def test_stealing_from_an_insured_person(self):
random.seed(0)
madonna = User("madonna")
madonna.set_value("cool_points", 1)
bowie = User("bowie")
bowie.set_value("cool_points", 10)
bowie.buy_insurance()
assert bowie.insured()
handbag = Command("handbag").save().allow_user("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
result = subject.steal()
assert (
result.metadata["stealing_result"]
== f"@madonna was blocked by @bowie's insurance! Num Attempts: 0"
)
assert not bowie.insured()
def test_caught_stealing(self):
random.seed(5)
madonna = User("madonna")
bowie = User("bowie")
handbag = Command("handbag").save().allow_user("bowie")
subject = Stealer(thief="madonna", target_sfx="handbag", victim="bowie")
assert madonna.mana() == 3
assert "handbag" not in madonna.commands()
assert "handbag" in bowie.commands()
result = subject.steal()
assert "handbag" not in madonna.commands()
assert "handbag" in bowie.commands()
assert isinstance(result, Result)
assert madonna.mana() == 0
assert (
"@madonna WAS CAUGHT STEALING! Chance of Success:"
in result.metadata["stealing_result"]
)
def steal(self) -> str:
if self.parser.target_user and self.parser.target_sfx:
result = Stealer(
thief=self.user,
victim=self.parser.target_user,
target_sfx=self.parser.target_sfx,
).steal()
if User(self.parser.target_user).creator() == self.user:
return f"You cannot steal from your own bot @{self.user} @{self.parser.target_user}"
elif User(self.user).creator() == self.parser.target_user:
return f"You cannot steal from your creator @{self.user} @{self.parser.target_user}"
else:
return StealFormatter(result).format()
else:
msg = f"@{self.user} you must specify who and what you want to steal."
if self.args:
msg += f" Invalid Args: {' '.join(self.args)}"
return msg