Skip to content

Commit 03440ca

Browse files
committed
Fix cheat sheet parser test expectations on importer branches
1 parent 63528a8 commit 03440ca

1 file changed

Lines changed: 14 additions & 4 deletions

File tree

application/tests/cheatsheets_parser_test.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,23 +34,33 @@ class Repo:
3434
repo.working_dir = loc
3535
cre = defs.CRE(name="blah", id="223-780")
3636
self.collection.add_cre(cre)
37-
with open(os.path.join(os.path.join(loc, "cheatsheets"), "cs.md"), "w") as mdf:
37+
with open(
38+
os.path.join(
39+
os.path.join(loc, "cheatsheets"),
40+
"Secrets_Management_Cheat_Sheet.md",
41+
),
42+
"w",
43+
) as mdf:
3844
mdf.write(cs)
3945
mock_clone.return_value = repo
4046
entries = cheatsheets_parser.Cheatsheets().parse(
4147
cache=self.collection, ph=PromptHandler(database=self.collection)
4248
)
4349
expected = defs.Standard(
4450
name="OWASP Cheat Sheets",
45-
hyperlink="https://github.com/foo/bar/tree/master/cs.md",
51+
hyperlink="https://cheatsheetseries.owasp.org/cheatsheets/Secrets_Management_Cheat_Sheet.html",
4652
section="Secrets Management Cheat Sheet",
47-
links=[defs.Link(document=cre, ltype=defs.LinkTypes.LinkedTo)],
53+
links=[
54+
defs.Link(
55+
document=cre, ltype=defs.LinkTypes.AutomaticallyLinkedTo
56+
)
57+
],
4858
)
4959
self.maxDiff = None
5060
for name, nodes in entries.results.items():
5161
self.assertEqual(name, cheatsheets_parser.Cheatsheets().name)
5262
self.assertEqual(len(nodes), 1)
53-
self.assertCountEqual(expected.todict(), nodes[0].todict())
63+
self.assertEqual(expected.todict(), nodes[0].todict())
5464

5565
cheatsheets_md = """ # Secrets Management Cheat Sheet
5666

0 commit comments

Comments
 (0)