if 'comp' in self.__dict__:
return
self.comp = {}
- self.comp['main'] = Component(component_name = 'main')
- self.comp['contrib'] = Component(component_name = 'contrib')
+ for name in ('main', 'contrib', 'non-free'):
+ self.comp[name] = Component(component_name = name)
self.session.add_all(self.comp.values())
def setup_locations(self):
self.setup_components()
self.loc = {}
self.loc['main'] = Location( \
- path = '/srv/ftp-master.debian.org/ftp/pool/', \
- component = self.comp['main'])
+ path = fixture('ftp/pool/'), component = self.comp['main'])
self.loc['contrib'] = Location( \
- path = '/srv/ftp-master.debian.org/ftp/pool/', \
- component = self.comp['contrib'])
+ path = fixture('ftp/pool/'), component = self.comp['contrib'])
self.session.add_all(self.loc.values())
def setup_poolfiles(self):
def tearDown(self):
self.session.rollback()
for class_ in self.classes_to_clean():
- self.session.query(class_).delete()
+ for object_ in self.session.query(class_):
+ self.session.delete(object_)
self.session.commit()
# usually there is no need to drop all tables here
#self.metadata.drop_all()