From e51817c849520892519296f458e92b9ab3585fdf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Robert=20Sch=C3=A4fer?= <git@roschaefer.de>
Date: Mon, 18 Feb 2019 23:49:54 +0100
Subject: [PATCH] Seed follow relations between users

---
 src/seed/data/index.js         |  1 -
 src/seed/data/users-follows.js | 26 --------------------------
 src/seed/factories/users.js    | 11 +++++++++++
 src/seed/seed-db.js            | 12 ++++++------
 4 files changed, 17 insertions(+), 33 deletions(-)
 delete mode 100644 src/seed/data/users-follows.js

diff --git a/src/seed/data/index.js b/src/seed/data/index.js
index 12008ccda..d4a99817e 100644
--- a/src/seed/data/index.js
+++ b/src/seed/data/index.js
@@ -1,6 +1,5 @@
 export default {
 
-  UserFollows: require('./users-follows.js').default,
   UserFriends: require('./users-friends.js').default,
 
   Organization: require('./organizations.js').default,
diff --git a/src/seed/data/users-follows.js b/src/seed/data/users-follows.js
deleted file mode 100644
index af6467e7f..000000000
--- a/src/seed/data/users-follows.js
+++ /dev/null
@@ -1,26 +0,0 @@
-export default function (data) {
-  return `
-    mutation {
-      u1_follow_u2: AddUserFollowing(
-        from: { id: "u1" },
-        to: { id: "u2" }
-      ) { from { id } }
-      u2_follow_u1: AddUserFollowing(
-        from: { id: "u2" },
-        to: { id: "u1" }
-      ) { from { id } }
-      u2_follow_u3: AddUserFollowing(
-        from: { id: "u2" },
-        to: { id: "u3" }
-      ) { from { id } }
-      u2_follow_u4: AddUserFollowing(
-        from: { id: "u2" },
-        to: { id: "u4" }
-      ) { from { id } }
-      u4_follow_u2: AddUserFollowing(
-        from: { id: "u4" },
-        to: { id: "u2" }
-      ) { from { id } }
-    }
-  `
-}
diff --git a/src/seed/factories/users.js b/src/seed/factories/users.js
index efea9a413..c9659f72c 100644
--- a/src/seed/factories/users.js
+++ b/src/seed/factories/users.js
@@ -10,6 +10,7 @@ export default function (params) {
     avatar = faker.internet.avatar(),
     badgeIds = [],
     blacklistedUserIds = [],
+    followedUserIds = [],
     disabled = false,
     deleted = false
   } = params
@@ -32,6 +33,15 @@ export default function (params) {
       `
   })
 
+  const followedUserRelations = followedUserIds.map((followedUserId) => {
+    return `
+      ${id}_follow_${followedUserId}: AddUserFollowing(
+        from: { id: "${id}" },
+        to: { id: "${followedUserId}" }
+      ) { from { id } }
+      `
+  });
+
   return `
     mutation {
       ${id}: CreateUser(
@@ -51,6 +61,7 @@ export default function (params) {
       }
     ${badgeRelations.join('\n')}
     ${blacklistedUserRelations.join('\n')}
+    ${followedUserRelations.join('\n')}
     }
   `
 }
diff --git a/src/seed/seed-db.js b/src/seed/seed-db.js
index 78b26eb21..8943c1406 100644
--- a/src/seed/seed-db.js
+++ b/src/seed/seed-db.js
@@ -17,12 +17,12 @@ import seed from './data'
     ])
 
     await Promise.all([
-      create('user', { id: 'u1', name: 'Peter Lustig',       role: 'admin',     badgeIds: ['b6'], email: 'admin@example.org' }),
-      create('user', { id: 'u2', name: 'Bob der Baumeister', role: 'moderator', badgeIds: ['b5'], email: 'moderator@example.org' }),
-      create('user', { id: 'u3', name: 'Jenny Rostock',      role: 'user',      badgeIds: ['b4'], email: 'user@example.org' }),
-      create('user', { id: 'u4', name: 'Tick',               role: 'user',      badgeIds: ['b3'], email: 'tick@example.org' }),
-      create('user', { id: 'u5', name: 'Trick',              role: 'user',      badgeIds: ['b2'], email: 'trick@example.org' }),
-      create('user', { id: 'u6', name: 'Track',              role: 'user',      badgeIds: ['b1'], email: 'track@example.org' }),
+      create('user', { id: 'u1', name: 'Peter Lustig',       role: 'admin',     followedUserIds: [    ], badgeIds: ['b6'], email: 'admin@example.org' }),
+      create('user', { id: 'u2', name: 'Bob der Baumeister', role: 'moderator', followedUserIds: ['u1'], badgeIds: ['b5'], email: 'moderator@example.org' }),
+      create('user', { id: 'u3', name: 'Jenny Rostock',      role: 'user',      followedUserIds: ['u2'], badgeIds: ['b4'], email: 'user@example.org' }),
+      create('user', { id: 'u4', name: 'Tick',               role: 'user',      followedUserIds: ['u3'], badgeIds: ['b3'], email: 'tick@example.org' }),
+      create('user', { id: 'u5', name: 'Trick',              role: 'user',      followedUserIds: ['u4'], badgeIds: ['b2'], email: 'trick@example.org' }),
+      create('user', { id: 'u6', name: 'Track',              role: 'user',      followedUserIds: ['u5'], badgeIds: ['b1'], email: 'track@example.org' }),
       create('user', { id: 'u7', name: 'Dagobert', role: 'user', badgeIds: ['b1', 'b2'], blacklistedUserIds: ['u4', 'u5', 'u6'], email: 'dagobert@example.org' })
     ])