82d417b9e6
Fresh start for the StarlingX automation framework. Change-Id: Ie265e0791024f45f71faad6315c2b91b022934d1
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from framework.database.connection.database_operation_manager import DatabaseOperationManager
|
|
from framework.database.objects.test_plan import TestPlan
|
|
from framework.logging.automation_logger import get_logger
|
|
from psycopg2.extras import RealDictCursor
|
|
|
|
|
|
class TestPlanOperation:
|
|
"""
|
|
Class for Test Plan Operations
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.database_operation_manager = DatabaseOperationManager()
|
|
|
|
def get_test_plan(self, test_plan_id) -> TestPlan:
|
|
"""
|
|
Gets the test plan with the given id
|
|
Args:
|
|
test_plan_id (): the test plan id
|
|
|
|
Returns: a test plan object
|
|
|
|
"""
|
|
|
|
query = f"SELECT * FROM test_plan WHERE test_plan_id={test_plan_id}"
|
|
|
|
results = self.database_operation_manager.execute_query(query, cursor_factory=RealDictCursor)
|
|
|
|
if results:
|
|
if len(results) > 1:
|
|
get_logger().log_info(f"WARNING: We have found more than one result matching the test plan id: {test_plan_id}")
|
|
result = results[0]
|
|
test_plan = TestPlan(result['test_plan_name'], result['description'], result['run_type_id'])
|
|
test_plan.set_test_plan_id(result['test_plan_id'])
|
|
test_plan.set_locked(result['locked'])
|
|
|
|
return test_plan
|
|
|
|
raise ValueError(f"There is no entry for test_plan_id={test_plan_id}")
|