ucampostgresvro 0.1.0__py3-none-any.whl

ucampostgresvro/__init__.py

@@ -0,0 +1 @@
+VERSION = "0.1.0"

ucampostgresvro/__main__.py

@@ -0,0 +1,134 @@
+import logging
+import sys
+from typing import Dict
+
+from ucampostgresvro import VERSION, utils
+from ucampostgresvro.DBA import DB
+from ucampostgresvro.exceptions import DbException
+from ucampostgresvro.secrets import password
+from ucampostgresvro.tools import DEFAULT_TABLES
+
+
+def setloggerdetail():
+    LOG = logging.getLogger(__name__)
+    stdout_handler = logging.StreamHandler(sys.stdout)
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s [%(levelname)s] %(message)s",
+        datefmt="%m/%d/%Y %H:%M:%S",
+        handlers=[stdout_handler],
+    )
+    return LOG
+
+
+def pre_setupconfig(db_params: Dict[str, str]) -> bool:
+    """Create the Database
+
+    Args:
+        db_params (Dict[str, str]): provide parameters for DB connection.
+
+    Returns:
+        bool: True if creation of database is suceess else False
+    """
+    result = []
+    if not utils.check_table_exists(DEFAULT_TABLES.get("user"), db_params):
+        result.append(utils.create_user_table(DEFAULT_TABLES.get("user"), db_params))
+
+    if not utils.check_table_exists(DEFAULT_TABLES.get("deploymentid"), db_params):
+        result.append(
+            utils.create_deployment_table(DEFAULT_TABLES.get("deploymentid"), db_params)
+        )
+
+    if not utils.check_table_exists(DEFAULT_TABLES.get("proj"), db_params):
+        result.append(
+            utils.create_project_table(
+                DEFAULT_TABLES.get("proj"), DEFAULT_TABLES.get("user"), db_params
+            )
+        )
+
+    if not utils.check_table_exists(DEFAULT_TABLES.get("grant"), db_params):
+        result.append(
+            utils.create_grant_table(
+                DEFAULT_TABLES.get("grant"), DEFAULT_TABLES.get("user"), db_params
+            )
+        )
+
+    if not utils.check_table_exists(DEFAULT_TABLES.get("costing"), db_params):
+        result.append(
+            utils.create_costing_table(
+                DEFAULT_TABLES.get("costing"),
+                DEFAULT_TABLES.get("deploymentid"),
+                DEFAULT_TABLES.get("proj"),
+                DEFAULT_TABLES.get("grant"),
+                db_params,
+            )
+        )
+
+    return False not in result
+
+
+def main():
+    LOG = setloggerdetail()
+    LOG.info(f"VERSION : {VERSION}")
+    db_params = {
+        "dbname": "vrapricing",
+        "user": "postgres",
+        "password": password,
+        "host": "infra-db.srv.uis.cam.ac.uk",  # or your database host
+        "port": "5432",  # default PostgreSQL port
+        "sslmode": "require",  # or 'verify-ca' or 'verify-full' based on your needs
+        "sslrootcert": "./ca.crt",  # path to your client certificate
+    }
+    db = DB(db_params)
+
+    if not pre_setupconfig(db_params):
+        raise DbException("ERROR: Tables are not created successfully")
+
+    # db.insert_vrauser("ll220", "len")
+    # print(db.get_vrauser("ll220"))
+    # db.update_vrauser("ll220", "bda20", 'Ben Argyle')
+    # print(db.get_vrauser())
+    # db.remove_vrauser('bda20')
+    # print(db.get_vrauser())
+
+    # db.insert_deployment_id("1231ee112ad11212")
+    # db.update_deployment_id("1231ee112ad11212", "1231a")
+    # print(db.get_deployment_id("1231a"))
+    # db.remove_deployment_id('1231a')
+    # db.insert_deployment_id("123")
+    # print(db.get_deployment_id())
+
+    # print(db.get_project())
+    # db.insert_project("0001",1,100.0)
+    # db.insert_project("0101",2,100.0)
+    # db.update_project("0001", "0002", 1, 200)
+    # db.remove_project("0002")
+    # db.insert_project("0001",1,100.0)
+    # db.insert_project("0002",1,200.0)
+    # print(db.get_project("0002",1,200))
+
+    # print(db.get_grant())
+    # db.insert_grant("0001",1,100.0)
+    # db.update_grant("0001", "0002", 1, 200)
+    # print(db.get_grant())
+    # db.remove_grant("0002")
+    # print(db.get_grant())
+    # db.insert_grant("0001",1,100.0)
+    # db.insert_grant("0002",1,200.0)
+    # print(db.get_grant("0002",1,200))
+
+    # print(db.get_costing())
+    # db.insert_costing(1, "Initial Resource", project_id=2, grant_id=None)
+    # db.insert_costing(1, "Initial Resource", project_id=None, grant_id=1)
+    # db.update_costing(1, "Duration Expansion", old_grant_id=None,
+    #                 old_project_id=2, grant_id=None, project_id=2)
+    # print(db.get_costing())
+    # print(db.get_costing())
+    # db.remove_costing(1, "Duration Expansion", 4, None)
+    # print(db.get_costing())
+
+    db.closedb()
+
+
+if __name__ == "__main__":
+    main()

