upd rageywide

Satont · Apr 8, 2023 · fdbe0be · fdbe0be
1 parent 0392fcf
commit fdbe0be
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/ent/migrate/migrations/atlas.sum b/ent/migrate/migrations/atlas.sum
@@ -1,4 +1,4 @@
-h1:TIORzPQ0gH91l3uBNc/QCex+GAAB93rfyynsMuZU0DA=
+h1:+Uq126QTEe4nFfdPa9fjq+ZlDcgw5trXo/I/+9/2Tgc=
 20230327181912_initial.sql h1:L6nniWh3O35p6lwgIG+NrRkkU2n2iG48Be5/zfPAz0I=
 20230401125338_TitleChangeNotification.sql h1:9u5qCYBNNL6RHtLdcW691tRhYyli7/pG2kBxYuHs1fA=
-20230408101010_add_stream_categories_history.sql h1:/zhrGC3To5cKmpw7ThFDDffZyrc6n22dZTBspPPG1TY=
+20230408113330_add_stream_categories_history.sql h1:LeWjpeGt1nRgvVP3/mCXNeWKOuhxBhqAQEpqW9Aadwc=
diff --git a/ent/schema/stream.go b/ent/schema/stream.go
@@ -3,6 +3,7 @@ package schema
 import (
 	"entgo.io/ent"
 	"entgo.io/ent/dialect"
+	"entgo.io/ent/dialect/entsql"
 	"entgo.io/ent/schema/edge"
 	"entgo.io/ent/schema/field"
 	"github.com/google/uuid"
@@ -49,6 +50,8 @@ func (Stream) Edges() []ent.Edge {
 			Required().
 			Unique().
 			Field("channel_id"),
-		edge.To("stream_categories", StreamCategory.Type),
+		edge.To("stream_categories", StreamCategory.Type).Annotations(entsql.Annotation{
+			OnDelete: entsql.Cascade,
+		}),
 	}
 }
diff --git a/internal/db/stream_impl_ent_test.go b/internal/db/stream_impl_ent_test.go
@@ -157,7 +157,7 @@ func TestStreamEntService_GetLatestByChannelID(t *testing.T) {
 				assert.Equal(t, newChannel.ID, stream.ChannelID, "Expects channel_id to be equal.")
 				assert.Equal(t, tt.wantedStreamID, stream.ID, "Expects stream_id to be equal.")
 				assert.Nil(t, stream.EndedAt, "Expects is_live to be equal.")
-				assert.Contains(t, stream.Categories, "Category", "Expects category to be equal.")
+				assert.Contains(t, stream.Categories[0].Name, "Category", "Expects category to be equal.")
 				assert.Contains(t, stream.Titles, "Title", "Expects title to be equal.")
 			}