mirror of
https://github.com/usmannasir/cyberpanel.git
synced 2025-11-03 20:05:58 +01:00
75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
|
||
|
|
from django.test import TestCase
|
||
|
|
import json
|
||
|
|
import requests
|
||
|
|
import urllib3
|
||
|
|
from plogical.CyberCPLogFileWriter import CyberCPLogFileWriter as logging
|
||
|
|
|
||
|
|
urllib3.disable_warnings()
|
||
|
|
|
||
|
|
|
||
|
|
# Create your tests here.
|
||
|
|
|
||
|
|
|
||
|
|
class TestDatabases(TestCase):
|
||
|
|
httpClient = requests.Session()
|
||
|
|
|
||
|
|
def MakeRequest(self, endPoint, data):
|
||
|
|
json_data = json.dumps(data)
|
||
|
|
path = 'https://cyberpanel.xyz:8090/%s' % (endPoint)
|
||
|
|
result = TestDatabases.httpClient.post(path, data=json_data, verify=False)
|
||
|
|
return json.loads(result.text)
|
||
|
|
|
||
|
|
def MakeRequestRaw(self, path):
|
||
|
|
result = requests.get(path)
|
||
|
|
return str(result.text)
|
||
|
|
|
||
|
|
def setupConnection(self):
|
||
|
|
try:
|
||
|
|
import MySQLdb as mysql
|
||
|
|
import MySQLdb.cursors as cursors
|
||
|
|
conn = mysql.connect(user='admin_hello', passwd='helloworld', cursorclass=cursors.SSCursor)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
return conn, cursor
|
||
|
|
except:
|
||
|
|
return 0, 0
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
## Verify login
|
||
|
|
|
||
|
|
data_ret = {'username': 'admin', 'password': '1234567'}
|
||
|
|
response = self.MakeRequest('verifyLogin', data_ret)
|
||
|
|
self.assertEqual(response['loginStatus'], 1)
|
||
|
|
|
||
|
|
def test_submitDBCreation(self):
|
||
|
|
## Create DB
|
||
|
|
|
||
|
|
data_ret = {'databaseWebsite': 'cyberpanel.xyz', 'dbName': 'hello', 'dbUsername': 'hello',
|
||
|
|
'dbPassword': 'helloworld', 'webUserName': 'admin'}
|
||
|
|
|
||
|
|
response = self.MakeRequest('dataBases/submitDBCreation', data_ret)
|
||
|
|
|
||
|
|
self.assertEqual(response['status'], 1)
|
||
|
|
|
||
|
|
## Check connection to database
|
||
|
|
|
||
|
|
connection, cursor = self.setupConnection()
|
||
|
|
|
||
|
|
self.assertNotEqual(connection, 0)
|
||
|
|
|
||
|
|
## Delete db
|
||
|
|
|
||
|
|
data_ret = {'dbName': 'admin_hello'}
|
||
|
|
response = self.MakeRequest('dataBases/submitDatabaseDeletion', data_ret)
|
||
|
|
self.assertEqual(response['status'], 1)
|
||
|
|
|
||
|
|
## Check connection to database
|
||
|
|
|
||
|
|
connection, cursor = self.setupConnection()
|
||
|
|
|
||
|
|
self.assertEqual(connection, 0)
|
||
|
|
|
||
|
|
|