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()