class BaseTest(TestCase): render_templates = False SQLALCHEMY_DATABASE_URI = 'sqlite://' def create_app(self): app.config['TESTING'] = True return app def setUp(self): db.create_all() self.faker = Faker() self.faker.gender = lambda: random.choice(['male', 'female', None]) self.faker.facebook_id = lambda: random.randint(1, 10000000000) def tearDown(self): db.session.remove() db.drop_all() def insert_fixture(self, limit=1): facebook_id = self.faker.facebook_id() users = [] for _ in range(limit): user = FBUser( facebook_id=str(facebook_id), name=self.faker.name(), username=self.faker.username(), gender=self.faker.gender() ) user.add_and_commit() users.append(user) # increment one instead of random to not have conflict facebook_id += 1 return users