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)