From c6367f582721d2186eaab61ba874c9f2328c3fe7 Mon Sep 17 00:00:00 2001
From: Usman Nasir <usman@cyberpersons.com>
Date: Wed, 27 Nov 2019 13:11:19 +0500
Subject: [PATCH] complete api tests

---
 api/tests.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++-
 api/views.py |  4 +--
 2 files changed, 82 insertions(+), 4 deletions(-)

diff --git a/api/tests.py b/api/tests.py
index 68c4071d2..abc81ac56 100755
--- a/api/tests.py
+++ b/api/tests.py
@@ -67,4 +67,84 @@ class TestAPI(TestCase):
         if self.MakeRequestRaw('http://api.cyberpanel.xyz').find('404') > -1:
             exists = 1
 
-        self.assertEqual(exists, 1)
\ No newline at end of file
+        self.assertEqual(exists, 1)
+
+    def test_getUserInfo(self):
+
+        ## Login
+
+        data_ret = {'adminUser': 'admin', 'adminPass': '1234567', 'username': 'admin'}
+        response = self.MakeRequest('getUserInfo', data_ret)
+
+        self.assertEqual(response['status'], 1)
+
+    def test_changeUserPassAPI(self):
+
+        ## Login
+
+        data_ret = {'adminUser': 'admin', 'adminPass': '1234567', 'websiteOwner': 'admin', 'ownerPassword': '1234567'}
+        response = self.MakeRequest('changeUserPassAPI', data_ret)
+
+        self.assertEqual(response['changeStatus'], 1)
+
+    def test_changePackageAPI(self):
+
+        ## Login
+
+        data_ret = {'adminUser': 'admin', 'adminPass': '1234567', 'websiteName': 'cyberpanel.xyz', 'packageName': 'Default'}
+        response = self.MakeRequest('changePackageAPI', data_ret)
+
+        self.assertEqual(response['changePackage'], 1)
+
+
+    def test_submitWebsiteStatus(self):
+
+        ## Suspend  check
+        data_ret = {'adminUser': 'admin', 'adminPass': '1234567', 'websiteName': 'cyberpanel.xyz', 'state': 'Suspend'}
+
+        response = self.MakeRequest('submitWebsiteStatus', data_ret)
+
+        time.sleep(5)
+
+        self.assertEqual(response['websiteStatus'], 1)
+
+        exists = 0
+
+        if self.MakeRequestRaw('http://cyberpanel.xyz').find('404') > -1 or self.MakeRequestRaw(
+                'http://cyberpanel.xyz').find('Access to this resource on the server is denied!') > -1:
+            exists = 1
+
+        self.assertEqual(exists, 1)
+
+        suspend = 0
+        import os
+        if os.path.exists('/usr/local/lsws/conf/vhosts/cyberpanel.xyz-suspended'):
+            suspend = 1
+
+        self.assertEqual(suspend, 1)
+
+        ## Unsuspend  check
+
+        data_ret = {'adminUser': 'admin', 'adminPass': '1234567', 'websiteName': 'cyberpanel.xyz', 'state': 'Unsuspend'}
+
+        response = self.MakeRequest('submitWebsiteStatus', data_ret)
+
+        time.sleep(5)
+
+        self.assertEqual(response['websiteStatus'], 1)
+
+        exists = 0
+
+        if self.MakeRequestRaw('http://cyberpanel.xyz').find('CyberPanel') > -1:
+            exists = 1
+
+        self.assertEqual(exists, 1)
+
+        suspend = 0
+        import os
+        if os.path.exists('/usr/local/lsws/conf/vhosts/cyberpanel.xyz'):
+            suspend = 1
+
+        self.assertEqual(suspend, 1)
+
+
diff --git a/api/views.py b/api/views.py
index 158d87da0..a87067f5d 100755
--- a/api/views.py
+++ b/api/views.py
@@ -93,7 +93,7 @@ def getUserInfo(request):
 
             try:
                 user = Administrator.objects.get(userName=username)
-                data_ret = {'status': 0,
+                data_ret = {'status': 1,
                             'firstName': user.firstName,
                             'lastName': user.lastName,
                             'email': user.email,
@@ -145,8 +145,6 @@ def changeUserPassAPI(request):
             websiteOwn.password = hashPassword.hash_password(ownerPassword)
             websiteOwn.save()
 
-
-
             data_ret = {'changeStatus': 1, 'error_message': "None"}
             json_data = json.dumps(data_ret)
             return HttpResponse(json_data)