def setUpClass(cls):
"""Set up for the class."""
super().setUpClass()
cls.aws_tables = list(AWS_CUR_TABLE_MAP.values())
cls.ocp_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.all_tables = list(AWS_CUR_TABLE_MAP.values()) + list(OCP_REPORT_TABLE_MAP.values())
cls.creator = ReportObjectCreator(cls.schema)
def setUpClass(cls):
"""Set up for the class."""
super().setUpClass()
cls.aws_tables = list(AWS_CUR_TABLE_MAP.values())
cls.ocp_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.all_tables = list(AWS_CUR_TABLE_MAP.values()) + list(OCP_REPORT_TABLE_MAP.values())
with ReportingCommonDBAccessor() as report_common_db:
cls.column_map = report_common_db.column_map
cls.creator = ReportObjectCreator(cls.schema, cls.column_map)
def setUpClass(cls):
"""Setup for the class."""
super().setUpClass()
cls.aws_tables = list(AWS_CUR_TABLE_MAP.values())
cls.ocp_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.all_tables = list(AWS_CUR_TABLE_MAP.values()) + \
list(OCP_REPORT_TABLE_MAP.values())
report_common_db = ReportingCommonDBAccessor()
cls.column_map = report_common_db.column_map
report_common_db.close_session()
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.accessor = OCPReportDBAccessor(schema=cls.schema)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(cls.schema)
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
with ReportingCommonDBAccessor() as report_common_db:
cls.column_map = report_common_db.column_map
cls.accessor = OCPReportDBAccessor('acct10001', cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.session = cls.accessor._session
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.creator = ReportObjectCreator(cls.accessor, cls.column_map,
cls.report_schema.column_types)
cls.date_accessor = DateAccessor()
billing_start = cls.date_accessor.today_with_timezone('UTC').replace(
day=1)
cls.manifest_dict = {
'assembly_id': '1234',
'billing_period_start_datetime': billing_start,
'num_total_files': 2,
'num_processed_files': 2,
'provider_id': 2
}
cls.manifest_accessor = ReportManifestDBAccessor()
def tearDown(self):
"""Return the database to a pre-test state."""
self.accessor._session.rollback()
aws_tables = list(AWS_CUR_TABLE_MAP.values())
with AWSReportDBAccessor(self.test_schema,
self.column_map) as aws_accessor:
aws_accessor._session.rollback()
for table_name in aws_tables:
tables = aws_accessor._get_db_obj_query(table_name).all()
for table in tables:
aws_accessor._session.delete(table)
aws_accessor.commit()
ocp_tables = list(OCP_REPORT_TABLE_MAP.values())
with OCPReportDBAccessor(self.test_schema,
self.column_map) as ocp_accessor:
for table_name in ocp_tables:
tables = ocp_accessor._get_db_obj_query(table_name).all()
for table in tables:
ocp_accessor._session.delete(table)
ocp_accessor.commit()
manifests = self.manifest_accessor._get_db_obj_query().all()
for manifest in manifests:
self.manifest_accessor.delete(manifest)
self.manifest_accessor.commit()
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.common_accessor = ReportingCommonDBAccessor()
cls.column_map = cls.common_accessor.column_map
cls.accessor = OCPReportDBAccessor(schema='acct10001', column_map=cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(cls.schema, cls.column_map)
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.common_accessor = ReportingCommonDBAccessor()
cls.column_map = cls.common_accessor.column_map
cls.ocp_provider_uuid = '3c6e687e-1a09-4a05-970c-2ccf44b0952e'
cls.accessor = OCPReportDBAccessor(schema='acct10001', column_map=cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(cls.schema, cls.column_map)
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
with ReportingCommonDBAccessor() as report_common_db:
cls.column_map = report_common_db.column_map
cls.accessor = OCPReportDBAccessor(cls.schema, cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.creator = ReportObjectCreator(cls.schema, cls.column_map)
cls.date_accessor = DateAccessor()
cls.manifest_accessor = ReportManifestDBAccessor()
def setUp(self):
"""Shared variables used by ocp common tests."""
super().setUp()
self.accessor = OCPReportDBAccessor(schema=self.schema)
self.provider_accessor = ProviderDBAccessor(
provider_uuid=self.ocp_test_provider_uuid)
self.report_schema = self.accessor.report_schema
self.creator = ReportObjectCreator(self.schema)
self.all_tables = list(OCP_REPORT_TABLE_MAP.values())
self.provider_uuid = self.provider_accessor.get_provider().uuid
reporting_period = self.creator.create_ocp_report_period(
provider_uuid=self.provider_uuid)
report = self.creator.create_ocp_report(
reporting_period, reporting_period.report_period_start)
self.creator.create_ocp_usage_line_item(reporting_period, report)
self.creator.create_ocp_storage_line_item(reporting_period, report)
self.creator.create_ocp_node_label_line_item(reporting_period, report)
def setUp(self):
super().setUp()
self.common_accessor = ReportingCommonDBAccessor()
self.column_map = self.common_accessor.column_map
self.accessor = OCPReportDBAccessor(schema=self.schema,
column_map=self.column_map)
self.provider_accessor = ProviderDBAccessor(
provider_uuid=self.ocp_test_provider_uuid)
self.report_schema = self.accessor.report_schema
self.creator = ReportObjectCreator(self.schema, self.column_map)
self.all_tables = list(OCP_REPORT_TABLE_MAP.values())
self.provider_uuid = self.provider_accessor.get_provider().uuid
reporting_period = self.creator.create_ocp_report_period(
provider_uuid=self.provider_uuid)
report = self.creator.create_ocp_report(
reporting_period, reporting_period.report_period_start)
self.creator.create_ocp_usage_line_item(reporting_period, report)
self.creator.create_ocp_storage_line_item(reporting_period, report)
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.accessor = OCPReportDBAccessor(schema=cls.schema)
cls.all_tables = list(OCP_REPORT_TABLE_MAP.values())
cls.dh = DateHelper()