diff --git a/tests/cogs/downloader/test_downloader.py b/tests/cogs/downloader/test_downloader.py index 9c894622c..9b3eab07d 100644 --- a/tests/cogs/downloader/test_downloader.py +++ b/tests/cogs/downloader/test_downloader.py @@ -110,6 +110,18 @@ async def test_add_repo(monkeypatch, repo_manager): assert squid.available_modules == [] +@pytest.mark.asyncio +async def test_remove_repo(monkeypatch, repo_manager): + monkeypatch.setattr("redbot.cogs.downloader.repo_manager.Repo._run", fake_run_noprint) + + await repo_manager.add_repo( + url="https://github.com/tekulvw/Squid-Plugins", name="squid", branch="rewrite_cogs" + ) + assert repo_manager.get_repo("squid") is not None + await repo_manager.delete_repo("squid") + assert repo_manager.get_repo("squid") is None + + @pytest.mark.asyncio async def test_current_branch(bot_repo): branch = await bot_repo.current_branch() diff --git a/tests/cogs/test_economy.py b/tests/cogs/test_economy.py index 9ecdacb2f..d502ae617 100644 --- a/tests/cogs/test_economy.py +++ b/tests/cogs/test_economy.py @@ -56,3 +56,27 @@ async def test_bank_can_spend(bank, member_factory): acc = await bank.get_account(mbr) canspendnow = await bank.can_spend(mbr, 100) assert canspendnow + + +@pytest.mark.asyncio +async def test_set_bank_name(bank, guild_factory): + guild = guild_factory.get() + await bank.set_bank_name("Test Bank", guild) + name = await bank.get_bank_name(guild) + assert name == "Test Bank" + + +@pytest.mark.asyncio +async def test_set_currency_name(bank, guild_factory): + guild = guild_factory.get() + await bank.set_currency_name("Coins", guild) + name = await bank.get_currency_name(guild) + assert name == "Coins" + + +@pytest.mark.asyncio +async def test_set_default_balance(bank, guild_factory): + guild = guild_factory.get() + await bank.set_default_balance(500, guild) + default_bal = await bank.get_default_balance(guild) + assert default_bal == 500