def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() db.create_all() Ruolo.insert_roles() Utente.insert_test_users()
def setUp(self): self.app = create_app('testing') self.app_context = self.app.app_context() self.app_context.push() db.create_all() Tag.insert_test_tags() Corso.insert_test_corsi() Serata.insert_test_serate() Ruolo.insert_roles() Utente.insert_test_users() self.client = self.app.test_client()
def create_test_db(): print("Start creating test db") FLASK_CONFIG = os.getenv("FLASK_CONFIG", "None") app = create_app(FLASK_CONFIG) app_context = app.app_context() app_context.push() from project.serate.models import Serata from project.corsi.models import Corso from project.tags.models import Tag from project.ruoli.models import Ruolo from project.utenti.models import Utente from project.blog.models import Post from project.commenti.models import Comment from random import randint from faker import Faker try: user_list = Utente.query.all() course_list = Corso.query.all() post_list = Post.query.all() comment_list = Comment.query.all() post_list = Post.query.all() ruolo_list = Ruolo.query.all() print("DB Tables already exists") except Exception as message: print(f"No db data exist, inserting them:") def users(count=100): fake = Faker("it_IT") i = 0 while i < count: u = Utente( email=fake.email(), username=fake.user_name(), password="******", confirmed=True, name=fake.name(), location=fake.city(), about_me=fake.text(), member_since=fake.past_date(), ) db.session.add(u) try: db.session.commit() i += 1 except IntegrityError: db.session.rollback() def posts(count=100): fake = Faker("it_IT") user_count = Utente.query.count() for i in range(count): u = Utente.query.offset(randint(0, user_count - 1)).first() p = Post(body=fake.text(), timestamp=fake.past_date(), author=u) db.session.add(p) db.session.commit() def comments(count=100): fake = Faker("it_IT") user_count = Utente.query.count() post_count = Post.query.count() for i in range(count): u = Utente.query.offset(randint(0, user_count - 1)).first() p = Post.query.offset(randint(0, post_count - 1)).first() c = Comment(body=fake.text(), timestamp=fake.past_date(), post=p, author=u) db.session.add(c) db.session.commit() print("Creating structure") db.create_all() db.session.commit() print("Creating roles") Ruolo.insert_roles() print("Creating fake users") users(2) print("Creating test users") Utente.insert_test_users() print("Creating tags") Tag.insert_test_tags() print("Creating corsi") Corso.insert_test_corsi() print("Creating serate") Serata.insert_test_serate() print("Creating posts fake") posts(3) print("Creating commenti fake") comments(3) print("\nDB Dummy data inserted succesfully") db.session.remove() app_context.pop()
if CREATE_ALL: # Utilizzo dell'application factory app = create_app('development') app_context = app.app_context() app_context.push() print("Creating structure") db.create_all() print("Creating roles") Ruolo.insert_roles() print("Creating fake users") users(20) print("Creating test users") Utente.insert_test_users() print("Creating tags") Tag.insert_test_tags() print("Creating corsi") Corso.insert_test_corsi() print("Creating serate") Serata.insert_test_serate() print("Creating posts fake") posts() print("Creating commenti fake") comments()