def fake():
"Populate tables using fake data"
fake = faker.Faker()
upload_dir = os.path.join(manager.app.instance_path, 'uploads')
if not os.path.isdir(upload_dir):
os.mkdir(upload_dir)
users = [User.generate(fake) for _ in range(0, 10)]
user = users[0]
for obj in users:
db.session.add(obj)
db.session.commit()
pdfdata = base64.decodestring(EMPTY_PDF.strip())
docs = [Document.generate(pdfdata) for _ in range(0, 10)]
for doc in docs:
comments = [Comment.generate(fake, doc.id) for _ in range(0, 4)]
annotations = [Annotation.generate(fake, doc.id, user.id)
for _ in range(0, 2)]
for obj in docs + comments + annotations:
db.session.add(obj)
db.session.commit()
def test_generate_models(self):
fake = faker.Faker()
user = User.generate(fake)
doc = Document.generate('pdfdata')
comm = Comment.generate(fake, doc)
ann = Annotation.generate(fake, doc, user)
