Add some more flask tests whilst I'm here

tonybaloney · Feb 5, 2020 · 2e1c476 · 2e1c476
1 parent 0a05861
commit 2e1c476
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/test/java/security/validators/FlaskDebugModeInspectionTest.kt b/src/test/java/security/validators/FlaskDebugModeInspectionTest.kt
@@ -89,4 +89,24 @@ class FlaskDebugModeInspectionTest: SecurityTestTask() {
         """.trimIndent()
         testCodeCallExpression(code, 0, FlaskDebugModeCheck, "test.py", FlaskDebugModeInspection())
     }
+
+    @Test
+    fun `test flask not imported`(){
+        var code = """
+            from banana import Flask
+            
+            app = Flask()
+            blah.run()
+        """.trimIndent()
+        testCodeCallExpression(code, 0, FlaskDebugModeCheck, "test.py", FlaskDebugModeInspection())
+    }
+
+    @Test
+    fun `test not reference parent`(){
+        var code = """
+            from flask import Flask
+            "1".run()
+        """.trimIndent()
+        testCodeCallExpression(code, 0, FlaskDebugModeCheck, "test.py", FlaskDebugModeInspection())
+    }
 }