More model fixes

3 files changed, 18 insertions, 5 deletions

diff --git a/backend/Elements.Data/Models/Suggestion.cs b/backend/Elements.Data/Models/Suggestion.cs
index 3d626fc..bd936a4 100644
--- a/backend/Elements.Data/Models/Suggestion.cs
+++ b/backend/Elements.Data/Models/Suggestion.cs
@@ -5,9 +5,12 @@ public class Suggestion
     public int Id { get; set; }
     public required string Name { get; init; }
     public required byte[] Icon { get; init; }
-    public required Element FirstIngredient { get; init; }
-    public required Element SecondIngredient { get; init; }
+    public required int FirstElementId { get; init; }
+    public required int SecondElementId { get; init; }
     public required DateTime VotingEnd { get; init; }
-    public required int Votes { get; set; } = 1;
     public required int UserId { get; init; }
+
+    public Element FirstElement { get; set; } = null!;
+    public Element SecondElement { get; set; } = null!;
+    public ICollection<Vote> Votes { get; } = new List<Vote>();
 }
\ No newline at end of file
diff --git a/backend/Elements.Data/Models/User.cs b/backend/Elements.Data/Models/User.cs
index 63be88a..bf3231b 100644
--- a/backend/Elements.Data/Models/User.cs
+++ b/backend/Elements.Data/Models/User.cs
@@ -5,6 +5,8 @@ public class User
     public int Id { get; set; }
     public required string GoogleId { get; init; }
     public required string Name { get; set; }
-    public required ICollection<Element> Elements { get; set; }
-    public required ICollection<Suggestion> Suggestions { get; set; }
+    
+    public ICollection<Element> Elements { get; } = new List<Element>();
+    public ICollection<Suggestion> Suggestions { get; } = new List<Suggestion>();
+    public ICollection<Vote> Votes { get; } = new List<Vote>();
 }
\ No newline at end of file
diff --git a/backend/Elements.Data/Models/Vote.cs b/backend/Elements.Data/Models/Vote.cs
new file mode 100644
index 0000000..2979692
--- /dev/null
+++ b/backend/Elements.Data/Models/Vote.cs
@@ -0,0 +1,8 @@
+namespace Elements.Data.Models;
+
+public class Vote
+{
+    public int Id { get; set; }
+    public required int UserId { get; init; }
+    public required int SuggestionId { get; set; }
+}
\ No newline at end of file