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
102
103
104
105
106
|
import unittest
from unittest.mock import patch, PropertyMock
import api.v1.spamcheck_pb2 as spam
from app import config
from app.spammable import generic
from tests.app.helpers import MockContext, MockML
class TestGeneric(unittest.TestCase):
def test_generic_attrs(self):
obj = spam.Generic(text="test", type="random_generic")
g = generic.Generic(obj, MockContext())
to_dict = g.to_dict()
self.assertEqual(
"test", to_dict["description"], "Generic description not set correctly"
)
self.assertEqual(
"random_generic", g.type(), "Generic type not set correctly"
)
def test_verdict_project_not_allowed(self):
g = generic.Generic(spam.Generic(text="test"), MockContext())
g._project_allowed = False
v = g.verdict()
self.assertEqual(
spam.SpamVerdict.NOOP, v.verdict, "Disallowed project should return NOOP"
)
def test_score(self):
g = generic.Generic(spam.Generic(text="test"), MockContext())
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.calculate_verdict(0.39),
"Confidence less than 0.4 should be allowed",
)
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.calculate_verdict(0.41),
"Confidence between 0.4 and 0,5 should be allowed",
)
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.calculate_verdict(0.55),
"Confidence between 0.5 and 0.9 should be allowed",
)
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.calculate_verdict(0.9),
"Confidence of 0.9 or greater should be allowed",
)
def test_verdict(self):
generic.classifier = MockML(1.0)
g = generic.Generic(spam.Generic(text="test"), MockContext())
g._project_allowed = False
self.assertEqual(
spam.SpamVerdict.NOOP,
g.verdict().verdict,
"Disallowed project should return NOOP",
)
g._project_allowed = True
g._email_allowed = True
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.verdict().verdict,
"Allowed email should return ALLOW",
)
g._email_allowed = False
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.verdict().verdict,
"ML inference of 1.0 should be allowed",
)
generic.classifier.set_score(0.1)
self.assertEqual(
spam.SpamVerdict.ALLOW,
g.verdict().verdict,
"ML inference of 0.1 should be allowed",
)
def test_verdict_no_ml(self):
generic.classifier = None
g = generic.Generic(spam.Generic(text="test"), MockContext())
g.project_allowed = True
self.assertEqual(
spam.SpamVerdict.NOOP,
g.verdict().verdict,
"Generic ML not loaded should return NOOP",
)
def test_generic_property(self):
g = generic.Generic(spam.Generic(text="spam"), MockContext())
g.allowed_domains = {"gitlab.com"}
self.assertEqual("spam", g.spammable.text, "Generic text should have been set")
self.assertEqual(False, g._email_allowed, "Blank email should not be allowed")
args = {"user": {"emails": [{"email": "test@gitlab.com", "verified": True}]}}
new_generic = spam.Generic(**args)
g.spammable = new_generic
self.assertEqual(
True,
g._email_allowed,
"Email should be allowed after updating issue",
)
|
