auto_test.py

import os

blacklist = set()


def save_blacklist():
    global blacklist
    with open("blacklist.txt", "w") as file:
        for item in blacklist:
            file.write(f"{item}\n")


def load_blacklist():
    global blacklist
    blacklist.clear()
    if os.path.exists("blacklist.txt"):
        with open("blacklist.txt", "r") as file:
            for line in file:
                blacklist.add(line.strip())


def test_load_blacklist(tmpdir):
    # Create a temporary blacklist.txt file
    temp_file = tmpdir.join("blacklist.txt")
    temp_file.write("Notepad\nCalculator\n")

    # Change the working directory to the temporary directory
    os.chdir(tmpdir)

    load_blacklist()

    # Check if the blacklist contains the correct items
    assert len(blacklist) == 2


def test_add_to_blacklist():
    # Clear the blacklist and add an item
    blacklist.clear()
    blacklist.add("NewApp")

    # Save and reload the blacklist
    save_blacklist()
    load_blacklist()

    # Check that the item is in the blacklist
    assert "NewApp" in blacklist


def test_remove_from_blacklist():
    # Setup the blacklist with some items
    blacklist.clear()
    blacklist.add("AppToRemove")

    # Remove the item
    blacklist.remove("AppToRemove")

    # Save and reload the blacklist
    save_blacklist()
    load_blacklist()

    # Check that the item is not in the blacklist
    assert "AppToRemove" not in blacklist