1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
from datetime import datetime, timezone
import pytest
from flask import url_for
from sec_certs_page import mail, mongo
from sec_certs_page.fips.tasks import notify
@pytest.fixture(params=["6b86b273ff34fce1", "3c365ff931ecb0e3"])
def certificate(request):
return mongo.db.fips.find_one({"_id": request.param})
@pytest.fixture(params=["all", "vuln"])
def subscription(user, certificate, request):
sub = {
"username": user[0].username,
"timestamp": datetime.now(timezone.utc),
"certificate": {
"name": certificate["web_data"]["module_name"],
"hashid": certificate["_id"],
"type": "fips",
"url": url_for("fips.entry", hashid=certificate["_id"]),
},
"updates": request.param,
"type": "changes",
}
res = mongo.db.subs.insert_one(sub)
sub["_id"] = res.inserted_id
yield sub
mongo.db.subs.delete_one({"_id": res.inserted_id})
def test_send_notification(user, mocker, subscription):
user, password = user
m = mocker.patch.object(mail, "send")
dgst = subscription["certificate"]["hashid"]
if subscription["updates"] == "vuln":
pytest.skip("Skip vuln only sub.")
diffs = list(mongo.db.fips_diff.find({"dgst": dgst}))
notify(str(diffs[-1]["run_id"]))
for call_args in m.call_args_list:
message = call_args.args[0]
if user.email in message.recipients:
break
else:
assert False
@pytest.mark.slow
def test_cve_notification(user, mocker, subscription):
dgst = subscription["certificate"]["hashid"]
diffs = list(mongo.db.fips_diff.find({"type": "change", "dgst": dgst}))
m = mocker.patch.object(mail, "send")
for diff in diffs:
vuln_diff = False
if h := diff["diff"]["__update__"].get("heuristics"):
for action, val in h.items():
if "related_cves" in val:
vuln_diff = True
break
notify(str(diff["run_id"]))
if vuln_diff or subscription["updates"] == "all":
assert m.call_count == 1
else:
assert m.call_count == 0
m.reset_mock()
@pytest.fixture()
def subscription_new(user):
sub = {
"username": user[0].username,
"timestamp": datetime.now(timezone.utc),
"type": "new",
"which": "fips",
}
res = mongo.db.subs.insert_one(sub)
sub["_id"] = res.inserted_id
yield sub
mongo.db.subs.delete_one({"_id": res.inserted_id})
def test_new_certificate_notification(user, mocker, certificate, subscription_new):
user, password = user
dgst = certificate["_id"]
diffs = list(mongo.db.cc_diff.find({"type": "new", "dgst": dgst}))
m = mocker.patch.object(mail, "send")
for diff in diffs:
notify(str(diff["run_id"]))
for call_args in m.call_args_list:
message = call_args.args[0]
if user.email in message.recipients:
break
else:
assert False
|
