Update tests

tests/test_cracker.py

@@ -7,11 +7,11 @@
 
 import unittest
 import fenjing
+from flask import render_template_string
 from typing import Union
 from fenjing.cracker import Cracker
 from fenjing.submitter import BaseSubmitter, HTTPResponse
 from fenjing import const
-import jinja2
 import logging
 
 
@@ -24,7 +24,7 @@ def submit_raw(self, raw_payload: str) -> Union[HTTPResponse, None]:
         if not self.waf(raw_payload):
             return HTTPResponse(200, "Nope")
         try:
-            result = jinja2.Template(source=raw_payload).render()
+            result = render_template_string(raw_payload)
             return HTTPResponse(200, result)
         except Exception:
             return HTTPResponse(500, "Internal Server Error")

tests/test_full_payload_gen.py

@@ -4,9 +4,11 @@
 
 import fenjing
 import logging
+from flask import render_template_string
+
+
 from fenjing import FullPayloadGen, const
 import unittest
-import jinja2
 import random
 
 fenjing.full_payload_gen.logger.setLevel(logging.ERROR)
@@ -45,7 +47,7 @@ def test_string(self):
             assert payload is not None
             # why?
             # cause the stupid type checker thinks the 'payload' below would still be None
-            result = jinja2.Template(payload).render()
+            result = render_template_string(payload)
             self.assertIn(string, result)
             for word in self.blacklist:
                 self.assertNotIn(word, payload)
@@ -58,7 +60,7 @@ def test_os_popen_read(self):
         assert payload is not None
         # why?
         # cause the stupid type checker thinks the 'payload' below would still be None
-        result = jinja2.Template(payload).render()
+        result = render_template_string(payload)
         self.assertIn("fen jing", result)
         for word in self.blacklist:
             self.assertNotIn(word, payload)
@@ -121,7 +123,7 @@ def test_string(self):
             assert payload is not None
             # why?
             # cause the stupid type checker thinks the 'payload' below would still be None
-            result = jinja2.Template(payload).render()
+            result = render_template_string(payload)
             self.assertIn(string, result)
             for word in self.blacklist:
                 self.assertNotIn(word, payload)
@@ -134,7 +136,7 @@ def test_os_popen_read(self):
         assert payload is not None
         # why?
         # cause the stupid type checker thinks the 'payload' below would still be None
-        result = jinja2.Template(payload).render()
+        result = render_template_string(payload)
         self.assertIn("fen jing", result)
         for word in self.blacklist:
             self.assertNotIn(word, payload)
@@ -196,7 +198,7 @@ def test_string(self):
             assert payload is not None
             # why?
             # cause the stupid type checker thinks the 'payload' below would still be None
-            result = jinja2.Template(payload).render()
+            result = render_template_string(payload)
             self.assertIn(string, result)
             for word in self.blacklist:
                 self.assertNotIn(word, payload)
@@ -209,7 +211,7 @@ def test_os_popen_read(self):
         assert payload is not None
         # why?
         # cause the stupid type checker thinks the 'payload' below would still be None
-        result = jinja2.Template(payload).render()
+        result = render_template_string(payload)
         self.assertIn("fen jing", result)
 
         for word in self.blacklist:
@@ -291,7 +293,7 @@ def test_string(self):
                 assert payload is not None
                 # why?
                 # cause the stupid type checker thinks the 'payload' below would still be None
-                result = jinja2.Template(payload).render()
+                result = render_template_string(payload)
                 self.assertIn(string, result)
                 for word in blacklist:
                     self.assertNotIn(word, payload)
@@ -303,7 +305,7 @@ def test_os_popen_read(self):
             assert payload is not None
             # why?
             # cause the stupid type checker thinks the 'payload' below would still be None
-            result = jinja2.Template(payload).render()
+            result = render_template_string(payload)
             self.assertIn("fen jing", result)
             for word in blacklist:
                 self.assertNotIn(word, payload)

tests/test_payload_gen.py

@@ -4,6 +4,9 @@
 
 import unittest
 import fenjing
+from flask import render_template_string
+
+
 from fenjing.payload_gen import PayloadGenerator
 from fenjing import const
 import logging