Commit 2cb264ad authored by Waleed Akbar's avatar Waleed Akbar
Browse files

Slight changes to verify KPI_DB test

parent dd371d2d

src/kpi_management/kpi_manager/database/service/Kpi_DB.py

+7 −9
Original line number Diff line number Diff line
@@ -34,16 +34,14 @@ class Kpi_DB: 
        # self.create_database(self.db_engine)        # to add database 

        self.Session = sessionmaker(bind=self.db_engine)



    @staticmethod

    def create_database(engine : sqlalchemy.engine.Engine) -> None:

        if not sqlalchemy_utils.database_exists(engine.url):

            LOGGER.info("Database created. {:}".format(engine.url))

            sqlalchemy_utils.create_database(engine.url)

    def create_database(self) -> None:

        if not sqlalchemy_utils.database_exists(self.db_engine.url):

            LOGGER.info("Database created. {:}".format(self.db_engine.url))

            sqlalchemy_utils.create_database(self.db_engine.url)



    @staticmethod

    def drop_database(engine : sqlalchemy.engine.Engine) -> None:

        if sqlalchemy_utils.database_exists(engine.url):

            sqlalchemy_utils.drop_database(engine.url)

    def drop_database(self) -> None:

        if sqlalchemy_utils.database_exists(self.db_engine.url):

            sqlalchemy_utils.drop_database(self.db_engine.url)



    def create_tables(self):

        try:

src/kpi_management/kpi_manager/database/tests/KpiDBtests.py

+3 −0
Original line number Diff line number Diff line
@@ -21,5 +21,8 @@ LOGGER = logging.getLogger(__name__) 
def test_verify_Tables():

    LOGGER.info('>>> test_verify_Tables : START <<< ')

    kpiDBobj = Kpi_DB()

    kpiDBobj.drop_database()

    kpiDBobj.verify_tables()

    kpiDBobj.create_database()

    kpiDBobj.create_tables()

    kpiDBobj.verify_tables()