# -*- coding: utf-8 -*- from __future__ import unicode_literals 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 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 Package data_ret = {'databaseWebsite': 'cyberpanel.xyz', 'dbName': 'hello', 'dbUsername': 'hello', 'dbPassword': 'helloworld', 'webUserName': 'admin'} response = self.MakeRequest('dataBases/submitDBCreation', data_ret) logging.writeToFile(str(response)) self.assertEqual(response['status'], 1) ## Modify Package data_ret = {'packageName': 'admin_HelloWorld', 'diskSpace': 500, 'bandwidth': 50, 'dataBases': 500, 'ftpAccounts': 50, 'emails': 50, 'allowedDomains': 50, 'allowFullDomain': 1} response = self.MakeRequest('packages/saveChanges', data_ret) logging.writeToFile(str(response)) self.assertEqual(response['status'], 1) ## Modify Confirm data_ret = {'packageName': 'admin_HelloWorld'} response = self.MakeRequest('packages/submitModify', data_ret) logging.writeToFile(str(response)) self.assertEqual(response['modifyStatus'], 1) self.assertEqual(response['dataBases'], 500) self.assertEqual(response['diskSpace'], 500) ## Delete Package data_ret = {'packageName': 'admin_HelloWorld'} response = self.MakeRequest('packages/submitDelete', data_ret) logging.writeToFile(str(response)) self.assertEqual(response['status'], 1)