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):
"""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 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):
"""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 = AWSReportDBAccessor('acct10001', cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.session = cls.accessor._session
cls.all_tables = list(AWS_CUR_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,
'provider_id': 1,
}
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."""
cls.common_accessor = ReportingCommonDBAccessor()
cls.column_map = cls.common_accessor.column_map
cls.accessor = AWSReportDBAccessor(
schema='acct10001',
column_map=cls.column_map
)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(
cls.accessor,
cls.column_map,
cls.report_schema.column_types
)
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.foreign_key_tables = [
AWS_CUR_TABLE_MAP['bill'],
AWS_CUR_TABLE_MAP['product'],
AWS_CUR_TABLE_MAP['pricing'],
AWS_CUR_TABLE_MAP['reservation']
]
billing_start = datetime.datetime.utcnow().replace(day=1)
cls.manifest_dict = {
'assembly_id': '1234',
'billing_period_start_datetime': billing_start,
'num_total_files': 2,
'provider_id': 1
}
cls.manifest_accessor = ReportManifestDBAccessor()
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.accessor = AWSReportDBAccessor(cls.schema)
cls.report_schema = cls.accessor.report_schema
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.creator = ReportObjectCreator(cls.schema)
cls.date_accessor = DateAccessor()
cls.manifest_accessor = ReportManifestDBAccessor()
def tearDown(self):
"""Tear down the test."""
with AWSReportDBAccessor(schema=self.test_schema,
column_map=self.column_map) as accessor:
for table_name in list(AWS_CUR_TABLE_MAP.values()):
table_obj = getattr(accessor.report_schema, table_name)
accessor._session.query(table_obj).delete()
# for table in tables:
# accessor._session.delete(table)
accessor.commit()
def setUp(self):
"""Set up the test class with required objects."""
super().setUp()
self.accessor = AWSReportDBAccessor(schema=self.schema)
self.all_tables = list(AWS_CUR_TABLE_MAP.values())
self.foreign_key_tables = [
AWS_CUR_TABLE_MAP["bill"],
AWS_CUR_TABLE_MAP["product"],
AWS_CUR_TABLE_MAP["pricing"],
AWS_CUR_TABLE_MAP["reservation"],
]
def setUpClass(cls):
"""Set up the test class with required objects."""
cls.common_accessor = ReportingCommonDBAccessor()
cls.column_map = cls.common_accessor.column_map
cls.accessor = ReportDBAccessor(schema='testcustomer',
column_map=cls.column_map)
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.foreign_key_tables = [
AWS_CUR_TABLE_MAP['bill'], AWS_CUR_TABLE_MAP['product'],
AWS_CUR_TABLE_MAP['pricing'], AWS_CUR_TABLE_MAP['reservation']
]
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 = AWSReportDBAccessor('acct10001', cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.creator = ReportObjectCreator(cls.schema, cls.column_map)
cls.date_accessor = DateAccessor()
cls.manifest_accessor = ReportManifestDBAccessor()
def setUpClass(cls):
"""Set up the test class with required objects."""
super().setUpClass()
cls.accessor = AWSReportDBAccessor(schema=cls.schema)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(cls.schema)
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.foreign_key_tables = [
AWS_CUR_TABLE_MAP["bill"],
AWS_CUR_TABLE_MAP["product"],
AWS_CUR_TABLE_MAP["pricing"],
AWS_CUR_TABLE_MAP["reservation"],
]
def setUp(self):
"""Set up the test class with required objects."""
super().setUp()
self.common_accessor = ReportingCommonDBAccessor()
self.column_map = self.common_accessor.column_map
self.accessor = AWSReportDBAccessor(schema=self.schema, column_map=self.column_map)
self.all_tables = list(AWS_CUR_TABLE_MAP.values())
self.foreign_key_tables = [
AWS_CUR_TABLE_MAP['bill'],
AWS_CUR_TABLE_MAP['product'],
AWS_CUR_TABLE_MAP['pricing'],
AWS_CUR_TABLE_MAP['reservation'],
]
def setUpClass(cls):
"""Set up the test class with required objects."""
cls.common_accessor = ReportingCommonDBAccessor()
cls.column_map = cls.common_accessor.column_map
cls.accessor = AWSReportDBAccessor(schema='acct10001',
column_map=cls.column_map)
cls.report_schema = cls.accessor.report_schema
cls.creator = ReportObjectCreator(cls.accessor, cls.column_map,
cls.report_schema.column_types)
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
cls.foreign_key_tables = [
AWS_CUR_TABLE_MAP['bill'], AWS_CUR_TABLE_MAP['product'],
AWS_CUR_TABLE_MAP['pricing'], AWS_CUR_TABLE_MAP['reservation']
]
def setUpClass(cls):
"""Setup for the class."""
cls.all_tables = list(AWS_CUR_TABLE_MAP.values())
report_common_db = ReportingCommonDBAccessor()
column_map = report_common_db.column_map
report_common_db.close_session()
cls.accessor = ReportDBAccessor(schema='testcustomer',
column_map=column_map)
cls.creator = ReportObjectCreator(
cls.accessor,
column_map,
cls.accessor.report_schema.column_types
)
def setUpClass(cls):
"""Set up the test class with required objects."""
cls.accessor = ReportingCommonDBAccessor()
cls.report_tables = list(AWS_CUR_TABLE_MAP.values())
def setUp(self):
"""Set up the test class with required objects."""
self.report_tables = list(AWS_CUR_TABLE_MAP.values())
def setUp(self):
"""Set up the test class with required objects."""
super().setUp()
self.accessor = ReportingCommonDBAccessor()
self.report_tables = list(AWS_CUR_TABLE_MAP.values())
