From 5f9f134eb0b5ea40dbf2accc24bb757f456572bf Mon Sep 17 00:00:00 2001
From: David Mehren <git@herrmehren.de>
Date: Wed, 19 May 2021 21:09:15 +0200
Subject: [PATCH] Seed: Generate multiple notes and authorships

Signed-off-by: David Mehren <git@herrmehren.de>
---
 src/seed.ts | 47 ++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 36 insertions(+), 11 deletions(-)

diff --git a/src/seed.ts b/src/seed.ts
index 4329d8c0a..6a76de437 100644
--- a/src/seed.ts
+++ b/src/seed.ts
@@ -5,6 +5,8 @@
  */
 
 import { createConnection } from 'typeorm';
+import { Author } from './authors/author.entity';
+import { Session } from './users/session.entity';
 import { User } from './users/user.entity';
 import { Note } from './notes/note.entity';
 import { Revision } from './revisions/revision.entity';
@@ -37,22 +39,45 @@ createConnection({
     Tag,
     AuthToken,
     Identity,
+    Author,
+    Session,
   ],
   synchronize: true,
   logging: false,
+  dropSchema: true,
 })
   .then(async (connection) => {
-    const user = User.create('hardcoded', 'Test User');
-    const note = Note.create(undefined, 'test');
-    const revision = Revision.create(
-      'This is a test note',
-      'This is a test note',
-    );
-    note.revisions = Promise.all([revision]);
-    note.userPermissions = [];
-    note.groupPermissions = [];
-    user.ownedNotes = [note];
-    await connection.manager.save([user, note, revision]);
+    const users = [];
+    users.push(User.create('hardcoded', 'Test User 1'));
+    users.push(User.create('hardcoded_2', 'Test User 2'));
+    users.push(User.create('hardcoded_3', 'Test User 3'));
+    const notes: Note[] = [];
+    notes.push(Note.create(undefined, 'test'));
+    notes.push(Note.create(undefined, 'test2'));
+    notes.push(Note.create(undefined, 'test3'));
+
+    for (let i = 0; i < 3; i++) {
+      const author = connection.manager.create(Author, Author.create(1));
+      const user = connection.manager.create(User, users[i]);
+      author.user = user;
+      const revision = Revision.create(
+        'This is a test note',
+        'This is a test note',
+      );
+      const authorship = Authorship.create(author, 1, 42);
+      revision.authorships = [authorship];
+      notes[i].revisions = Promise.all([revision]);
+      notes[i].userPermissions = [];
+      notes[i].groupPermissions = [];
+      user.ownedNotes = [notes[i]];
+      await connection.manager.save([
+        notes[i],
+        user,
+        revision,
+        authorship,
+        author,
+      ]);
+    }
     const foundUser = await connection.manager.findOne(User);
     if (!foundUser) {
       throw new Error('Could not find freshly seeded user. Aborting.');