def setup_class(cls): if cls.run_setup_bind is None: assert cls.bind is not None assert cls.run_deletes in (None, 'each') if cls.run_inserts == 'once': assert cls.run_deletes is None if cls.tables is None: cls.tables = adict() if cls.other_artifacts is None: cls.other_artifacts = adict() if cls.bind is None: setattr(cls, 'bind', cls.setup_bind()) if cls.metadata is None: setattr(cls, 'metadata', sa.MetaData()) if cls.metadata.bind is None: cls.metadata.bind = cls.bind if cls.run_define_tables == 'once': cls.define_tables(cls.metadata) cls.metadata.create_all() cls.tables.update(cls.metadata.tables) if cls.run_inserts == 'once': cls._load_fixtures() cls.insert_data()
def setup_class(cls): if cls.run_setup_classes == 'each': assert cls.run_setup_mappers != 'once' assert cls.run_deletes in (None, 'each') if cls.run_inserts == 'once': assert cls.run_deletes is None assert not hasattr(cls, 'keep_mappers') assert not hasattr(cls, 'keep_data') if cls.tables is None: cls.tables = adict() if cls.classes is None: cls.classes = adict() if cls.other_artifacts is None: cls.other_artifacts = adict() if cls.metadata is None: setattr(cls, 'metadata', sa.MetaData()) if cls.metadata.bind is None: cls.metadata.bind = getattr(cls, 'engine', config.db) if cls.run_define_tables == 'once': cls.define_tables(cls.metadata) cls.metadata.create_all() cls.tables.update(cls.metadata.tables) if cls.run_setup_classes == 'once': baseline = subclasses(BasicEntity) cls.setup_classes() cls._register_new_class_artifacts(baseline) if cls.run_setup_mappers == 'once': baseline = subclasses(BasicEntity) cls.setup_mappers() cls._register_new_class_artifacts(baseline) if cls.run_inserts == 'once': cls._load_fixtures() cls.insert_data()