Revize 204bd181

Cleanup in seeding

Backend/Core/Seeding/Seeder.cs

@@ -16 +16 @@
         {
             if (!context.Users.Any(u => u.Role == ERole.ADMINISTRATOR))
             {
-                DocumentContent content = new DocumentContent()
-                {
-                    Content = "Dummy content"
-                };
-                context.DocumentContents.Add(content);
-
-                User user = new User()
-                {
-                    Username = "testuser",
-                    Name = "Test",
-                    Surname = "User",
-                    Password = BCrypt.Net.BCrypt.HashPassword("password"),
-                    Role = ERole.ANNOTATOR
-                };
-                context.Users.Add(user);
-
-                User admin = new User()
-                {
-                    Username = "admin",
-                    Name = "Admin",
-                    Surname = "User",
-                    Password = BCrypt.Net.BCrypt.HashPassword("admin"),
-                    Role = ERole.ADMINISTRATOR
-                };
-                context.Users.Add(admin);
-
-                Document document = new Document()
-                {
-                    Content = content,
-                    Name = "Test document",
-                    DateAdded = DateTime.Now,
-                    Length = 10,
-                    RequiredAnnotations = 3,
-                    UserAdded = user
-                };
-                context.Documents.Add(document);
-                context.SaveChanges();
-
-                Annotation annotation = new Annotation()
-                {
-                    DateAssigned = DateTime.Now,
-                    DateLastChanged = DateTime.Now,
-                    Document = document,
-                    Note = "",
-                    State = EState.NEW,
-                    User = user,
-                    UserAssigned = user
-                };
-                context.Annotations.Add(annotation);
-
-                Class clss = new Class()
-                {
-                    Color = "rgb(...)",
-                    Description = "Dummy description",
-                    Name = "Dummy class name"
-                };
-                context.Classes.Add(clss);
-
-                annotation.Classes.Add(clss);
-
-                TagCategory category = new TagCategory()
-                {
-                    Name = "Dummy category",
-                    Color = "rgb(...)",
-                    Description = "Dummy category description"
-                };
-                context.TagCategories.Add(category);
-
-                Tag tag = new Tag()
-                {
-                    Name = "Dummy tag",
-                    Category = category,
-                    Description = "Dummy tag",
-                    Color = "rgb(...)"
-                };
-                context.Tags.Add(tag);
-                SubTag subTag = new SubTag()
-                {
-                    Description = "Dummy subtag",
-                    Name = "Dummy subtag",
-                    Tag = tag
-                };
-                context.SubTags.Add(subTag);
-
-                AnnotationTag annotationTag = new AnnotationTag()
-                {
-                    Annotation = annotation,
-                    Instance = 0,
-                    Length = 1,
-                    Position = 0,
-                    Tag = tag,
-                    Note = "Dummy note"
-                };
-                context.AnnotationTags.Add(annotationTag);
-
+                var users = AddUsers(context);
+                var classes = AddClasses(context);
+                DummyTags.AddDummyTags(context);
                 context.SaveChanges();
             }
         }
@@ -131 +39 @@
                 context.SaveChanges();
             }
         }
+
+        private static List<Class> AddClasses(DatabaseContext context)
+        {
+            List<Class> classes = new List<Class>();
+            Class class1 = new Class()
+            {
+                Name = "Class1",
+                Description = "Class1 description",
+                Color = "Class 1 color"
+            };
+            context.Classes.Add(class1);
+            classes.Add(class1);
+            return classes;
+        }
+
+        private static List<User> AddUsers(DatabaseContext context)
+        {
+            List<User> users = new List<User>();
+            User annotator1 = new User()
+            {
+                Username = "annotator1",
+                Name = "Anno",
+                Surname = "Tator1",
+                Password = BCrypt.Net.BCrypt.HashPassword("password"),
+                Role = ERole.ANNOTATOR
+            };
+            context.Users.Add(annotator1);
+            users.Add(annotator1);
+
+            User annotator2 = new User()
+            {
+                Username = "annotator2",
+                Name = "Anno",
+                Surname = "Tator2",
+                Password = BCrypt.Net.BCrypt.HashPassword("password"),
+                Role = ERole.ANNOTATOR
+            };
+            context.Users.Add(annotator2);
+            users.Add(annotator2);
+
+            User annotator3 = new User()
+            {
+                Username = "annotator3",
+                Name = "Anno",
+                Surname = "Tator3",
+                Password = BCrypt.Net.BCrypt.HashPassword("password"),
+                Role = ERole.ANNOTATOR
+            };
+            context.Users.Add(annotator3);
+            users.Add(annotator3);
+
+            User annotator4 = new User()
+            {
+                Username = "annotator4",
+                Name = "Anno",
+                Surname = "Tator4",
+                Password = BCrypt.Net.BCrypt.HashPassword("password"),
+                Role = ERole.ANNOTATOR
+            };
+            context.Users.Add(annotator4);
+            users.Add(annotator4);
+
+            User annotator5 = new User()
+            {
+                Username = "annotator5",
+                Name = "Anno",
+                Surname = "Tator5",
+                Password = BCrypt.Net.BCrypt.HashPassword("password"),
+                Role = ERole.ANNOTATOR
+            };
+            context.Users.Add(annotator5);
+            users.Add(annotator5);
+
+            User admin = new User()
+            {
+                Username = "admin",
+                Name = "Admin",
+                Surname = "User",
+                Password = BCrypt.Net.BCrypt.HashPassword("admin"),
+                Role = ERole.ADMINISTRATOR
+            };
+            context.Users.Add(admin);
+            users.Add(admin);
+
+            return users;
+        }
     }
 }