From d1e4d7cc8f8ded14e588df248d9541b6c34f8f9b Mon Sep 17 00:00:00 2001
From: Usman Nasir <usman@cyberpersons.com>
Date: Sun, 24 Nov 2019 14:32:20 +0500
Subject: [PATCH] test_checkRewriteFile

---
 websiteFunctions/tests.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/websiteFunctions/tests.py b/websiteFunctions/tests.py
index 93d5d67f1..7e286b3f8 100755
--- a/websiteFunctions/tests.py
+++ b/websiteFunctions/tests.py
@@ -376,6 +376,31 @@ phpinfo();
 
         self.assertEqual(response['configstatus'], 1)
 
+    def test_checkRewriteFile(self):
+
+        ## Inserting something in rewrite file
+
+        data_ret = {'virtualHost': 'cyberpanel.xyz', 'rewriteRules': 'hello world'}
+
+        response = self.MakeRequest('websites/saveRewriteRules', data_ret)
+
+        self.assertEqual(response['rewriteStatus'], 1)
+
+        ## Check again
+
+        data_ret = {'virtualHost': 'cyberpanel.xyz'}
+
+        response = self.MakeRequest('websites/getRewriteRules', data_ret)
+
+        self.assertEqual(response['rewriteStatus'], 1)
+
+        exists = 0
+
+        if response['rewriteRules'].find('hello world') > -1:
+            exists = 1
+
+        self.assertEqual(exists, 1)
+