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()