Skip to content

Commit

Permalink
Manager -> Loadmaster
Browse files Browse the repository at this point in the history
  • Loading branch information
leondz committed Sep 9, 2024
1 parent 83f3e69 commit 5e9b450
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 11 deletions.
4 changes: 2 additions & 2 deletions garak/payloads.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,8 +125,8 @@ def __init__(self, name, path) -> None:
self._load()


class Manager:
"""The payload Manager manages payload groups. It'll inventory them on disk,
class Loadmaster:
"""The payload Loadmaster manages payload groups. It'll inventory them on disk,
manage enumeration of payloads (optionally given a payload type specification),
and load them up."""

Expand Down
2 changes: 1 addition & 1 deletion garak/probes/grandma.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class Win10(Probe):
doc_uri = "https://www.independent.co.uk/tech/chatgpt-microsoft-windows-11-grandma-exploit-b2360213.html"

def __init__(self, config_root=_config):
win10_payload = garak.payloads.Manager().get(
win10_payload = garak.payloads.Loadmaster().get(
"product keys / windows 10"
) # connecting probes w/ payloads can grow to become complex
product_names = win10_payload.payloads
Expand Down
16 changes: 8 additions & 8 deletions tests/test_payloads.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,13 @@


PAYLOAD_NAMES = list(
garak.payloads.Manager().search()
garak.payloads.Loadmaster().search()
) # default includes local custom payloads to help test them


@pytest.mark.parametrize("payload_name", PAYLOAD_NAMES)
def test_core_payloads(payload_name):
f = garak.payloads.Manager()
f = garak.payloads.Loadmaster()
p = f.get(payload_name)
assert isinstance(
p, garak.payloads.PayloadGroup
Expand All @@ -37,22 +37,22 @@ def payload_typology():
return types


def test_payload_Manager():
f = garak.payloads.Manager()
assert isinstance(f, garak.payloads.Manager)
def test_payload_Loadmaster():
f = garak.payloads.Loadmaster()
assert isinstance(f, garak.payloads.Loadmaster)


def test_payload_list():
f = garak.payloads.Manager()
f = garak.payloads.Loadmaster()
for payload_name in f.search():
assert isinstance(
payload_name, str
), "Payload names from Manager().search() must be strings"
), "Payload names from Loadmaster().search() must be strings"


@pytest.mark.parametrize("payload_name", PAYLOAD_NAMES)
def test_payloads_have_valid_tags(payload_name, payload_typology):
f = garak.payloads.Manager()
f = garak.payloads.Loadmaster()
p = f.get(payload_name)
for typee in p.types:
assert (
Expand Down

0 comments on commit 5e9b450

Please sign in to comment.