Skip to content
Snippets Groups Projects
Commit 2cb264ad authored by Waleed Akbar's avatar Waleed Akbar
Browse files

Slight changes to verify KPI_DB test

parent dd371d2d
No related branches found
No related tags found
2 merge requests!294Release TeraFlowSDN 4.0,!207Resolve "(CTTC) Separation of Monitoring"
...@@ -34,16 +34,14 @@ class Kpi_DB: ...@@ -34,16 +34,14 @@ class Kpi_DB:
# self.create_database(self.db_engine) # to add database # self.create_database(self.db_engine) # to add database
self.Session = sessionmaker(bind=self.db_engine) self.Session = sessionmaker(bind=self.db_engine)
@staticmethod def create_database(self) -> None:
def create_database(engine : sqlalchemy.engine.Engine) -> None: if not sqlalchemy_utils.database_exists(self.db_engine.url):
if not sqlalchemy_utils.database_exists(engine.url): LOGGER.info("Database created. {:}".format(self.db_engine.url))
LOGGER.info("Database created. {:}".format(engine.url)) sqlalchemy_utils.create_database(self.db_engine.url)
sqlalchemy_utils.create_database(engine.url)
@staticmethod def drop_database(self) -> None:
def drop_database(engine : sqlalchemy.engine.Engine) -> None: if sqlalchemy_utils.database_exists(self.db_engine.url):
if sqlalchemy_utils.database_exists(engine.url): sqlalchemy_utils.drop_database(self.db_engine.url)
sqlalchemy_utils.drop_database(engine.url)
def create_tables(self): def create_tables(self):
try: try:
......
...@@ -21,5 +21,8 @@ LOGGER = logging.getLogger(__name__) ...@@ -21,5 +21,8 @@ LOGGER = logging.getLogger(__name__)
def test_verify_Tables(): def test_verify_Tables():
LOGGER.info('>>> test_verify_Tables : START <<< ') LOGGER.info('>>> test_verify_Tables : START <<< ')
kpiDBobj = Kpi_DB() kpiDBobj = Kpi_DB()
kpiDBobj.drop_database()
kpiDBobj.verify_tables()
kpiDBobj.create_database()
kpiDBobj.create_tables() kpiDBobj.create_tables()
kpiDBobj.verify_tables() kpiDBobj.verify_tables()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment