From 119229cce723730c4f9c6d155845effe0402a913 Mon Sep 17 00:00:00 2001 From: Adam Williamson Date: Wed, 26 Apr 2023 17:41:25 -0700 Subject: [PATCH] Add a needle cleanup script, enhance the needle check script This adds a new script - cleanup-needles.py - to use for cleaning up old needles. It has to be used in conjunction with a database query; the comment at the top explains how to do that query and export the needed information. It produces a git commit with needles that haven't matched since a certain date (specified in the sql query) removed, subject to a 'keeplist' of needles we keep even if they seem to be old. I also enhanced check-needles.py to check for cases where tests seem to be trying to match a tag we have no needles for. This was necessary to find cases where the cleanup script was too aggressive (i.e. the things that wound up in the 'keeplist'), but it also turned out to find quite a lot of cases where the code really *was* looking for a needle that had gone in a previous cleanup or that never existed; the commits before this one clean up a lot of those cases. The code to decide which string literals are needle tags is pretty dumb and hacky and needs some manual cueing sometimes - that's what the `# testtag` changes in this commit are for. Making it smarter would probably require this script to get a lot more complicated and either incorporate or become a tokenizer, which I don't really want to do. Signed-off-by: Adam Williamson --- check-needles.py | 170 +++++++++++++++++++++--------- cleanup-needles.py | 77 ++++++++++++++ lib/freeipa.pm | 4 +- tests/_software_selection.pm | 4 +- tests/desktop_update_graphical.pm | 3 +- 5 files changed, 203 insertions(+), 55 deletions(-) create mode 100755 cleanup-needles.py diff --git a/check-needles.py b/check-needles.py index 6114b87d..b4f8b9b7 100755 --- a/check-needles.py +++ b/check-needles.py @@ -19,9 +19,19 @@ # # Author: Adam Williamson -"""This is a check script which checks for unused needles. If none of -the tags a needle declares is referenced in the tests, it is -considered unused. +"""This is a check script which checks for: + +1. Unused needles - if none of the tags a needle declares is referenced +in the tests, it is considered unused. +2. Tag assertions with no needle - if a test seems to be asserting or +checking for a tag, but there is no needle with that tag. The code to +decide what string literals in the tests are tags is not perfect. If +a literal that *is* a tag is not being counted as one, you may need to +rejig the code, or add `# testtag` to the end of the line to cue this +script to consider it a tag. If a tag does not have a - or _ in it, it +must be added to the `knowns` list. +3. Image files in the needles directory with no matching JSON file. +4. JSON files in the needles directory with no matching image file. """ import glob @@ -38,110 +48,152 @@ LIBPATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "lib") DOUBLEQUOTERE = re.compile('"(.*?)"') SINGLEQUOTERE = re.compile("'(.*?)'") -# first we're gonna build a big list of all string literals +# first we're gonna build a big list of all string literals that look +# like they're needle tags testpaths = glob.glob(f"{TESTSPATH}/**/*.pm", recursive=True) testpaths.extend(glob.glob(f"{LIBPATH}/**/*.pm", recursive=True)) -testliterals = [] +testtags = [] for testpath in testpaths: # skip if it's a symlink if os.path.islink(testpath): continue # otherwise, scan it for string literals with open(testpath, "r") as testfh: - testtext = testfh.read() - for match in DOUBLEQUOTERE.finditer(testtext): - testliterals.append(match[1]) - for match in SINGLEQUOTERE.finditer(testtext): - testliterals.append(match[1]) + testlines = testfh.readlines() + for line in testlines: + matchfuncs = ( + "assert_screen", + "assert_and_click", + "check_screen", + "start_with_launcher", + "send_key_until_needlematch", + "# testtag" + ) + for matchfunc in matchfuncs: + if matchfunc == "# testtag" and matchfunc in line: + # for the comment tag we should take all literals from + # the whole line + start = 0 + else: + # for match functions we should only take literals + # after the function name + start = line.find(matchfunc) + # fortunately `find` returns -1 for 'no match' + if start > -1: + for match in DOUBLEQUOTERE.finditer(line[start:]): + testtags.append(match[1]) + for match in SINGLEQUOTERE.finditer(line[start:]): + testtags.append(match[1]) + if matchfunc == "send_key_until_needlematch": + # strip last match because it'll be the key to hit + testtags.pop() +# filter the list a bit for matches that aren't tags. Almost all our +# tags have a - or an _ in them, except a small number of weirdos; +# this filters out a lot of false matches (like keypresses) +knowns = ("_", "-", "bootloader", "browser", "firefox") +testtags = [tag for tag in testtags if + tag and + not tag.isdigit() and + not tag.isupper() and + any(known in tag for known in knowns)] + +# keep this around for the tagnoneedle check later; we can't use +# the 'synthetic' tags in that check as some of them we know don't +# have needles (e.g. the range(30,100) background tags, most of those +# don't exist yet) +realtesttags = set(tag for tag in testtags if "$" not in tag) # now let's do some whitelisting, for awkward cases where we know that # we concatenate string literals and stuff # versioned backgrounds and release IDs for rel in range(30, 100): - testliterals.append(f"{rel}_background") - testliterals.append(f"{rel}_background_dark") - testliterals.append(f"version_{rel}_ident") + testtags.append(f"{rel}_background") + testtags.append(f"{rel}_background_dark") + testtags.append(f"version_{rel}_ident") # anaconda id needles, using tell_source for source in ("workstation", "generic", "server"): - testliterals.append(f"leftbar_{source}") - testliterals.append(f"topbar_{source}") + testtags.append(f"leftbar_{source}") + testtags.append(f"topbar_{source}") # keyboard layout switching, using desktop_switch_layout for environment in ("anaconda", "gnome"): for layout in ("native", "ascii"): - testliterals.append(f"{environment}_layout_{layout}") + testtags.append(f"{environment}_layout_{layout}") # package set selection, using get_var('PACKAGE_SET') for pkgset in ("kde", "workstation", "minimal"): - testliterals.append(f"anaconda_{pkgset}_highlighted") - testliterals.append(f"anaconda_{pkgset}_selected") + testtags.append(f"anaconda_{pkgset}_highlighted") + testtags.append(f"anaconda_{pkgset}_selected") # desktop_login stuff for user in ("jack", "jim"): - testliterals.append(f"login_{user}") - testliterals.append(f"user_confirm_{user}") + testtags.append(f"login_{user}") + testtags.append(f"user_confirm_{user}") # partitioning stuff, there's a bunch of this, all in anaconda.pm # multiple things use this for part in ("swap", "root", "efi", "boot", "bootefi", "home", "vda2"): - testliterals.append(f"anaconda_part_select_{part}") - testliterals.append(f"anaconda_blivet_part_inactive_{part}") + testtags.append(f"anaconda_part_select_{part}") + testtags.append(f"anaconda_blivet_part_inactive_{part}") # select_disks for num in range(1, 10): - testliterals.append(f"anaconda_install_destination_select_disk_{num}") + testtags.append(f"anaconda_install_destination_select_disk_{num}") # custom_scheme_select for scheme in ("standard", "lvmthin", "btrfs", "lvm"): - testliterals.append(f"anaconda_part_scheme_{scheme}") + testtags.append(f"anaconda_part_scheme_{scheme}") # custom_blivet_add_partition for dtype in ("lvmvg", "lvmlv", "lvmthin", "raid"): - testliterals.append(f"anaconda_blivet_part_devicetype_{dtype}") + testtags.append(f"anaconda_blivet_part_devicetype_{dtype}") for fsys in ("ext4", "xfs", "btrfs", "ppc_prep_boot", "swap", "efi_filesystem", "biosboot"): - testliterals.append(f"anaconda_blivet_part_fs_{fsys}") - testliterals.append(f"anaconda_blivet_part_fs_{fsys}_selected") + testtags.append(f"anaconda_blivet_part_fs_{fsys}") + testtags.append(f"anaconda_blivet_part_fs_{fsys}_selected") +# this is variable-y in custom_blivet_resize_partition but we only +# call it with 'GiB' (in disk_custom_blivet_resize_lvm.pm) +testtags.append("anaconda_blivet_size_unit_GiB") # this is variable-y in custom_change_type but we only actually have # one value -testliterals.append("anaconda_part_device_type_raid") +testtags.append("anaconda_part_device_type_raid") # custom_change_fs for fsys in ("xfs", "ext4"): - testliterals.append(f"anaconda_part_fs_{fsys}") - testliterals.append(f"anaconda_part_fs_{fsys}_selected") + testtags.append(f"anaconda_part_fs_{fsys}") + testtags.append(f"anaconda_part_fs_{fsys}_selected") # Needles for Help viewer for section in ("desktop", "networking", "sound", "files", "user", "hardware", "accessibility", "tipstricks", "morehelp"): - testliterals.append(f"help_section_{section}") - testliterals.append(f"help_section_content_{section}") + testtags.append(f"help_section_{section}") + testtags.append(f"help_section_content_{section}") # Needles for Calculator for button in ("div", "divider", "zero", "one", "two", "three", "four", "five", "six","seven", "eight", "nine", "mod", "percent", "pi", "root", "square", "sub"): - testliterals.append(f"calc_button_{button}") + testtags.append(f"calc_button_{button}") for result in ("BokZw", "Czo4s", "O9qsL", "WIxiR", "b5y2B", "h7MfO", "qxuBK", "tWshx", "uC8Ul", "3LAG3"): - testliterals.append(f"calc_result_{result}") + testtags.append(f"calc_result_{result}") # Needles for Contacts for hashname in ("jlJmL", "7XGzO", "ps61y", "OvXj~", "GqYOp", "VEFrP"): - testliterals.append(f"contacts_name_{hashname}") - testliterals.append(f"contacts_contact_listed_{hashname}") - testliterals.append(f"contacts_contact_existing_{hashname}") - testliterals.append(f"contacts_contact_doubled_{hashname}") - testliterals.append(f"contacts_contact_altered_{hashname}") - testliterals.append(f"contacts_contact_added_{hashname}") + testtags.append(f"contacts_name_{hashname}") + testtags.append(f"contacts_contact_listed_{hashname}") + testtags.append(f"contacts_contact_existing_{hashname}") + testtags.append(f"contacts_contact_doubled_{hashname}") + testtags.append(f"contacts_contact_altered_{hashname}") + testtags.append(f"contacts_contact_added_{hashname}") for info in ("home", "personal", "work"): - testliterals.append(f"contacts_label_{info}") + testtags.append(f"contacts_label_{info}") # Needles for Maps for location in ("vilnius", "denali", "wellington", "poysdorf", "pune"): - testliterals.append(f"maps_select_{location}") - testliterals.append(f"maps_found_{location}") - testliterals.append(f"maps_info_{location}") + testtags.append(f"maps_select_{location}") + testtags.append(f"maps_found_{location}") + testtags.append(f"maps_info_{location}") # Needles for Gnome Panel for percentage in ("zero", "fifty", "hundred"): - testliterals.append(f"panel_volume_bar_{percentage}") - testliterals.append(f"panel_volume_indicator_{percentage}") + testtags.append(f"panel_volume_bar_{percentage}") + testtags.append(f"panel_volume_indicator_{percentage}") # variable-y in custom_change_device but we only have one value -testliterals.append("anaconda_part_device_sda") +testtags.append("anaconda_part_device_sda") # for Anaconda help related needles. -testliterals.extend(f"anaconda_help_{fsys}" for fsys in ('install_destination', +testtags.extend(f"anaconda_help_{fsys}" for fsys in ('install_destination', 'installation_progress', 'keyboard_layout', 'language_support', 'network_host_name', 'root_password', 'select_packages', 'installation_source', 'time_date', 'user_creation', 'language_selection', 'language', 'summary_link')) -testliterals.extend(f"anaconda_main_hub_{fsys}" for fsys in ('language_support', 'selec_packages', +testtags.extend(f"anaconda_main_hub_{fsys}" for fsys in ('language_support', 'selec_packages', 'time_date', 'create_user','keyboard_layout')) # retcode tracker @@ -151,6 +203,7 @@ ret = 0 unused = [] noimg = [] noneedle = [] +needletags = set() needlepaths = glob.glob(f"{NEEDLEPATH}/**/*.json", recursive=True) for needlepath in needlepaths: @@ -160,10 +213,21 @@ for needlepath in needlepaths: noimg.append(needlepath) with open(needlepath, "r") as needlefh: needlejson = json.load(needlefh) - if any(tag in testliterals for tag in needlejson["tags"]): + needletags.update(needlejson["tags"]) + if any(tag in testtags for tag in needlejson["tags"]): continue unused.append(needlepath) +# check for tags with no needle +tagnoneedle = realtesttags - needletags +# allowlist +# this is a weird one: we theoretically know this needle exists but we +# don't know what it looks like because the function has been broken +# as long as the test has existed. once +# https://gitlab.gnome.org/GNOME/gnome-font-viewer/-/issues/64 is +# fixed we can create this needle and drop this entry +tagnoneedle.discard("fonts_c059_installed") + # reverse check, for images without a needle file imgpaths = glob.glob(f"{NEEDLEPATH}/**/*.png", recursive=True) for imgpath in imgpaths: @@ -189,4 +253,10 @@ if noneedle: for img in noneedle: print(img) +if tagnoneedle: + ret += 8 + print("Tag(s) without needle(s) found!") + for tag in tagnoneedle: + print(tag) + sys.exit(ret) diff --git a/cleanup-needles.py b/cleanup-needles.py new file mode 100755 index 00000000..fc6ba40e --- /dev/null +++ b/cleanup-needles.py @@ -0,0 +1,77 @@ +#!/usr/bin/python3 + +"""Dumb script for staging an old needle cleanup. + +First, log in to the stg server (always use stg because some needles may +only be used on Power) and do this: + psql -h db-openqa01.iad2.fedoraproject.org -U openqastg -d openqa-stg -W +and enter the password (from /etc/openqa/database.ini). Now do this, +changing the date in the `select` command to an appropriate one - a few +months before the current date: + \o oldneedles.txt + select filename from needles where date_trunc('day', last_matched_time) < '2023-01-01' or last_matched_time is null; + ctrl-d (to quit) +now copy oldneedles.txt off the server, and run this script on it. It will +stage a git commit that removes all the identified needles. +""" + +import datetime +import os +import subprocess +import sys + +try: + fname = sys.argv[1] +except IndexError: + sys.exit("You must pass the file with the query output as the argument!") + +with open(fname, "r", encoding="utf-8") as fh: + lines = fh.readlines() + +# strip the column name and underlines +lines = lines[2:] + +# needles we know we want to keep around: these are ones that are +# encountered very rarely, but which *do* have a legitimate reason +# to exist. often the exact needle we have would not match any more +# anyway, but keeping it around prevents check-needles.py from +# complaining, and gives us a template to create a working needle +# from the next time we encounter the rare situation +keeplist = ( + # 'system crashes to emergency mode / dracut' cases + "emergency_rescue_nopassword", + "root_logged_in-dracut", + # text install just doesn't fail this way very often + "anaconda_main_hub_text_unfinished", + # upgrade tests don't fail on system-upgrade reboot very often + "upgrade_fail", + # text install just doesn't fail very often + "anaconda_text_error", +) + +changed = False +for line in lines: + # query output lines start with a space, when we hit one that does + # not, we've done all the query output lines and can quit + if not line.startswith(" "): + break + line = line.strip() + if any(keep in line for keep in keeplist): + continue + line = f"needles/{line}" + # the db has needles we deleted before in it, so let's not bother + # trying to remove them again + if os.path.exists(line): + basename = line[:-4] + command = ("git", "rm", f"{basename}json", f"{basename}png") + subprocess.run(command) + changed = True + +# create the commit +if changed: + today = datetime.date.today().strftime("%Y-%m-%d") + command = ("git", "commit", "-a", "-s", "-m", f"Old needle cleanup {today}") + subprocess.run(command) +else: + print("Nothing to do!") + sys.exit() diff --git a/lib/freeipa.pm b/lib/freeipa.pm index 21ff8a4d..2f31d488 100644 --- a/lib/freeipa.pm +++ b/lib/freeipa.pm @@ -36,8 +36,8 @@ sub start_webui { my ($user, $password) = @_; # if we logged in as 'admin' we should land on the admin 'Active # users' screen, otherwise we should land on the user's own page - my $user_screen = "freeipa_webui_user"; - $user_screen = "freeipa_webui_users" if ($user eq 'admin'); + my $user_screen = "freeipa_webui_user"; # testtag + $user_screen = "freeipa_webui_users" if ($user eq 'admin'); # testtag type_string "startx /usr/bin/firefox -width 1024 -height 768 https://ipa001.test.openqa.fedoraproject.org\n"; assert_screen ["freeipa_webui_login", $user_screen], 60; diff --git a/tests/_software_selection.pm b/tests/_software_selection.pm index 9a9edb83..53d032f6 100644 --- a/tests/_software_selection.pm +++ b/tests/_software_selection.pm @@ -39,14 +39,14 @@ sub run { # select desired environment # go through the list 20 times at max (to prevent infinite loop when it's missing) - for (my $i = 0; !check_screen("anaconda_" . $packageset . "_highlighted", 1) && $i < 20; $i++) { + for (my $i = 0; !check_screen("anaconda_${packageset}_highlighted", 1) && $i < 20; $i++) { send_key "down"; } send_key "spc"; # check that desired environment is selected - assert_screen "anaconda_" . $packageset . "_selected"; + assert_screen "anaconda_${packageset}_selected"; assert_and_click "anaconda_spoke_done"; diff --git a/tests/desktop_update_graphical.pm b/tests/desktop_update_graphical.pm index 62d6bcb4..120309f2 100644 --- a/tests/desktop_update_graphical.pm +++ b/tests/desktop_update_graphical.pm @@ -76,7 +76,8 @@ sub run { sleep 5; } - my $tags = ['desktop_package_tool_update_download', 'desktop_package_tool_update_apply']; + my $tags = ['desktop_package_tool_update_download', 'desktop_package_tool_update_apply']; # testtag + # Apply updates, moving the mouse every two minutes to avoid the # idle screen blank kicking in. Depending on whether this is KDE # or GNOME and what Fedora release, we may see 'apply' right away,