mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-21 18:27:59 -05:00
[Config] Asynchronous getters (#907)
* Make config get async * Asyncify alias * Asyncify bank * Asyncify cog manager * IT BOOTS * Asyncify core commands * Asyncify repo manager * Asyncify downloader * Asyncify economy * Asyncify alias TESTS * Asyncify economy TESTS * Asyncify downloader TESTS * Asyncify config TESTS * A bank thing * Asyncify Bank cog * Warning message in docs * Update docs with await syntax * Update docs with await syntax
This commit is contained in:
@@ -11,13 +11,14 @@ def bank(config):
|
||||
return bank
|
||||
|
||||
|
||||
def test_bank_register(bank, ctx):
|
||||
default_bal = bank.get_default_balance(ctx.guild)
|
||||
assert default_bal == bank.get_account(ctx.author).balance
|
||||
@pytest.mark.asyncio
|
||||
async def test_bank_register(bank, ctx):
|
||||
default_bal = await bank.get_default_balance(ctx.guild)
|
||||
assert default_bal == (await bank.get_account(ctx.author)).balance
|
||||
|
||||
|
||||
async def has_account(member, bank):
|
||||
balance = bank.get_balance(member)
|
||||
balance = await bank.get_balance(member)
|
||||
if balance == 0:
|
||||
balance = 1
|
||||
await bank.set_balance(member, balance)
|
||||
@@ -27,11 +28,11 @@ async def has_account(member, bank):
|
||||
async def test_bank_transfer(bank, member_factory):
|
||||
mbr1 = member_factory.get()
|
||||
mbr2 = member_factory.get()
|
||||
bal1 = bank.get_account(mbr1).balance
|
||||
bal2 = bank.get_account(mbr2).balance
|
||||
bal1 = (await bank.get_account(mbr1)).balance
|
||||
bal2 = (await bank.get_account(mbr2)).balance
|
||||
await bank.transfer_credits(mbr1, mbr2, 50)
|
||||
newbal1 = bank.get_account(mbr1).balance
|
||||
newbal2 = bank.get_account(mbr2).balance
|
||||
newbal1 = (await bank.get_account(mbr1)).balance
|
||||
newbal2 = (await bank.get_account(mbr2)).balance
|
||||
assert bal1 - 50 == newbal1
|
||||
assert bal2 + 50 == newbal2
|
||||
|
||||
@@ -40,16 +41,16 @@ async def test_bank_transfer(bank, member_factory):
|
||||
async def test_bank_set(bank, member_factory):
|
||||
mbr = member_factory.get()
|
||||
await bank.set_balance(mbr, 250)
|
||||
acc = bank.get_account(mbr)
|
||||
acc = await bank.get_account(mbr)
|
||||
assert acc.balance == 250
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_bank_can_spend(bank, member_factory):
|
||||
mbr = member_factory.get()
|
||||
canspend = bank.can_spend(mbr, 50)
|
||||
assert canspend == (50 < bank.get_default_balance(mbr.guild))
|
||||
canspend = await bank.can_spend(mbr, 50)
|
||||
assert canspend == (50 < await bank.get_default_balance(mbr.guild))
|
||||
await bank.set_balance(mbr, 200)
|
||||
acc = bank.get_account(mbr)
|
||||
canspendnow = bank.can_spend(mbr, 100)
|
||||
acc = await bank.get_account(mbr)
|
||||
canspendnow = await bank.can_spend(mbr, 100)
|
||||
assert canspendnow
|
||||
|
||||
Reference in New Issue
Block a user