Merge pull request #92 from nineinchnick/issue-63

add missing server isResilient property
OWASP · Mar 24, 2020 · 8cd2312 · 8cd2312
2 parents 4a25881 + dce00a5
commit 8cd2312
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytm/pytm.py b/pytm/pytm.py
@@ -488,6 +488,7 @@ class Server(Element):
     hasAccessControl = varBool(False)
     implementsCSRFToken = varBool(False)
     handlesResourceConsumption = varBool(False)
+    isResilient = varBool(False)
     authenticationScheme = varString("")
     validatesInput = varBool(False)
     validatesHeaders = varBool(False)

diff --git a/tests/test_pytmfunc.py b/tests/test_pytmfunc.py
@@ -263,8 +263,7 @@ def test_DO01(self):
         web = Server("Web Server")
         process1.handlesResourceConsumption = False
         process1.isResilient = False
-        web.handlesResourceConsumption = False
-        web.isResilient = False
+        web.handlesResourceConsumption = True
         ThreatObj = Threat(next(item for item in threats_json if item["SID"] == "DO01"))
         self.assertTrue(ThreatObj.apply(process1))
         self.assertTrue(ThreatObj.apply(web))