ucampostgresvro/ca.crt

@@ -0,0 +1,19 @@
+-----BEGIN CERTIFICATE-----
+MIIDGTCCAgGgAwIBAgIUGSvY0cDL0E2opwi3ovKt0Ctuf0swDQYJKoZIhvcNAQEL
+BQAwGzEZMBcGA1UEAwwQaW5mcmEtZGIgcm9vdCBDQTAgFw0yMTA3MTUxMTMwNTBa
+GA8yMTIxMDYyMTExMzA1MFowGzEZMBcGA1UEAwwQaW5mcmEtZGIgcm9vdCBDQTCC
+ASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAL8Z6OlwQ1b4W2QO+0bhMjxQ
+xnTZeiWg+jOwrbWdOgxcjOt0KB0lCMuIVcBiMkVdh32Ov0q+FpinJ97/ZnZBhiYf
+QqrNYjsfBUsWy/q9x/b0RcSHtkGrtHsU3YloeyzzdB6jIg8SC35p0Ht6BQuYrP07
+NHjvMzZKbS/+PZIDs5fhyQVl6isEy1xvqJ/nDlRg2HItYOd7lzXC46E7AIj2lqIv
+GCUKAszFnf6OsZEwr0XoKeZqgPbrsCY6RqcT3uWUM86C3F8b6wee/9AvapElvWg5
+eMoIgW3oCvpBSgVuCKWj5YEycz5kQ14LIxacocSSoghyhN3ce72gjtOw16ZkKyEC
+AwEAAaNTMFEwHQYDVR0OBBYEFDtnsz4XBSEfkTCuTgAeiGmubTrXMB8GA1UdIwQY
+MBaAFDtnsz4XBSEfkTCuTgAeiGmubTrXMA8GA1UdEwEB/wQFMAMBAf8wDQYJKoZI
+hvcNAQELBQADggEBALiBdm+Ka7R6AQePfMjVNoqPqjueorIhxazcoSNrJkl5+lzc
+lpB8NznWG6XxTJxvXQ2Am2/Gv3SOPZaceTJFD2G54AuQlrH2ic5KUvZJaJQ5uYCt
+DaH5U96O9mJ2GtVMsgPocbTH2lSkCCkfMmiP80lj6Ms/gSa3oWfhacCl8OhcjI+D
+Etz5dGtVbqavp4DK4b8nt29Ok3v4mWZ6UAnCr3ci8JGFMcmPS6Nh+Tqy6nB7yj4Z
+TgjzT3vqtTROfhE25xSAIZFsA0YM4+pTGJ85W9hZewT52HG8K0bV54/EIIOXHFyu
+SnMIaMvdpveWbgzocgJY54H0i9k+dsjC0dal40w=
+-----END CERTIFICATE-----

ucampostgresvro/exceptions.py

@@ -0,0 +1,2 @@
+class DbException(Exception):
+    pass

ucampostgresvro/tests/conftest.py

