def setUp(self):
""" Set up data used in tests """
self.config = mock.Mock()
def get(*args, **kwargs):
if args[0] == 'app_path':
return 'tests/latte/'
elif args[0] == 'sleep_time':
return 5
elif args[0] == 'stats_path':
return 'stats/'
elif args[0] == 'ignore_keywords':
return ['ignore', 'keyw']
self.config.get = get
self.config.getint = get
engine = create_engine('sqlite:///:memory:')
self.session = sessionmaker(bind=engine)
Base.metadata.create_all(engine)
self.timetracker = TimeTracker(self.config, session=self.session())
