web-base/test/test.py

#!/usr/bin/env python3

import os
import sys
import argparse
import random
import string
import unittest

import mysql.connector
import psycopg2

from installTest import InstallTestCase
from apiTest import ApiTestCase

CONFIG_FILES = ["../core/Configuration/Database.class.php","../core/Configuration/JWT.class.php","../core/Configuration/Mail.class.php"]

def randomName(length):
   letters = string.ascii_lowercase
   return ''.join(random.choice(letters) for i in range(length))

def performTest(args):
    suite = unittest.TestSuite()
    suite.addTest(InstallTestCase(args))
    suite.addTest(ApiTestCase())
    runner = unittest.TextTestRunner()
    runner.run(suite)

def testMysql(args):

    # Create a temporary database
    if args.database is None:
        args.database = "webbase_test_%s" % randomName(6)
        config = {
            "host": args.host,
            "user": args.username,
            "passwd": args.password,
            "port": args.port
        }

        print("[ ] Connecting to dbms…")
        connection = mysql.connector.connect(**config)
        print("[+] Success")
        cursor = connection.cursor()
        print("[ ] Creating temporary databse %s" % args.database)
        cursor.execute("CREATE DATABASE %s" % args.database)
        print("[+] Success")

    # perform test
    try:
        args.type = "mysql"
        performTest(args)
    finally:
        if cursor is not None:
            print("[ ] Deleting temporary database")
            cursor.execute("DROP DATABASE %s" % args.database)
            cursor.close()
            print("[+] Success")

    if connection is not None:
        print("[ ] Closing connection…")
        connection.close()

def testPostgres(args):

    # Create a temporary database
    if args.database is None:
        args.database = "webbase_test_%s" % randomName(6)
        connection_string = "host=%s port=%d user=%s password=%s" % (args.host, args.port, args.username, args.password)

        print("[ ] Connecting to dbms…")
        connection = psycopg2.connect(connection_string)
        connection.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
        print("[+] Success")
        cursor = connection.cursor()
        print("[ ] Creating temporary databse %s" % args.database)
        cursor.execute("CREATE DATABASE %s" % args.database)
        print("[+] Success")

    # perform test
    try:
        args.type = "postgres"
        performTest(args)
    finally:
        if cursor is not None:
            print("[ ] Deleting temporary database")
            cursor.execute("DROP DATABASE %s" % args.database)
            cursor.close()
            print("[+] Success")

    if connection is not None:
        print("[ ] Closing connection…")
        connection.close()

if __name__ == "__main__":

    supportedDbms = {
        "mysql": 3306,
        "postgres": 5432,
        "oracle": 1521
    }

    parser = argparse.ArgumentParser(description='Web-Base database test suite')
    parser.add_argument('dbms', metavar='DBMS', type=str, help='the dbms to setup, must be one of: %s' % ", ".join(supportedDbms.keys()))
    parser.add_argument('--username', '-u', type=str, help='the username used for connecting to the dbms, default: root', default='root')
    parser.add_argument('--password', '-p', type=str, help='the password used for connecting to the dbms, default: (empty)', default='')
    parser.add_argument('--host', '-H', type=str, help='the host where the dbms is running on, default: localhost', default='localhost')
    parser.add_argument('--port', '-P', type=int, help='the port where the dbms is running on, default: (depends on dbms)')
    parser.add_argument('--database', '-d', type=str, help='the name of the database for the test suite, default: randomly chosen and created')
    parser.add_argument('--force', action='store_const', help='Delete existing configuration files', const=True)

    args = parser.parse_args()
    if args.dbms not in supportedDbms:
        print("Unsupported dbms. Supported values: %s" % ", ".join(supportedDbms.keys()))
        exit(1)

    for f in CONFIG_FILES:
        if os.path.isfile(f):
            if not args.force:
                print("File %s exists. The testsuite is required to perform tests on a clean environment. Specify --force to delete those files" % f)
                exit(1)
            else:
                os.remove(f)

    if args.port is None:
        args.port = supportedDbms[args.dbms]

    if args.dbms == "mysql":
        testMysql(args)
    elif args.dbms == "postgres":
        testPostgres(args)

    for f in CONFIG_FILES:
        if os.path.isfile(f):
            os.remove(f)