@@ -0,0 +1,63 @@
+import pytest
+
+from ucampostgresvro.tests import utils as tools
+from ucampostgresvro.exceptions import DbException
+
+
+data_params = tools.db_params
+
+
+@pytest.fixture
+def user_db_fixture():
+    # Setup
+    if tools.check_db_connection(data_params):
+        db = tools.setup_user_table_func(data_params)
+        yield db
+        # tear_down
+        tools.teardown_user_table_func(data_params)
+    else:
+        raise DbException("Could not connect to the database after several retries.")
+
+
+@pytest.fixture
+def deploymentid_db_fixture():
+    # Setup
+    if tools.check_db_connection(data_params):
+        db = tools.setup_deployment_table_func(data_params)
+        yield db
+        tools.teardown_deployment_table_func(data_params)
+    else:
+        raise DbException("Could not connect to the database after several retries.")
+
+
+@pytest.fixture
+def project_db_fixture():
+    # Setup
+    if tools.check_db_connection(data_params):
+        db = tools.setup_project_table_func(data_params)
+        yield db
+        tools.teardown_project_table_func(data_params)
+    else:
+        raise DbException("Could not connect to the database after several retries.")
+
+
+@pytest.fixture
+def grant_db_fixture():
+    # Setup
+    if tools.check_db_connection(data_params):
+        db = tools.setup_grant_table_func(data_params)
+        yield db
+        tools.teardown_grant_table_func(data_params)
+    else:
+        raise DbException("Could not connect to the database after several retries.")
+
+
+@pytest.fixture
+def costing_db_fixture():
+    # Setup
+    if tools.check_db_connection(data_params):
+        db = tools.setup_costing_table_func(data_params)
+        yield db
+        tools.teardown_costing_table_func(data_params)
+    else:
+        raise DbException("Could not connect to the database after several retries.")

ucampostgresvro/tests/dbconnect.py

@@ -0,0 +1,10 @@
+import os
+
+
+db_params = {
+    "dbname": os.environ['POSTGRES_DB'],
+    "user": os.environ['POSTGRES_USER'],
+    "password": os.environ['POSTGRES_PASSWORD'],
+    "host": os.environ['POSTGRES_HOST'],
+    "port": os.environ['POSTGRES_PORT']
+}

ucampostgresvro/tests/test_DB.py

