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