From 341b5bb2347c30344662675936b90b325efe5520 Mon Sep 17 00:00:00 2001 From: Vijay Venkatesh Kumar Date: Thu, 17 Nov 2022 14:51:38 -0500 Subject: Heartbeat code refactoring code optimization & test improvement Issue-ID: DCAEGEN2-2953 Signed-off-by: Vijay Venkatesh Kumar Change-Id: I99229d966c13ad666ac994ab5a582aeeaa306639 Signed-off-by: Vijay Venkatesh Kumar --- miss_htbt_service/mod/trapd_vnf_table.py | 250 ------------------------------- 1 file changed, 250 deletions(-) delete mode 100644 miss_htbt_service/mod/trapd_vnf_table.py (limited to 'miss_htbt_service/mod/trapd_vnf_table.py') diff --git a/miss_htbt_service/mod/trapd_vnf_table.py b/miss_htbt_service/mod/trapd_vnf_table.py deleted file mode 100644 index f9fecbb..0000000 --- a/miss_htbt_service/mod/trapd_vnf_table.py +++ /dev/null @@ -1,250 +0,0 @@ -# ============LICENSE_START======================================================= -# Copyright (c) 2017-2022 AT&T Intellectual Property. All rights reserved. -# Copyright (c) 2019 Pantheon.tech. All rights reserved. -# Copyright (c) 2020 Deutsche Telekom. All rights reserved. -# Copyright (c) 2021 Samsung Electronics. All rights reserved. -# Copyright (c) 2021 Fujitsu Ltd. -# ================================================================================ -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# ============LICENSE_END========================================================= -# -# Author Kiran Mandal (km386e) - -""" -trapd_vnf_table verifies the successful creation of DB Tables. -""" -import logging -import os -import yaml -import os.path as path -import time -import subprocess - -import db_monitoring as dbmon -import htbtworker as pm -import misshtbtd as db -import cbs_polling as cbs - -prog_name = os.path.basename(__file__) -hb_properties_file = path.abspath(path.join(__file__, "../../config/hbproperties.yaml")) -_logger = logging.getLogger(__name__) - - -def hb_properties(): - # Read the hbproperties.yaml for postgress and CBS related data - s = open(hb_properties_file, "r") - a = yaml.full_load(s) - ip_address = a["pg_ipAddress"] - port_num = a["pg_portNum"] - user_name = a["pg_userName"] - password = a["pg_passwd"] - db_name = a["pg_dbName"] - db_name = db_name.lower() - cbs_polling_required = a["CBS_polling_allowed"] - cbs_polling_interval = a["CBS_polling_interval"] - s.close() - return ip_address, port_num, user_name, password, db_name, cbs_polling_required, cbs_polling_interval - - -def verify_DB_creation_1(user_name, password, ip_address, port_num, db_name): - connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name) - try: - _db_status = pm.db_table_creation_check(connection_db, "vnf_table_1") - except Exception: - return None - - return _db_status - - -def verify_DB_creation_2(user_name, password, ip_address, port_num, db_name): - - connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name) - try: - _db_status = pm.db_table_creation_check(connection_db, "vnf_table_2") - except Exception: - return None - - return _db_status - - -def verify_DB_creation_hb_common(user_name, password, ip_address, port_num, db_name): - - connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name) - try: - _db_status = pm.db_table_creation_check(connection_db, "hb_common") - except Exception: - return None - - return _db_status - - -def verify_cbspolling(): - os.environ["pytest"] = "test" - os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static" - cbs.poll_cbs(10) - os.unsetenv("pytest") - os.unsetenv("SERVICE_NAME") - - -def verify_fetch_json_file(): - os.environ["pytest"] = "test" - os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static" - os.environ["CONSUL_HOST"] = "localhost" - os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static" - try: - db.fetch_json_file() - result = True - except Exception: - result = False - print(result) - os.unsetenv("pytest") - os.unsetenv("SERVICE_NAME") - os.unsetenv("CONSUL_HOST") - os.unsetenv("HOSTNAME") - return result - - -def verify_misshtbtdmain(): - os.environ["pytest"] = "test" - os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static" - os.environ["CONSUL_HOST"] = "localhost" - os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static" - - try: - db.main() - result = True - except Exception: - result = False - print(result) - os.unsetenv("pytest") - os.unsetenv("SERVICE_NAME") - os.unsetenv("CONSUL_HOST") - os.unsetenv("HOSTNAME") - return result - - -def verify_dbmonitoring(): - os.environ["pytest"] = "test" - os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static" - os.environ["CONSUL_HOST"] = "localhost" - os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static" - try: - jsfile = db.fetch_json_file() - ip_address, port_num, user_name, password, db_name, cbs_polling_required, cbs_polling_interval = hb_properties() - hbc_pid, hbc_state, hbc_src_name, hbc_time = db.read_hb_common( - user_name, password, ip_address, port_num, db_name - ) - dbmon.db_monitoring(hbc_pid, jsfile, user_name, password, ip_address, port_num, db_name) - result = True - except Exception as e: - print("Message process error - %s" % e) - result = False - print(result) - os.unsetenv("pytest") - os.unsetenv("SERVICE_NAME") - os.unsetenv("CONSUL_HOST") - os.unsetenv("HOSTNAME") - return result - - -def verify_dbmon_startup(): - try: - p = subprocess.Popen(["./miss_htbt_service/db_monitoring.py"], stdout=subprocess.PIPE, shell=True) - time.sleep(1) - except Exception: - return None - return True - - -def verify_sendControlLoop_VNF_ONSET(): - try: - pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/" - _CL_return = dbmon.sendControlLoopEvent( - "ONSET", - pol_url, - "1.0", - "vFireWall", - "pscope", - "VNF", - "srcname1", - 1541234567, - "SampleCLName", - "1.0", - "genVnfName", - ) - except Exception: - return None - return _CL_return - - -def verify_sendControlLoop_VM_ONSET(): - try: - pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/" - _CL_return = dbmon.sendControlLoopEvent( - "ONSET", - pol_url, - "1.0", - "vFireWall", - "pscope", - "VM", - "srcname1", - 1541234567, - "SampleCLName", - "1.0", - "genVnfName", - ) - except Exception: - return None - return _CL_return - - -def verify_sendControlLoop_VNF_ABATED(): - try: - pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/" - _CL_return = dbmon.sendControlLoopEvent( - "ABATED", - pol_url, - "1.0", - "vFireWall", - "pscope", - "VNF", - "srcname1", - 1541234567, - "SampleCLName", - "1.0", - "genVnfName", - ) - except Exception: - return None - return _CL_return - - -def verify_sendControlLoop_VM_ABATED(): - try: - pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/" - _CL_return = dbmon.sendControlLoopEvent( - "ABATED", - pol_url, - "1.0", - "vFireWall", - "pscope", - "VM", - "srcname1", - 1541234567, - "SampleCLName", - "1.0", - "genVnfName", - ) - except Exception: - return None - return _CL_return -- cgit 1.2.3-korg