@@ -0,0 +1,339 @@
+from ucampostgresvro.tests.utils import tables
+
+
+def test_user_insertion(user_db_fixture):
+    db = user_db_fixture
+    result = db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    info = db.get_vrauser(None, None, tables.get("user"))
+    assert len(info) == 1
+    assert result
+
+
+def test_user_update(user_db_fixture):
+    db = user_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    result = db.update_vrauser("im530", "ll220", "Len", tables.get("user"))
+    info = db.get_vrauser(None, None, tables.get("user"))
+    assert len(info) == 1
+    assert result
+    assert info[0][1] == "ll220"
+
+
+def test_user_remove(user_db_fixture):
+    db = user_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    result = db.remove_vrauser("im530", tables.get("user"))
+    info = db.get_vrauser(None, None, tables.get("user"))
+    assert len(info) == 0
+    assert result
+
+
+def test_user_fetchall(user_db_fixture):
+    db = user_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    db.insert_vrauser("im532", "Ishan", tables.get("user"))
+    info = db.get_vrauser(None, None, tables.get("user"))
+    assert len(info) == 2
+    assert info[0][1] == "im530"
+    assert info[1][1] == "im532"
+
+
+def test_user_fetch_one(user_db_fixture):
+    db = user_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    db.insert_vrauser("im532", "Ishan", tables.get("user"))
+    info = db.get_vrauser("im530", None, tables.get("user"))
+    assert len(info) == 1
+    assert info[0][1] == "im530"
+
+
+def test_deployment_insertion(deploymentid_db_fixture):
+    db = deploymentid_db_fixture
+    result = db.insert_deployment_id("121212", tables.get("deploymentid"))
+    info = db.get_deployment_id(None, tables.get("deploymentid"))
+    assert len(info) == 1
+    assert result
+
+
+def test_deployment_update(deploymentid_db_fixture):
+    db = deploymentid_db_fixture
+    db.insert_deployment_id("12345", tables.get("deploymentid"))
+    result = db.update_deployment_id("12345", "9876", tables.get("deploymentid"))
+    info = db.get_deployment_id(None, tables.get("deploymentid"))
+    assert len(info) == 1
+    assert result
+    assert info[0][1] == "9876"
+
+
+def test_deployment_remove(deploymentid_db_fixture):
+    db = deploymentid_db_fixture
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    result = db.remove_deployment_id("121212", tables.get("deploymentid"))
+    info = db.get_deployment_id(None, tables.get("deploymentid"))
+    assert len(info) == 0
+    assert result
+
+
+def test_deployment_fetchall(deploymentid_db_fixture):
+    db = deploymentid_db_fixture
+    db.insert_deployment_id("1212", tables.get("deploymentid"))
+    db.insert_deployment_id("1256", tables.get("deploymentid"))
+    info = db.get_deployment_id(None, tables.get("deploymentid"))
+    assert len(info) == 2
+    assert info[0][1] == "1212"
+    assert info[1][1] == "1256"
+
+
+def test_deployment_fetch_one(deploymentid_db_fixture):
+    db = deploymentid_db_fixture
+    db.insert_deployment_id("1212", tables.get("deploymentid"))
+    db.insert_deployment_id("1256", tables.get("deploymentid"))
+    info = db.get_deployment_id("1256", tables.get("deploymentid"))
+    assert len(info) == 1
+    assert info[0][1] == "1256"
+
+
+def test_proj_insertion(project_db_fixture):
+    db = project_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    result = db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    info = db.get_project(None, tables.get("proj"))
+    assert len(info) == 1
+    assert result
+
+
+def test_proj_update(project_db_fixture):
+    db = project_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    result = db.update_project(
+        "123abc", "9876xyz", users[0][0], 220.0, tables.get("proj")
+    )
+    info = db.get_project(None, tables.get("proj"))
+    assert len(info) == 1
+    assert result
+    assert info[0][1] == "9876xyz"
+    assert info[0][3] == 220.0
+
+
+def test_proj_remove(project_db_fixture):
+    db = project_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    info = db.get_project(None, tables.get("proj"))
+    result = db.remove_project("123abc", tables.get("proj"))
+    info = db.get_project(None, tables.get("proj"))
+    assert result
+    assert len(info) == 0
+
+
+def test_proj_fetchall(project_db_fixture):
+    db = project_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    db.insert_project("123xyz", users[0][0], 200.0, tables.get("proj"))
+    info = db.get_project(None, tables.get("proj"))
+    assert len(info) == 2
+    assert info[0][1] == "123abc"
+    assert info[1][1] == "123xyz"
+
+
+def test_proj_fetch_one(project_db_fixture):
+    db = project_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    db.insert_project("123xyz", users[0][0], 200.0, tables.get("proj"))
+    info = db.get_project("123xyz", tables.get("proj"))
+    assert len(info) == 1
+    assert info[0][1] == "123xyz"
+
+
+def test_grant_insertion(grant_db_fixture):
+    db = grant_db_fixture
+    result = db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+    result = db.insert_grant("123zyx", users[0][0], 200.0, tables.get("grant"))
+    info = db.get_grant(None, tables.get("grant"))
+    assert len(info) == 1
+    assert result
+
+
+def test_grant_update(grant_db_fixture):
+    db = grant_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_grant("123abc", users[0][0], 100.0, tables.get("grant"))
+    result = db.update_grant(
+        "123abc", "9876xyz", users[0][0], 220.0, tables.get("grant")
+    )
+    info = db.get_grant(None, tables.get("grant"))
+    assert len(info) == 1
+    assert result
+    assert info[0][1] == "9876xyz"
+    assert info[0][3] == 220.0
+
+
+def test_grant_remove(grant_db_fixture):
+    db = grant_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_grant("123abc", users[0][0], 100.0, tables.get("grant"))
+    result = db.remove_grant("123abc", tables.get("grant"))
+    info = db.get_grant(None, tables.get("grant"))
+    assert len(info) == 0
+    assert result
+
+
+def test_grant_fetchall(grant_db_fixture):
+    db = grant_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_grant("123abc", users[0][0], 100.0, tables.get("grant"))
+    db.insert_grant("123xyz", users[0][0], 200.0, tables.get("grant"))
+    info = db.get_grant(None, tables.get("grant"))
+    assert len(info) == 2
+    assert info[0][1] == "123abc"
+    assert info[1][1] == "123xyz"
+
+
+def test_grant_fetch_one(grant_db_fixture):
+    db = grant_db_fixture
+    db.insert_vrauser("im530", "Ishan", tables.get("user"))
+    users = db.get_vrauser(None, None, tables.get("user"))
+    db.insert_grant("123abc", users[0][0], 100.0, tables.get("grant"))
+    db.insert_grant("123xyz", users[0][0], 200.0, tables.get("grant"))
+    info = db.get_grant("123xyz", tables.get("grant"))
+    assert len(info) == 1
+    assert info[0][1] == "123xyz"
+
+
+def test_costing_insertion(costing_db_fixture):
+    db = costing_db_fixture
+    db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    deploy = db.get_deployment_id(None, tables.get("deploymentid"))
+
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    proj = db.get_project(None, tables.get("proj"))
+
+    result = db.insert_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+    info = db.get_costing(None, None, None, None, tables.get("costing"))
+    assert len(info) == 1
+    assert result
+
+
+def test_costing_update(costing_db_fixture):
+    db = costing_db_fixture
+    db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    deploy = db.get_deployment_id(None, tables.get("deploymentid"))
+
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    proj = db.get_project(None, tables.get("proj"))
+
+    db.insert_grant("998zxc", users[0][0], 100.0, tables.get("grant"))
+    grant = db.get_grant(None, tables.get("grant"))
+
+    db.insert_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+
+    info = db.get_costing(None, None, None, None, tables.get("costing"))
+    result = db.update_costing(
+        deploy[0][0],
+        "Resource Expansion",
+        None,
+        proj[0][0],
+        grant[0][0],
+        None,
+        tables.get("costing"),
+    )
+
+    info = db.get_costing(None, None, None, None, tables.get("costing"))
+    assert len(info) == 1
+    assert result
+    assert info[0][4] == 1
+
+
+def test_costing_remove(costing_db_fixture):
+    db = costing_db_fixture
+    db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    deploy = db.get_deployment_id(None, tables.get("deploymentid"))
+
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    proj = db.get_project(None, tables.get("proj"))
+
+    db.insert_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+
+    result = db.remove_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+
+    info = db.get_costing(None, None, None, None, tables.get("costing"))
+    assert len(info) == 0
+    assert result
+
+
+def test_costing_fetchall(costing_db_fixture):
+    db = costing_db_fixture
+    db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    deploy = db.get_deployment_id(None, tables.get("deploymentid"))
+
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    proj = db.get_project(None, tables.get("proj"))
+
+    db.insert_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+    db.insert_costing(
+        deploy[0][0], "Duration Expansion", proj[0][0], None, tables.get("costing")
+    )
+
+    info = db.get_costing(None, None, None, None, tables.get("costing"))
+    assert len(info) == 2
+    assert info[0][2] == "Resource Expansion"
+    assert info[1][2] == "Duration Expansion"
+
+
+def test_costing_fetch_one(costing_db_fixture):
+    db = costing_db_fixture
+    db.insert_vrauser("im530", "Ishan", "test_user")
+    users = db.get_vrauser(None, None, tables.get("user"))
+
+    db.insert_deployment_id("121212", tables.get("deploymentid"))
+    deploy = db.get_deployment_id(None, tables.get("deploymentid"))
+
+    db.insert_project("123abc", users[0][0], 100.0, tables.get("proj"))
+    proj = db.get_project(None, tables.get("proj"))
+
+    db.insert_costing(
+        deploy[0][0], "Resource Expansion", proj[0][0], None, tables.get("costing")
+    )
+    db.insert_costing(
+        deploy[0][0], "Duration Expansion", proj[0][0], None, tables.get("costing")
+    )
+
+    info = db.get_costing(
+        deploy[0][0], "Duration Expansion", proj[0][0], None, tables.get("costing")
+    )
+    assert len(info) == 1
+    assert info[0][2] == "Duration Expansion"

ucampostgresvro/tests/utils.py

@@ -0,0 +1,116 @@
+import psycopg2
+import time
+
+from ucampostgresvro import utils
+from ucampostgresvro.DBA import DB
+from ucampostgresvro.exceptions import DbException
+from ucampostgresvro.tests.dbconnect import db_params
+
+db_params = db_params
+
+tables = {
+    "user": "test_user",
+    "deploymentid": "test_deployment",
+    "grant": "test_grants",
+    "proj": "test_projects",
+    "costing": "test_costing",
+}
+
+
+def check_db_connection(data_params):
+    retries = 3
+    state = False
+    while retries > 0:
+        try:
+            conn = psycopg2.connect(**db_params)
+            state = True
+            break
+        except psycopg2.OperationalError:
+            time.sleep(2)  # Wait for 2 seconds before retrying
+            retries -= 1
+    if state:
+        conn.close()
+    return state
+
+
+def setup_user_table_func(data_params):
+    utils.create_user_table(tables.get("user"), data_params)
+    db = DB(db_params)
+    return db
+
+
+def teardown_user_table_func(data_params):
+    return utils.drop_table(tables.get("user"), data_params)
+
+
+def setup_deployment_table_func(data_params):
+    utils.create_deployment_table(tables.get("deploymentid"), data_params)
+    db = DB(db_params)
+    return db
+
+
+def teardown_deployment_table_func(data_params):
+    return utils.drop_table(tables.get("deploymentid"), data_params)
+
+
+def setup_grant_table_func(data_params):
+    db = DB(db_params)
+    if utils.create_user_table(tables.get("user"), data_params):
+        if utils.create_grant_table(
+            tables.get("grant"), tables.get("user"), data_params
+        ):
+            return db
+    raise DbException("Creation of the grant DB failed.")
+
+
+def teardown_grant_table_func(data_params):
+    result = []
+    result.append(utils.drop_table(tables.get("grant"), data_params))
+    result.append(utils.drop_table(tables.get("user"), data_params))
+    return False not in result
+
+
+def setup_project_table_func(data_params):
+    db = DB(db_params)
+    if utils.create_user_table(tables.get("user"), data_params):
+        utils.create_project_table(tables.get("proj"), tables.get("user"), data_params)
+        return db
+    raise DbException("Creation of the purchase order DB failed.")
+
+
+def teardown_project_table_func(data_params):
+    result = []
+    result.append(utils.drop_table(tables.get("proj"), data_params))
+    result.append(utils.drop_table(tables.get("user"), data_params))
+    return False not in result
+
+
+def setup_costing_table_func(data_params):
+    db = DB(db_params)
+    if utils.create_user_table(tables.get("user"), data_params):
+        if utils.create_deployment_table(tables.get("deploymentid"), data_params):
+            if utils.create_project_table(
+                tables.get("proj"), tables.get("user"), data_params
+            ):
+                if utils.create_grant_table(
+                    tables.get("grant"), tables.get("user"), data_params
+                ):
+                    if utils.create_costing_table(
+                        tables.get("costing"),
+                        tables.get("deploymentid"),
+                        tables.get("proj"),
+                        tables.get("grant"),
+                        data_params,
+                    ):
+                        return db
+    raise DbException("Creation of the costing DB failed.")
+
+
+def teardown_costing_table_func(data_params):
+    result = []
+    result.append(utils.drop_table(tables.get("costing"), data_params))
+    result.append(utils.drop_table(tables.get("proj"), data_params))
+    result.append(utils.drop_table(tables.get("grant"), data_params))
+    result.append(utils.drop_table(tables.get("user"), data_params))
+    result.append(utils.drop_table(tables.get("deploymentid"), data_params))
+    return False not in result